TMElyralab / Comfyui-MusePose

Other
287 stars 28 forks source link

KeyError: 'TopdownPoseEstimator is not in the mmdet::model registry. #23

Open jaechanjo opened 4 weeks ago

jaechanjo commented 4 weeks ago

In Colab, cuda version is 11.8/ python version 3.10.12 torch 2.1.2 mmcv 2.1.0 mmengine 0.10.4 mmdet 3.3.0 mmpose 1.3.1

Error raises in "muse_pose_align" node, """ !!! Exception during processing !!! Traceback (most recent call last): File "/content/drive/MyDrive/ComfyUI/execution.py", line 151, in recursive_execute output_data, output_ui = get_output_data(obj, input_data_all) File "/content/drive/MyDrive/ComfyUI/execution.py", line 81, in get_output_data return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True) File "/content/drive/MyDrive/ComfyUI/execution.py", line 74, in map_node_over_list results.append(getattr(obj, func)(*slice_dict(input_data_all, i))) File "/content/drive/MyDrive/ComfyUI/custom_nodes/Comfyui-MusePose/nodes.py", line 554, in posealign_func return run_align_video_with_filterPose_translate_smooth(args) File "/content/drive/MyDrive/ComfyUI/custom_nodes/Comfyui-MusePose/nodes.py", line 276, in run_align_video_with_filterPose_translate_smooth detector = DWposeDetector( File "/content/drive/MyDrive/ComfyUI/custom_nodes/Comfyui-MusePose_old/pose/script/dwpose.py", line 75, in init self.pose_estimation = Wholebody(pose_config, pose_ckpt, device) File "/content/drive/MyDrive/ComfyUI/custom_nodes/Comfyui-MusePose_old/pose/script/wholebody.py", line 33, in init self.detector = init_detector(det_config, det_ckpt, device=device) File "/content/drive/MyDrive/ComfyUI/custom_nodes/Comfyui-MusePose/mmdetection/mmdet/apis/inference.py", line 66, in init_detector model = MODELS.build(config.model) File "/usr/local/lib/python3.10/dist-packages/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, **kwargs, registry=self) File "/usr/local/lib/python3.10/dist-packages/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "/usr/local/lib/python3.10/dist-packages/mmengine/registry/build_functions.py", line 100, in build_from_cfg raise KeyError( KeyError: 'TopdownPoseEstimator is not in the mmdet::model registry. Please check whether the value of TopdownPoseEstimator is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module'

"""