使用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