数据集的准备

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

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

图13

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

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

图14

模型训练

(1)训练所需要的环境如下:

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

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

图15

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

图16

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

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

图17


今天不学习,明天变垃圾!!!

04-08 00:04