FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Improve `deeplabcut.pose_estimation_pytorch` imports by n-poulsen · Pull Request #2833 · DeepLabCut/DeepLabCut · GitHub

Improve deeplabcut.pose_estimation_pytorch imports - #2833

Merged
n-poulsen merged 20 commits into
pytorch_dlcfrom
niels/dlc3_diagram
Jan 10, 2025
Merged

Improve deeplabcut.pose_estimation_pytorch imports#2833
n-poulsen merged 20 commits into
pytorch_dlcfrom
niels/dlc3_diagram

Conversation

Copy link
Copy Markdown
Contributor

Improves the way the most useful classes/methods from deeplabcut.pose_estimation_pytorch can be imported.

Additionally, makes the following improvements/fixes:

  • a new dlc_torch.config.make_basic_project_config for COCO projects, to make the creation of model configuration files easier (see deeplabcut/pose_estimation_pytorch/README.md for an example)
  • use get_pose_inference_runner and get_detector_inference_runner in analyze_image_folder (instead of get_inference_runners) to decouple detector and pose model loading.
  • video_inference no longer needs to be given a task, as the task was not useful (if a detector runner is given, object detection will be run before pose estimation)
  • load_head_weights was not correctly given to train in train_network
  • Loader.ground_truth_bboxes now also returns image sizes, which makes evaluation of detectors easier

maximpavliv left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

LGTM, thanks @n-poulsen !
Btw, I added a couple of comments on the diagram on Figma

n-poulsen merged commit d03da3c into pytorch_dlc Jan 10, 2025
n-poulsen deleted the niels/dlc3_diagram branch January 10, 2025 16:26
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants


Back | FazBrowse Home | New Git URL