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:測試