Improve `deeplabcut.pose_estimation_pytorch` imports by n-poulsen · Pull Request #2833 · DeepLabCut/DeepLabCut
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_configfor COCO projects, to make the creation of model configuration files easier (seedeeplabcut/pose_estimation_pytorch/README.mdfor an example) - use
get_pose_inference_runnerandget_detector_inference_runnerinanalyze_image_folder(instead ofget_inference_runners) to decouple detector and pose model loading. video_inferenceno longer needs to be given atask, as the task was not useful (if a detector runner is given, object detection will be run before pose estimation)load_head_weightswas not correctly given totrainintrain_networkLoader.ground_truth_bboxesnow also returns image sizes, which makes evaluation of detectors easier