【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，否则为0
OPENCV=0  # 如果调用摄像头，还需要设置OPENCV为1，否则为0
OPENMP=0  # 如果使用OPENMP设置为1，否则为0
DEBUG=0 # 如果使用DEBUG设置为1，否则为0

保存并推出，然后在命令行里make

make

3. 准备数据集

3.1. 标注数据

使用labelImg进行标注
使用git下载labelImg：

git clone https://github