pytorch框架下fasterRCNN海上目标自定义数据集训练测试

代码链接：https://github.com/bubbliiiing/faster-rcnn-pytorch

本地环境如软件版本：

win10

anaconda5

pytorch1.1

1 依赖库安装：

No module named 'cupy'  解决办法：因为本地是cuda9.0环境，所以pip install cupy-cuda90，cuda10则为cupy-cuda100；

2.数据集，是自己勾画的海上目标

4 目录结构，关于数据集如何使用可以参考keras版本yolov3示例，

数据集有三个类别，所以classnum为3，不要忘记修改label的id，训练和测试过程如下；