三、訓練自己的yolo3模型

yolo3整體的文件夾構架如下:
在這裏插入圖片描述
本文使用VOC格式進行訓練。
訓練前將 標籤文件(.xml) 放在 VOCdevkit 文件夾下的 VOC2007 文件夾下的 Annotation 中。

在這裏插入圖片描述
訓練前將 圖片文件 放在 VOCdevkit 文件夾下的 VOC2007 文件夾下的 JPEGImages 中。
在這裏插入圖片描述

一個圖片文件 對應 一個標籤文件:

在訓練前利用 voc2yolo3.py 文件生成對應的txt。
在這裏插入圖片描述
再運行根目錄下的voc_annotation.py,運行前需要將classes改成你自己的classes。

classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]

在這裏插入圖片描述


就會生成對應的 2007_train.txt,每一行對應其 圖片位置 及其 真實框的位置 和 分類


在這裏插入圖片描述

在訓練前需要修改model_data裏面的voc_classes.txt文件,需要將classes改成你自己的classes。


在這裏插入圖片描述

運行train.py即可開始訓練。


在這裏插入圖片描述

 

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章