用自己資料訓練faster-rcnn---cpu
模型下載:
修改模型中的部分程式碼:
因為是用CPU跑模型,所以要對程式碼進行適當修改。
用自己的資料集訓練faster-rcnn:
1.擁有訓練的資料以及資料對應的xml檔案。
我這裡用的是以前人臉座標資料,用的標註工具是LableImg,當然還有其他的標註資料的方法。這裡貼一個從0開始做訓練資料的部落格:https://www.cnblogs.com/qw12/p/6185126.html。注意:訓練的資料是jpg 格式的
2.做好的資料放在主檔案指定位置:
訓練圖片:tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/JPEGImages
訓練圖片對應的xml檔案:tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/Annotations
訓練、測試、驗證的圖片名檔案:tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/ImageSets/Main/val.txt
tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/ImageSets/Main/train.txt
tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/ImageSets/Main/test.txt
tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/ImageSets/Main/trainval.txt
results 資料夾(必須有,不然會報錯):tf-faster-rcnn-master/data/VOCdevkit2007/VOC2007/results/VOC2007/Main/
訓練網路:
結合這幾篇部落格最終成功執行,因為是CPU,30幾秒一個iter 。