訓練 smallcorgi/Faster-RCNN_TF 模型(附ImageNet model百度雲下載地址)


  1. 下載訓練、驗證、測試數據和 VOCdevkit,下載地址:
   http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
   http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
   http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

   2. 解壓已經下載的數據至同一個文件夾下,文件夾名字修改為 VOCdevkit
     tar xvf VOCtrainval_06-Nov-2007.tar
     tar xvf VOCtest_06-Nov-2007.tar
     tar xvf VOCdevkit_08-Jun-2007.tar

   解壓后的路徑層次如下:
     $VOCdevkit/             # development kit
     $VOCdevkit/VOCcode /       # VOC utility code
     $VOCdevkit/VOC2007              # image sets, annotations, etc.
   # ... 等其他子目錄 ...

    3. 創建PASCAL VOC 數據庫鏈接,可以將解壓好的數據移至 $FRCN_ROOT/data 目錄下
     cd $FRCN_ROOT/data
     ln -s $VOCdevkit VOCdevkit2007

    4. 下載預訓練的 ImageNet 模型,下載地址為:
   鏈接:https://pan.baidu.com/s/1-Pbi1MLe2HgZvzVhKA7l_A  密碼:veuk


    5. 將下載的ImageNet模型名稱改為VGG_imagenet.npy,並移至 $FRCN_ROOT/data/pretrain_model/ 目錄下
     mv VGG_imagenet.npy $FRCN_ROOT/data/pretrain_model/VGG_imagenet.npy

    6. 運行訓練和測試的sh文件
      cd $FRCN_ROOT
      ./experiments/scripts/faster_rcnn_end2end.sh $DEVICE $DEVICE_ID VGG16 pascal_voc

      其中的 $DEVICE 和 $DEVICE_ID 分別可以填 gpu 和 0,如下:
      ./experiments/scripts/faster_rcnn_end2end.sh gpu 0 VGG16 pascal_voc
 

訓練結束,執行到test的時候會出現一直等待的問題

Waiting for /home/username/Faster-RCNN_TF/output/faster_rcnn_end2end/voc_2007_trainval/VGGnet_fast_rcnn_iter_70000.ckpt to exist...

可以修改tools/test_net.py 中的while判斷條件語句

while not os.path.exists(args.model) and args.wait:

改為

while not os.path.exists(args.model+'.meta') and args.wait:


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM