這個(gè)實(shí)驗(yàn)是基于# dBeker/Faster-RCNN-TensorFlow-Python3做的
1、下載這個(gè)tensorflow的faster-RCNN版本秽五,并按照他的README.md完成配置里逆。
git clone https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.git
進(jìn)入其文件夾系馆,之后會(huì)看到如下所示的東西里面有三個(gè)文件夾刑巧,其中data是我們放數(shù)據(jù)的文件夾,里面有cache逆皮、coco宅粥、demo和scripts四個(gè)子文件夾,新建一個(gè)文件夾imagenet_weights电谣,將下載下來的VGG16模型放至該文件夾中秽梅,再將下載下來的PASCAL VOC2007的訓(xùn)練數(shù)據(jù)集放至其中,以供訓(xùn)練模型使用辰企。
2风纠、將以上東西都配置好了之后就可以進(jìn)行模型的訓(xùn)練了
python train.py
就可以完成訓(xùn)練,訓(xùn)練出來的模型會(huì)被放在“./default/voc_2007_trainval/default”目錄中牢贸,到時(shí)候調(diào)用demo.py的時(shí)候就可以直接調(diào)用demo.py
3、接下來是最讓人頭疼的調(diào)試demo.py的步驟了
一定要將demo.py中這一行代碼給注釋掉
tfmodel = os.path.join('output', demonet, DATASETS[dataset][0], 'default', NETS[demonet][0])
因?yàn)椴蛔⑨尩暨@一行代碼的話镐捧,會(huì)出現(xiàn)找不到訓(xùn)練出來的模型的問題潜索。
其次臭增,還要將以下代碼塊也一并注釋掉
if not os.path.isfile(tfmodel + '.meta'):
print(tfmodel)
raise IOError(('{:s} not found.\nDid you download the proper networks from '
'our server and place them properly?').format(tfmodel + '.meta'))
同時(shí)添加一行代碼
tfmodel = './default/voc_2007_trainval/default/vgg16_faster_rcnn_iter_40000.ckpt'
完成以上步驟還不足以完成demo.py的訓(xùn)練,還需要將\lib\config\config.py中的
tf.app.flags.DEFINE_string('network', "vgg16", "The network to be used as backbone")
轉(zhuǎn)換成
tf.app.flags.DEFINE_string('net', "vgg16", "The network to be used as backbone")
這樣便可完成demo.py的測試竹习。