Skip to content

pytorch_dlc - Small updates and bug fixes #2747

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
n-poulsen merged 10 commits into from
Oct 1, 2024
Merged

pytorch_dlc - Small updates and bug fixes #2747

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
34e350c
added option for gradient masking
n-poulsen Aug 13, 2024
1a0403f
added grad masking parameters for all configs
n-poulsen Aug 13, 2024
c8812e8
default: do not teleport keypoints to heatmaps for DEKR
n-poulsen Sep 12, 2024
e94c5b8
PAF predictor bug fix: assemblies None if no predictions
n-poulsen Sep 30, 2024
22e8311
eval network API - use the config path that was given
n-poulsen Sep 30, 2024
bce546d
add logger tests
n-poulsen Sep 30, 2024
ec22ee8
add detector type when creating training dataset
n-poulsen Sep 30, 2024
dfc6445
fix image logger NaN bug
n-poulsen Sep 30, 2024
a1e60f9
add cropping parameters to metadata
n-poulsen Sep 30, 2024
19aca31
edit detector eval interval
n-poulsen Sep 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
add detector type when creating training dataset
  • Loading branch information
@n-poulsen
n-poulsen committed Sep 30, 2024
commit ec22ee882beceb765e68d1679dfadb8ed33efb6b
1 change: 1 addition & 0 deletions deeplabcut/generate_training_dataset/trainingsetmanipulation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -969,6 +969,7 @@ def create_training_dataset(
num_shuffles,
Shuffles,
net_type=net_type,
detector_type=detector_type,
trainIndices=trainIndices,
testIndices=testIndices,
userfeedback=userfeedback,
Expand Down