使用mmpose库时出现了问题

Question

我打算用mmpose进行人体姿态识别，一开始mmcv版本不对，后来所有东西都配置好之后出现下面几个问题：
遇到的错误表明在加载 mmcv 的扩展模块时失败了，这可能是因为以下几个原因：
1 缺少依赖：可能缺少一些系统库或依赖库。
2DLL 文件问题：可能是由于 DLL 文件缺失或不匹配。
3 Python 版本和环境问题：环境中可能存在 Python 版本或库不兼容的问题。
这是报错代码

(dwpose) PS F:\DWPOSE> python F:\DWPOSE\DWPose-onnx\mmpose\demo\image_demo.py video\img1.jpg F:\DWPOSE\DWPose-onnx\mmpose\ceshi\configs\td-hm_hrnet-w48_8xb32-210e_coco-256x192.py F:\DWPOSE\DWPose-onnx\mmpose\ceshi\configs\td-hm_hrnet-w48_8xb32-210e_coco-256x192-0e67c616_20220913.pth --out-file vis_results.jpg --draw-heatmap
C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmengine\optim\optimizer\zero_optimizer.py:11: DeprecationWarning: TorchScript support for functional optimizers is deprecated and will be removed in a future PyTorch release. Consider using the torch.compile optimizer instead.
  from torch.distributed.optim import \
C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmcv\cnn\bricks\transformer.py:33: UserWarning: Fail to import `MultiScaleDeformableAttention from mmcv.ops.multi_scale_deform_attn, You should install mmcv rather than mmcv-lite if you need this module.
  warnings.warn('Fail to import `MultiScaleDeformableAttention from '
Traceback (most recent call last):
  File "F:\DWPOSE\DWPose-onnx\mmpose\demo\image_demo.py", line 6, in <module>
    from mmpose.apis import inference_topdown, init_model
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmpose\apis\__init__.py", line 2, in <module>
    from .inference import (collect_multi_frames, inference_bottomup,
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmpose\apis\inference.py", line 17, in <module>
    from mmpose.models.builder import build_pose_estimator
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmpose\models\__init__.py", line 8, in <module>
    from .heads import *  # noqa
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmpose\models\heads\__init__.py", line 11, in <module>
    from .transformer_heads import EDPoseHead
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmpose\models\heads\transformer_heads\__init__.py", line 2, in <module>
    from .edpose_head import EDPoseHead
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmpose\models\heads\transformer_heads\edpose_head.py", line 14, in <module>
    from mmcv.ops import MultiScaleDeformableAttention
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmcv\ops\__init__.py", line 3, in <module>
    from .active_rotated_filter import active_rotated_filter
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmcv\ops\active_rotated_filter.py", line 10, in <module>
    ext_module = ext_loader.load_ext(
  File "C:\Users\ssx\AppData\Roaming\Python\Python38\site-packages\mmcv\utils\ext_loader.py", line 13, in load_ext
    ext = importlib.import_module('mmcv.' + name)
  File "F:\Anaconda\envs\dwpose\lib\importlib\__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
ImportError: DLL load failed while importing _ext: 找不到指定的程序。

Answer 1

🕒下午好，题主！目前：🟢[在线]
📢参考通义千问、由郭老师整理提供！
📣今日诗号：日出江花红胜火，春来江水绿如蓝。

你遇到的问题是因为 mmcv-lite 不包含 MultiScaleDeformableAttention 模块，而你的代码中需要使用这个模块。

解决方案：

1. 卸载 mmcv-lite：

   pip uninstall mmcv-lite
   

2. 安装完整的 mmcv： 在安装 mmcv 之前，请先确保你的 CUDA 和 C++ 环境已经配置好。然后运行以下命令安装 mmcv：

   pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
   

   将 {cu_version} 替换为你的 CUDA 版本（例如 11.3），将 {torch_version} 替换为你的 PyTorch 版本（例如 1.10.0）。

   你也可以根据自己的系统和 CUDA 版本在 https://download.openmmlab.com/mmcv/dist/index.html 上找到对应的 mmcv 版本。

3. 验证安装：

   安装完成后，运行以下代码验证是否成功导入了 MultiScaleDeformableAttention 模块：

   from mmcv.ops import MultiScaleDeformableAttention
   
   print(MultiScaleDeformableAttention)
   

   如果没有报错，说明安装成功。

其他建议：

• 确保你的所有依赖库版本都与 mmpose 兼容。你可以在 mmpose 的官方文档中找到兼容性信息。
• 检查你的 Python 环境是否配置正确，并尝试在一个新的环境中安装 mmpose 和 mmcv。

希望这些解决方案能够帮助你解决问题。如果问题仍然存在，请提供更多信息，例如你的操作系统、CUDA 版本、PyTorch 版本、mmcv 版本和 mmpose 版本，以便我更好地帮助你。