数据集的准备
在官网中下载VOC2012数据集(或在本文附件5中下载)
图13
使用voc_annotation.py生成2012_train.txt和2012_val.txt标注文件,如图14所示:
图14
模型训练
(1)训练所需要的环境如下:
(2)下载附件5 yolov3-keras,解压后,使用vs code打开项目文件夹(图15),并使用上述环境运行train.py进行模型训练。
图15
图16
(3)训练结束后,项目根目录下会生成logs/000文件夹,文件夹中存放的是网络结构文件.h5和日志文件。前者用于.pb文件的生成,而后者可用来分析训练状态与可视化网络,如图17所示。
图17
今天不学习,明天变垃圾!!!