Visitar URL original
SuperAnimal Model Updates by n-poulsen · Pull Request #2756 · DeepLabCut/DeepLabCut · GitHub
Skip to content

SuperAnimal Model Updates #2756

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 55 commits into from
Oct 18, 2024
Merged

SuperAnimal Model Updates #2756

Changes from 1 commit
Commits
Show all changes
55 commits
Select commit Hold shift + click to select a range
d6ee416
refactored weight init for better modularity
Sep 18, 2024
59511e3
refactored model downloading
Oct 1, 2024
a0fdeaf
update examples
Oct 1, 2024
58bf846
removed unused code
Oct 1, 2024
23b66f5
various bug fixes
Oct 1, 2024
2764b38
do not create pose_cfg for pytorch models
Oct 1, 2024
881f174
bug fix: keypoint matching
Oct 1, 2024
718d61b
bug fix: keypoint matching
Oct 1, 2024
f1422e1
fixed dlc scorer for zeroshot video
Oct 2, 2024
8f8510f
bug fix - super animal scorer
Oct 2, 2024
44613ab
fix max individuals when updating config
Oct 2, 2024
3a7b161
better check for annotation files
Oct 2, 2024
2a4c6a3
small improvements
Oct 2, 2024
02a279d
update SA detector augmentation
Oct 2, 2024
5c0656f
added SuperBird configs
Oct 2, 2024
7df067f
bug fix: config creation, set method
Oct 2, 2024
8fc6950
bug fixes: memory replay training
Oct 2, 2024
431e913
bug fix: evaluating with memory replay
Oct 2, 2024
04fa197
fix video analysis with memory replay
Oct 2, 2024
42d1081
added SuperBird colormap
Oct 2, 2024
c42f474
added SuperBird to pseudo-label
Oct 2, 2024
819d0e2
raise error when training with memory replay without detector
Oct 2, 2024
da13693
bug fix: raising error when no conversion table
Oct 2, 2024
d939e3c
fix pose config creation - joints for test config
Oct 3, 2024
b0b2208
bug fix
Oct 3, 2024
657b704
add default BU augmentation when Fine-Tuning SA model without detector
Oct 3, 2024
2e6c120
bug fix
Oct 3, 2024
3f87fee
clean video inference. add individuals to metadata.
Oct 3, 2024
e0c8302
bug fix: test pose cfg creation
Oct 3, 2024
a0bff4e
fix make labeled video with memory replay
Oct 3, 2024
fe00587
clean weight init
Oct 3, 2024
d54e7e1
bug fix
Oct 3, 2024
1177ef8
doc improvements and fixes to project configs
Oct 3, 2024
978845b
doc improvements
Oct 3, 2024
e588712
remove extra FIXME
Oct 3, 2024
0221957
doc fixes
Oct 3, 2024
0d008df
fix setup.py
Oct 3, 2024
16e4154
updated SuperAnimal COLAB
Oct 3, 2024
8b18244
updated docs
Oct 3, 2024
f62f1bd
update DLCLibrary version
Oct 3, 2024
26ae7db
fix modelzoo GUI
Oct 3, 2024
714d383
bug fix: tensorflow superanimal video creation
Oct 3, 2024
1b44363
fix cannot pickle memoryview, only use multianimalproject key when an…
Oct 7, 2024
fcab1ce
add cropping argument to SuperAnimal inference
Oct 7, 2024
ad3f7f8
add make_pytorch_test_config to config __init__
Oct 7, 2024
e94ce12
clean superbird config
Oct 7, 2024
480a2ac
do not pin detector dataloader memory
Oct 8, 2024
f18a02f
update SuperAnimal resnet_50 config
Oct 9, 2024
bdab2cd
add mobilenet fasterrcnn superanimal config
Oct 9, 2024
61e9ef6
bug fix: make_labeled_video SA finetune with mem replay, single anima…
Oct 9, 2024
d83503f
fix SuperAnimal memory replay evaluation
Oct 9, 2024
9e04427
update code for new models
Oct 10, 2024
4655b62
build_weight_init with str/path instead of loaded config
Oct 10, 2024
6b0883d
update dlclibrary to use get_available_datasets
Oct 15, 2024
0c6b4e4
updated config when loading SuperAnimal config
Oct 18, 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
bug fix
  • Loading branch information
Niels Poulsen
Niels Poulsen committed Oct 3, 2024
commit d54e7e12abbb580499099d710c0aac2b02b463e7
2 changes: 1 addition & 1 deletion deeplabcut/core/weight_init.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ def to_dict(self) -> dict:
if self.dataset is not None:
data["dataset"] = self.dataset

data["snapshot_path"] = self.snapshot_path
data["snapshot_path"] = str(self.snapshot_path)
if self.detector_snapshot_path is not None:
data["detector_snapshot_path"] = str(self.detector_snapshot_path)

Expand Down