Skip to content

Implement create_pretrained_project() for Pytorch engine #2897

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
maximpavliv merged 14 commits into from
Feb 25, 2025
Merged

Implement create_pretrained_project() for Pytorch engine #2897

Changes from 1 commit
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
f1994da
Tmp fix create_pretrained_project: create TF trainingset metadata
maximpavliv Feb 6, 2025
50a2cb2
Type hints
maximpavliv Feb 17, 2025
94a5308
Split create_pretrained_project() into tf and torch
maximpavliv Feb 18, 2025
7d17382
Add individuals param to create_new_project()
maximpavliv Feb 20, 2025
7791165
Fix list_videos_in_folder()
maximpavliv Feb 20, 2025
7faa988
Refactor _create_and_save_fake_metadata
maximpavliv Feb 20, 2025
e4b55b1
Refactor: _process_videos
maximpavliv Feb 20, 2025
f736668
Implement create_pretrained_project for torch
maximpavliv Feb 20, 2025
dd23192
Update create_pretrained_project usages
maximpavliv Feb 20, 2025
37aee7d
Refactoring
maximpavliv Feb 21, 2025
d13d213
Fix get_super_animal_project_cfg
maximpavliv Feb 20, 2025
9a825cb
Write integration script
maximpavliv Feb 20, 2025
0fcabd3
Fix other Superanimal tests
maximpavliv Feb 20, 2025
a67974b
Run black
maximpavliv Feb 24, 2025
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
Fix get_super_animal_project_cfg 
When calling read_config, the superanimal yaml config in the repo was modified
  • Loading branch information
@maximpavliv
maximpavliv committed Feb 25, 2025
commit d13d213a5719ddd7b61dc6feef4a26ed34419b9d
2 changes: 1 addition & 1 deletion deeplabcut/modelzoo/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def get_super_animal_project_cfg(super_animal: str) -> dict:
f"are {', '.join(super_animal_projects.keys())}."
)

return read_config(str(super_animal_projects[super_animal]))
return read_config_as_dict(super_animal_projects[super_animal])


def get_super_animal_scorer(
Expand Down