faster-RCNN的訓(xùn)練和demo的實(shí)現(xiàn)

這個(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的測試竹习。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末誊抛,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子整陌,更是在濱河造成了極大的恐慌拗窃,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,682評論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件泌辫,死亡現(xiàn)場離奇詭異随夸,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)震放,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,277評論 3 395
  • 文/潘曉璐 我一進(jìn)店門宾毒,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人殿遂,你說我怎么就攤上這事诈铛。” “怎么了墨礁?”我有些...
    開封第一講書人閱讀 165,083評論 0 355
  • 文/不壞的土叔 我叫張陵幢竹,是天一觀的道長。 經(jīng)常有香客問我恩静,道長焕毫,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,763評論 1 295
  • 正文 為了忘掉前任蜕企,我火速辦了婚禮咬荷,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘轻掩。我一直安慰自己幸乒,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,785評論 6 392
  • 文/花漫 我一把揭開白布唇牧。 她就那樣靜靜地躺著罕扎,像睡著了一般。 火紅的嫁衣襯著肌膚如雪丐重。 梳的紋絲不亂的頭發(fā)上腔召,一...
    開封第一講書人閱讀 51,624評論 1 305
  • 那天,我揣著相機(jī)與錄音扮惦,去河邊找鬼臀蛛。 笑死,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的浊仆。 我是一名探鬼主播客峭,決...
    沈念sama閱讀 40,358評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼抡柿!你這毒婦竟也來了舔琅?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,261評論 0 276
  • 序言:老撾萬榮一對情侶失蹤洲劣,失蹤者是張志新(化名)和其女友劉穎备蚓,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體囱稽,經(jīng)...
    沈念sama閱讀 45,722評論 1 315
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡郊尝,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,900評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了粗悯。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片虚循。...
    茶點(diǎn)故事閱讀 40,030評論 1 350
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖样傍,靈堂內(nèi)的尸體忽然破棺而出横缔,到底是詐尸還是另有隱情,我是刑警寧澤衫哥,帶...
    沈念sama閱讀 35,737評論 5 346
  • 正文 年R本政府宣布茎刚,位于F島的核電站,受9級特大地震影響撤逢,放射性物質(zhì)發(fā)生泄漏膛锭。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,360評論 3 330
  • 文/蒙蒙 一蚊荣、第九天 我趴在偏房一處隱蔽的房頂上張望初狰。 院中可真熱鬧,春花似錦互例、人聲如沸奢入。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,941評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽腥光。三九已至,卻和暖如春糊秆,著一層夾襖步出監(jiān)牢的瞬間武福,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,057評論 1 270
  • 我被黑心中介騙來泰國打工痘番, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留捉片,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,237評論 3 371
  • 正文 我出身青樓,卻偏偏與公主長得像界睁,于是被迫代替她去往敵國和親觉增。 傳聞我的和親對象是個(gè)殘疾皇子兵拢,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,976評論 2 355