1、安装CentOS 7安装OpenCV和YOLO
2、下载数据集VOC数据集下载地址
3、生成labels
由于硬件配置限制,本人只下载了2007的train数据做训练。
下载数据压缩包VOCtrainval_06-Nov-2007.tar
解压:tar -xvf VOCtrainval_06-Nov-2007.tar
将解压的目录VOCdevkit复制到darknet/script.py目录下(这里是为了不修改voc_label.py文件),
因为没有用所有的数据集
需将第7行:sets=[(‘2012’, ‘train’), (‘2012’, ‘val’), (‘2007’, ‘train’), (‘2007’, ‘val’), (‘2007’, ‘test’)]
改为:sets=[(‘2007’, ‘train’), (‘2007’, ‘val’)]
执行 python voc_lable.py
会生成几个文件列表,2007_train.txt,2007_val.txt
4、修改配置文件
修改darknet/cfg/voc.data
classes= 20
train = /root/software/darknet/scripts/2007_train.txt
valid = /root/software/darknet/