Retina-net 實踐 Keras版本

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精度評定指標

 

發佈了65 篇原創文章 · 獲贊 29 · 訪問量 19萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章