【YOLO】使用YOLOv3训练及测试自己的数据集
使用YOLOv3训练及测试自己的数据集
1. 环境
Ubuntu 16.04
Python 3.7.0
torch 1.6.0
2. git YOLOv3项目
2.1. git
git clone https://github.com/pjreddie/darknet
cd darknet
make
2.2. 修改Makefile
修改darknet文件夹中的Makefile
修改为:
GPU=1 # 如果使用GPU设置为1,CPU设置为0
CUDNN=0 # 如果使用CUDNN设置为1,
原创
2020-08-21 17:25:30 ·
1939 阅读 ·
0 评论