YOLOv5旋转目标框项目使用记录

一、项目来源

GitHub - BossZard/rotation-yolov5: rotation detection based on yolov5

二、环境

Windows 10

RTX2070

cuda_10.2.89_win10_network

cudnn-windows-x86_64-8.4.1.50_cuda10.2-archive

Python3.7

opencv_contrib_python-4.2.0.34-cp37-cp37m-win_amd64

torch-1.6.0-cp37-cp37m-win_amd64

torchvision-0.7.0-cp37-cp37m-win_amd64

shapely==1.7.1

三、遇到的问题

1、OSError: [WinError 126] 找不到指定的模块

（1）点击"J:\Users\wpk\anaconda3\envs\cuda102\lib\site-packages\shapely\geos.py"这个路径

（2）在这句代码打上断点

_lgeos = CDLL(os.path.join(sys.prefix, 'Library', 'bin', 'geos_c.dll'))

（3）用调试的方式执行，程序停在断点处，然后点击Console

 （4）在Console命令行输入下面代码并回车，得到geos_c.dll的路径，记geos_c.dll所在文件夹的路径为P（即我电脑的J:\\Users\\wpk\\anaconda3\\envs\\cuda102\\Library\\bin），后面需要用到。

os.path.join(sys.prefix, 'Library', 'bin', 'geos_c.dll')

（5）打开geos_c.dll文件所在的文件夹，查看是否有geos_c.dll文件。因为报了这个错误，所以一般情况下是没有geos_c.dll文件的。

然后在终端输入下面命令，查看shapely的安装路径

pip show shapely

 （6）打开该路径，并进入shapely\DLLs文件夹，可以看到geos.dll和geos_c.dll

（7）将geos.dll和geos_c.dll复制，粘贴到上述路径P即可解决。

 2、TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

将numpy1.21.6降为numpy1.19就好了，亲测有用。直接输入下面代码

pip install numpy==1.19