tf-faster-rcnn[cpu]實現目標檢測(二)

繼CPU下運行demo之後,現在開始使用res101進行voc數據的訓練
(1)在tf-faster-rcnn根目錄下,執行:

NET=res101
TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
mkdir -p output/${NET}/${TRAIN_IMDB}
cd output/${NET}/${TRAIN_IMDB}
ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default
cd ../../..

(2)下載res101模型,並建立軟連接

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/resnet_v1_101_2016_08_28.tar.gz
tar -xzvf resnet_v1_101_2016_08_28.tar.gz
mv resnet_v1_101.ckpt res101.ckpt
cd ../..

(3)下載voc數據集(提取碼: 3a8y),並修改名稱

mv VOCdevkit VOCdevkit2007

(4)運行(運行的參數中是GPU運行的參數設置,但是在經過第一次的修改後,爲gpu_id參數設置爲0的話,也是可以運行的)

./experiments/scripts/test_faster_rcnn.sh 0 pascal_voc_0712 res101
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章