1. 程式人生 > >用自己資料訓練faster-rcnn---cpu

用自己資料訓練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 。