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

你的名字 2022-03-27 10:38 414阅读 0赞

# 前言 #

上接：[https://blog.csdn.net/plSong\_CSDN/article/details/86348413][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][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.labels = tf.placeholder(
                    tf.float32,
                    [None, self.cell_size, self.cell_size, 5 + self.num_class])

labels中的“5 + self.num\_class”这个要根据自己实际情况就行修改。例如，如果你的训练有三类，name就应该是“5+3 = 8”

所以，要修改的

第二处：pascal\_voc.py文件中

labels = np.zeros(
                (self.batch_size, self.cell_size, self.cell_size, 8)) #25修改为8

第三处：pascal\_voc.py文件中

label = np.zeros((self.cell_size, self.cell_size, 8)) #25修改为8

下面就可以执行训练了。完成后，会提示训练完成。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BsU29uZ19DU0RO_size_16_color_FFFFFF_t_70][]

# 测试 #

训练完成后的模型在文件目录：yolo\_tensorflow-master\\data\\pascal\_voc\\output\\2019\_01\_12\_15\_27

其中：2019\_01\_12\_15\_27是对应训练的开始的时间。

里面保存着训练过程中，不同阶段的模型文件，三个为一组。将最后训练完成的一组拷贝到：yolo\_tensorflow-master\\data\\weights下。修改文件名为\*.ckpt.\*，如下：

![2019011310361786.png][]

在yolo\_tensorflow-master\\test\\文件夹下，放入相应的测试图片，修改test.py文件中对应的文件名

imname = 'test/person.jpg'

直接执行即可。

# 后记 #

整个流程题主走了一遍，中间也为某些错误纠结了很长时间。相对于[https://blog.csdn.net/plSong\_CSDN/article/details/85194719][https_blog.csdn.net_plSong_CSDN_article_details_85194719]这篇文章中的而言，训练效果非常差，在这我就不贴出来了。

欢迎大家留言讨论。

--------------------

★finished by songpl，2019.1.13

[https_blog.csdn.net_plSong_CSDN_article_details_86348413]: https://blog.csdn.net/plSong_CSDN/article/details/86348413
[https_blog.csdn.net_plSong_CSDN_article_details_85194719]: https://blog.csdn.net/plSong_CSDN/article/details/85194719
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3BsU29uZ19DU0RO_size_16_color_FFFFFF_t_70]: /images/20220327/222ce9c3824c413491423b4dd746f523.png
[2019011310361786.png]: /images/20220327/e6b227f2f798474192da8c81627b1663.png