Visitar URL original
`superanimal_humanbody` with FilteredDetector: add video adaptation by maximpavliv · Pull Request #3086 · DeepLabCut/DeepLabCut · GitHub
Skip to content
Merged
Changes from 1 commit
Commits
Show all changes
66 commits
Select commit Hold shift + click to select a range
ff54deb
Remove unused imports
maximpavliv Jul 8, 2025
0963bee
_build_tf_attributes - add pseudo threshold widgets back
maximpavliv Jul 8, 2025
2bb52de
Restore hiding torch_widget when TF Engine is selected
maximpavliv Jul 8, 2025
f0d4f5a
Hide adaptation settings when adaptation is unchecked
maximpavliv Jul 8, 2025
c5306b9
Hide/Show detector row on engine change
maximpavliv Jul 8, 2025
930c71f
Hide detector row when humanbody supermodel is selected
maximpavliv Jul 8, 2025
2438c13
Remove unused method
maximpavliv Jul 8, 2025
0765c8e
Hide torch adaptation widgets when humanbody is selected
maximpavliv Jul 8, 2025
f93530c
video_inference_superanimal: add create_labeled_video argument
maximpavliv Jun 30, 2025
64d663b
Rename method
maximpavliv Jul 8, 2025
6d7e7ef
Add create labeled video checkbox
maximpavliv Jul 8, 2025
207f644
Use set_combo_items() to update combo boxes
maximpavliv Jul 8, 2025
7bf43a0
Add batch size combo boxes for pose model and detector
maximpavliv Jul 8, 2025
3fd8ead
Replace MediaSelectorWidget with VideoSelectorWidget
maximpavliv Jul 9, 2025
a19e980
VideoSelectorWidget: add hide_videotype option
maximpavliv Jul 9, 2025
d50e5b4
Refactor: split _build_common_attributes() into smaller functions
maximpavliv Jul 10, 2025
88d4743
Refactor: split _build_tf_attributes and _build_torch_attributes into…
maximpavliv Jul 10, 2025
1435920
Refactoring: single _add_use_adaptation_row method for tf and torch
maximpavliv Jul 10, 2025
ba3012a
Merge branch 'maxim/restore_video_selection_widget' into maxim/modelz…
maximpavliv Jul 10, 2025
2abf37c
Trim superanimal_humanbody.yaml default project config
maximpavliv Jul 21, 2025
a1a6be1
Trim superanimal_humanbody_colors
maximpavliv Jul 21, 2025
dfbce1d
Correct get_checkpoint_epoch
maximpavliv Jul 21, 2025
1432a73
Add rtmpose_x modelzoo model config
maximpavliv Jul 21, 2025
a4d74cc
Add FilteredDetector
maximpavliv Jul 21, 2025
0cbfe59
Add get_filtered_coco_detector_inference_runner() method
maximpavliv Jul 21, 2025
84b230e
Add ScaleToUnitRange transform
maximpavliv Jul 21, 2025
c4c1318
Superanimal humanbody inference: use filtered detector runner
maximpavliv Jul 21, 2025
dc511cd
ModelZoo tab: make humanbody general case
maximpavliv Jul 21, 2025
479cc66
get_super_animal_scorer(): add torchvision_detector_name arg
maximpavliv Jul 21, 2025
6a14584
Remove superanimal_humanbody_video_inference.py module
maximpavliv Jul 21, 2025
6774812
Regularize get_super_animal_model_config_path()
maximpavliv Jul 21, 2025
424ac92
Regularize load_super_animal_config()
maximpavliv Jul 21, 2025
04438f8
Regularize download_super_animal_snapshot()
maximpavliv Jul 21, 2025
256977b
update_config(): superanimal_humanbody - compatible
maximpavliv Jul 21, 2025
8d4cf20
Revert video_inference()
maximpavliv Jul 21, 2025
edf4f9d
Revert create_df_from_prediction()
maximpavliv Jul 21, 2025
cded581
Restore CTDInferenceRunner
maximpavliv Jul 21, 2025
7eaf923
Remove TorchvisionDetectorInferenceRunner
maximpavliv Jul 21, 2025
feb34d1
Revert DetectorInferenceRunner
maximpavliv Jul 21, 2025
711a47e
superanimal_analyze_images() - make humanbody compatible
maximpavliv Jul 21, 2025
28e4dd0
Revert build_predictions_dataframe()
maximpavliv Jul 21, 2025
193f935
Revert get_inference_runners()
maximpavliv Jul 21, 2025
613b2ac
Revert detectors/fasterRCNN.py
maximpavliv Jul 21, 2025
d848741
Revert detectors/torchvision.py
maximpavliv Jul 21, 2025
bba8689
Revert base Runner
maximpavliv Jul 21, 2025
fa45d26
Merge branch 'maxim/superanimal_humanbody_filtered_detector' into max…
maximpavliv Jul 25, 2025
d8f5bdb
No video adaptation if superanimal_humanbody
maximpavliv Jul 25, 2025
1a03360
Remove deprecated skipped frames msg
maximpavliv Jul 26, 2025
b1d982e
Fix Run button style change
maximpavliv Jul 26, 2025
f77ccda
Universalize created videos msg
maximpavliv Jul 26, 2025
c6ecb0f
Superanimal_humanbody - add detector widget
maximpavliv Jul 26, 2025
25fa08d
Fix superanimal_humanbody unit test
maximpavliv Jul 31, 2025
425484b
Disable video adaptation for superanimal_humanbody
maximpavliv Jul 31, 2025
4b04013
Fix testscript_superanimal_inference.py
maximpavliv Jul 31, 2025
eea470e
Remove debug print
maximpavliv Jul 31, 2025
3d47a36
Black formatting
maximpavliv Jul 31, 2025
fb90004
Merge branch 'maxim/superanimal_humanbody_filtered_detector' into max…
maximpavliv Jul 31, 2025
749de6d
Black formatting
maximpavliv Jul 31, 2025
63d7208
Merge branch 'main' into maxim/modelzoo_tab_fixes
maximpavliv Sep 5, 2025
19a2696
Add superanimal_humanbody video adaptation
maximpavliv Sep 3, 2025
ce0d201
Add superanimal_humanbody video adaptation to GUI
maximpavliv Sep 3, 2025
c39db66
Add warning
maximpavliv Sep 4, 2025
bccf4f3
Raise exception in build_weight_init()
maximpavliv Sep 4, 2025
e9ff1a1
Black formatting
maximpavliv Sep 4, 2025
efca1c7
Read adapt_checkbox for superanimal_humanbody
maximpavliv Sep 10, 2025
c5eeb2e
Merge branch 'main' into maxim/superanimal_humanbody_video_adaptation
maximpavliv Sep 10, 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
Hide detector row when humanbody supermodel is selected
  • Loading branch information
maximpavliv committed Jul 8, 2025
commit 930c71f739cfe015486f3782957e227a1e992fce
18 changes: 5 additions & 13 deletions deeplabcut/gui/tabs/modelzoo.py
Original file line number Diff line number Diff line change
Expand Up @@ -663,18 +663,10 @@ def _update_detectors(self, super_animal: str) -> None:
if self.root.engine == Engine.TF:
self.detector_type_selector.addItems(["dlcrnet"])
else:
if super_animal == "superanimal_humanbody":
self.detector_type_selector.clear()
self.detector_type_selector.addItems([
"fasterrcnn_mobilenet_v3_large_fpn",
"fasterrcnn_resnet50_fpn_v2"
])
else:
try:
detectors = dlclibrary.get_available_detectors(super_animal)
self.detector_type_selector.addItems(detectors)
except KeyError:
pass
detectors = dlclibrary.get_available_detectors(super_animal)
self.detector_type_selector.addItems(detectors)
set_layout_contents_visible(self.detector_row, super_animal != "superanimal_humanbody")


@Slot(Engine)
def _on_engine_change(self, engine: Engine) -> None:
Expand All @@ -687,7 +679,7 @@ def _on_engine_change(self, engine: Engine) -> None:
self.tf_widget.show()

# Hide widgets in detector row
set_layout_contents_visible(self.detector_row, engine == Engine.PYTORCH)
set_layout_contents_visible(self.detector_row, engine == Engine.PYTORCH and self.model_combo.currentText() != "superanimal_humanbody")

# Initialize adaptation options based on current media type
current_media_type = self.media_selection_widget.media_type_widget.currentText()
Expand Down