第五章 YOLOv3训练自己的数据集

作者：Gausst松鼠会 | 2024-04-09 01:53:23

踩

第五章 YOLOv3训练自己的数据集

在官网中下载VOC2012数据集（或在本文附件5中下载）

图13

使用voc_annotation.py生成2012_train.txt和2012_val.txt标注文件，如图14所示：

图14

（1）训练所需要的环境如下：

Python==3.5.6

Keras==2.1.5

Tensorflow==1.10.0

（2）下载附件5 yolov3-keras，解压后，使用vs code打开项目文件夹（图15），并使用上述环境运行train.py进行模型训练。

图15

图16

（3）训练结束后，项目根目录下会生成logs/000文件夹，文件夹中存放的是网络结构文件.h5和日志文件。前者用于.pb文件的生成，而后者可用来分析训练状态与可视化网络，如图17所示。

图17

今天不学习，明天变垃圾！！！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/389530