logs/000/trained_weights_final.h5 放置訓練完的權重
keras-yolo3-master
Keras/Tensorflow+python+yolo3訓練自己的數據集
代碼:https://github.com/qqwweee/keras-yolo3
修改yolov3.cfg文件:https://blog.csdn.net/lilai619/article/details/79695109
使用yolo3訓練自己的數據集進行目標檢測
VOCdevkit/VOC2007/Annotations xml文件
VOCdevkit/VOC2007/JPEGImages jpg圖片
VOCdevkit/VOC2007/ImageSets/Main 下面四個文件,在VOC2007下創建文件test.py,
運行voc_annotation.py,修改classes,運行結果是2007_train.txt,手動刪除2007_,生成train.txt,
修改參數文件:yolo3.cfg
三處yolo 各修改filters:3*(5+len(classes));
classes:len(classes)
random:原來是1,顯存小改為0,
修改model_data下文件,coco_classes.txt:放入類名 voc_classes.txt:放入類名,注意不要有空行或空格
訓練:
python convert.py -w yolov3.cfg yolov3.weights model_data/yolo_weights.h5
python train.py
測試:
python yolo.py 或者python yolo_video.py video_path
