Tensorflow版本需要1.14.0及以上。之前用的是1.13.1,後面重裝了1.14.0
1. github:https://github.com/fizyr/keras-retinanet
git clone https://github.com/fizyr/keras-retinanet.git
2. follow README
--> 編譯環境 ①pip install numpy --user ② pip install . --user(這裏需要稍微等一下) ③python setup.py build_ext --inplace(這裏也要等一下)
3. 準備自己的數據,我這裏用的是pascal VOC數據,數據組織如下:
VOC2007
--JPEGImages(訓練數據,圖片.jpg)
--Annotations(標籤數據,.xml)
--ImageSets
--Main
--test.txt(測試的圖片名,沒有後綴)
--trainval.txt(訓練的圖片名,沒有後綴)
4. 修改參數並訓練、驗證
①需要修改pascal_voc.py 裏面的voc_classes類別,這裏不包括背景。
②根據自己的數據修改epoch,steps per epoch,(steps≈訓練樣本數,epoch=總迭代次數/steps)
③根據檢測目標大小修改anchor:在anchors.py下。
注:一個epoch之後會做一次evaluation,給了AP和F1精度評定指標