windows下yolov5_obb和nms_rotated的问题

Question

在运行yolov5-obb时出现了cannot import name 'nms_rotated_ext' from partially initialized module 'utils.nms_rotated' (most likely due to a circular import)的问题

环境是windows + python3.9

已尝试更改了pytorch版本，目前已尝试了
torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0
torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1
torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0
torch-2.0.1+cu117-cp39-cp39-win_amd64
torch-2.0.1+cu118-cp39-cp39-win_amd64

Answer 1

参考gpt：
结合自己分析给你如下建议：
您没有正确地安装yolov5-obb所需的依赖包，比如numpy, opencv-python, cython等。您可以尝试使用pip install -r requirements.txt命令来安装所有的依赖包。
您没有正确地编译yolov5-obb中的C++扩展模块，比如nms_rotated_ext。您可以尝试使用python setup.py build_ext --inplace命令来编译所有的扩展模块。
您的pytorch版本和cuda版本不匹配，导致无法加载nms_rotated_ext模块。您可以尝试使用pip uninstall torch torchvision torchaudio命令来卸载当前的pytorch版本，然后根据您的cuda版本选择合适的pytorch版本进行安装。