【YOLO初探】yolo_tensorflow_v1训练自己的数据集

前言

上接：https://blog.csdn.net/plSong_CSDN/article/details/86348413（yolo_tensorflow_v1使用YOLO_small.ckpt进行测试）

所需材料

python3.6

tensorflow-gpu  1.4.0,

cuda_8.0.61_win10

cudnn-8.0-windows10-x64-v6.0

TensorFlow 版本的源码 :https://github.com/hizhangp/yolo_tensorflow

数据集：使用的是文章https://blog.csdn.net/plSong_CSDN/article/details/85194719（keras-yolov3训练自己数据集）中的自己收集的五官数据集。

训练

step1：将准备好的数据放在yolo_tensorflow-master\data\pascal_voc\VOCdevkit\VOC2007目录下

step2：修改config.py文件，

第一处：CLASSES = ['class1', 'class2', 'class2'....] 

看这段代码：

f is_training:
            self.labe