keras txt 读取训练数据_使用keras版本的yoloV3训练并在VOC数据集上测试
使用keras版本的yoloV3训练并在VOC数据集上测试
发布时间：2018-05-11 16:44,
浏览次数：394
, 标签：
keras
yoloV
VOC
yoloV3主页：https://pjreddie.com/darknet/yolo/
keras-yolo3代码：https://github.com/qqwweee/keras-yolo3
VOC数据集：http://host.robots.ox.ac.uk/pascal/VOC/
一、环境需求：
tensorflow
keras
(编译运行时按照错误提示pip install 相应库文件即可)
二、Demo：
1、下载代码：
git clone https://github.com/qqwweee/keras-yolo3
2、下载yoloV3权重文件：
https://pjreddie.com/media/files/yolov3.weights
3、将darknet下的yolov3配置文件转换成keras适用的h5文件：
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
4、运行预测图像程序(预测视频程序跳过)：
python yolo.py
输入需要预测的图片路径即可，示例如下，可以看到结果还是蛮准确的，但对于某些小目标(如图片左半部分)仍然未能很好地检测出来：
三、训练：
1、将VOC数据集放在代码同一目录下，然后先生成需要进行训练的txt文件，包括图片路径、类别和bbox，一张图片有多个目标时用'
'隔开，根据使用VOC数据集的年份修改voc_annotation.py文件中的sets，根据需要训练的类别修改classes(如注释中只使用person类)。
sets=[('2007', 'train'), ('2007', 'val'), ('2007', 'test')] # classes =
['person'] classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus",
"car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike",
"person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]
运行生成txt文件程序，根据sets得到三个不同的txt文件，方便用于做训练、验证和测试：
python voc_annotation.py
2、修改train.py中的annotation_path为刚刚生成的txt文件，data_path是根据txt提供的训练图像提取出的数据信息并保存为了npz格式(注意，如果需要更换训练数据的话，需要删除上一次生成的npz文件，或者在给get_training_data函数传参时设置load_previous=False)，output_path是当前训练的模型保存路径，log_dir记录了每一次迭代生成的模型，classes_path是需要训练的目标种类，根据上一步修改过的classes调整(如注释中选择的txt文件中只包含自己需要训练的类别)，anchors_path是锚点文件(因还不太了解所以没有改过)。
annotation_path = '2007_train.txt' data_path = 'train.npz'
output_path = 'model_data/my_yolo.h5' log_dir = 'logs/000/'
#classes_path = 'model_data/my_classes.txt' classes_path =
'model_data/voc_classes.txt' anchors_path = 'model_data/yolo_anchors.txt'
同时，在训练时还会存在这样的情况：可能数据集中某些图像中不包含你想要训练的目标，那么可以在生成txt文件时就筛除掉，或者在读取训练数据时跳过没有数据的那一行，这里作者选择后者，修改train.py中get_training_data函数：
with open(annotation_path) as f: for line in f.readlines():
line = line.split(' ') # skip the rows contain no target
if len(line) <= 1: continue filename = line[0]
image = Image.open(filename)
这里需要注意的是，如果想要训练某一类或某几类特定的目标类别，必须要同时修改以上两步中voc_annotation.py和train.py相应的位置，否则可能会报维数不符的错误。
修改完成后，开始训练：
python train.py
随着epoch迭代，会得到越来越小的loss，直到loss达到某一阈值自动停止，或者epoch达到上限也会停止。可以在train.py中的train函数中修改预设的epochs。
3、修改yolo.py，使用自己训练的模型预测图像：
#self.model_path = 'model_data/yolo.h5' self.model_path =
'model_data/my_yolo.h5' self.anchors_path = 'model_data/yolo_anchors.txt'
#self.classes_path = 'model_data/coco_classes.txt' self.classes_path =
'model_data/my_classes.txt'
因为只在2007VOC训练集上进行30次的迭代训练，所以得到的模型效果不算好，在VOC测试图像上的预测效果还可以，但不够精确。如果测试图像更大，差的预测结果就比较明显了，因此需要更多训练数据来训练自己的模型。
四、在VOC2007上评估MAP
1、具体过程参考：YOLO在VOC2007数据集计算mAP 备忘
2、输出文件格式要求：Pascal VOC 数据集介绍
五、其他：
1、如果想要使用自己的权重继续训练，在train.py中的create_model函数中修改weights_path。如果不想使用预训练的权重，设置传给create_model函数的参数load_pretrained=False。