huggingface / controlnet_aux

Apache License 2.0
400 stars 86 forks source link

dwpose/yolox_config not included in the python package DW Pose failing #93

Open stevenjlm opened 9 months ago

stevenjlm commented 9 months ago

After running pip install controlnet-aux==0.0.7 and following the instructions from DW Pose:

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.1"
mim install "mmdet>=3.1.0"
mim install "mmpose>=1.1.0"

I am getting an error FileNotFoundError: [Errno 2] No such file or directory: '[...]/lib/python3.11/site-packages/controlnet_aux/dwpose/yolox_config/yolox_l_8xb8-300e_coco.py' Indeed in the lib/python3.11/site-packages/controlnet_aux/dwpose directory I do not see the yolox_config folder at all. And in the pypi package source it seems to be missing as well. Is there an installation step missing, or is the package missing files?

stevenjlm commented 9 months ago

If anyone else has this issue, the workaround is to download the script yolox_l_8xb8-300e_coco.py and pass its path as an argument:

det_config = "./dwpose_utils/yolox_l_8xb8-300e_coco.py"
pose_config = "./dwpose_utils/dwpose-l_384x288.py"
self.dwpose_detector = DWposeDetector(
    det_config=det_config, pose_config=pose_config, device="cuda"
)

After that, it's all working for me.