赞
踩
电脑配置:win10 CUDA 11.6
https://github.com/hukaixuan19970627/yolov5_obb
conda create -n obb python=3.9
激活虚拟环境
conda activate obb
pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
下载地址:https://github.com/hukaixuan19970627/yolov5_obb
cd yolov5_obb-master
pip install -r requirements.txt
cd utils/nms_rotated
python setup.py develop
这里报错Error compiling objects for extension,需要替换utils\nms_rotated\src下的poly_nms_cuda.cu。
替换后再运行:
python setup.py develop
参考issue:
https://github.com/hukaixuan19970627/yolov5_obb/issues/224
pip install numpy==1.22
pip install pillow==9.5
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。