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即可開始訓練。