當前位置:首頁 > 智能硬件 > 人工智能AI
[導(dǎo)讀] Tensorflow在更新1.0版本之后多了很多新功能,其中放出了很多用tf框架寫的深度網(wǎng)絡(luò)結(jié)構(gòu),大大降低了開發(fā)難度,利用現(xiàn)成的網(wǎng)絡(luò)結(jié)構(gòu),無論fine-tuning還是重新訓(xùn)練方便了不少。而且T

Tensorflow在更新1.0版本之后多了很多新功能,其中放出了很多用tf框架寫的深度網(wǎng)絡(luò)結(jié)構(gòu),大大降低了開發(fā)難度,利用現(xiàn)成的網(wǎng)絡(luò)結(jié)構(gòu),無論fine-tuning還是重新訓(xùn)練方便了不少。而且TensorFlow內(nèi)包含了一個強大的物體檢測API,我們可以利用這API來訓(xùn)練自己的數(shù)據(jù)集實現(xiàn)特殊的目標檢測。最近筆者終于跑通TensorFlow Object DetecTIon API的ssd_mobilenet_v1模型,這里記錄下如何完整跑通數(shù)據(jù)準備到模型使用的整個過程,相信對大家能有所幫助。

Object DetecTIon API提供了5種網(wǎng)絡(luò)結(jié)構(gòu)的預(yù)訓(xùn)練的權(quán)重,全部是用數(shù)據(jù)集進行訓(xùn)練,這五種模型分別是SSD+mobilenet、SSD+incepTIon_v2、R-FCN+resnet101、faster RCNN+resnet101、faster RCNN+incepTIon+resnet101。各個模型的精度和計算所需時間如下。下面及介紹下如何使用Object Detection去訓(xùn)練自己的模型。

這里TensorFlow的安裝就不再說明了,網(wǎng)上的教程一大把,大家可以找到很詳盡的安裝TensorFlow的文檔。

tensorflow 訓(xùn)練模型訓(xùn)練前準備工作

使用protobuf來配置模型和訓(xùn)練參數(shù),所以API正常使用必須先編譯protobuf庫,這里可以下載直接編譯好的pb庫( https://github.com/google/protobuf/releases ),解壓壓縮包后,把protoc加入到環(huán)境變量中:
$ cd tensorflow/models

$ protoc object_detection/protos/*.proto --python_out=.

(我是把protoc加到環(huán)境變量中,遇到找不到*.proto文件的報錯,后來把protoc.exe放到models/object_detection目錄下,重新執(zhí)行才可以)

然后將models和slim(tf高級框架)加入python環(huán)境變量:
PYTHONPATH=$PYTHONPATH:/your/path/to/tensorflow/models:/your/path/to/tensorflow/models/slim

tensorflow 訓(xùn)練模型數(shù)據(jù)準備

數(shù)據(jù)集需要轉(zhuǎn)化成PASCAL VOC結(jié)構(gòu),API提供了create_pascal_tf_record.py,把VOC結(jié)構(gòu)數(shù)據(jù)集轉(zhuǎn)換成.record格式。不過我們發(fā)現(xiàn)更簡單的方式,Datitran提供一種更簡單生產(chǎn).record格式的方法。

首先需要先要標注圖像相應(yīng)標簽,這里可以使用labelImg工具。每標注一張樣本,即生成一個xml的標注文件。然后,把這些標注的xml文件,按訓(xùn)練集與驗證集分別放置到兩個目錄下,在Datitran提供了xml_to_csv.py腳本。這里只要指定標注的目錄名即可。接下來,然后需要我們把對應(yīng)的csv格式轉(zhuǎn)換成.record格式。

def main():
# image_path = os.path.join(os.getcwd(), 'annotations')
image_path = r'D: raining-setsobject-detectionsunglasseslabel est'
xml_df = xml_to_csv(image_path)
xml_df.to_csv('sunglasses_test_labels.csv', index=None)
print('Successfully converted xml to csv.')

調(diào)用generate_tfrecord.py,注意要指定--csv_input與--output_path這兩個參數(shù)。執(zhí)行下面命令:
python generate_tfrecord.py --csv_input=sunglasses_test_labels.csv --output_path=sunglass_test.record

這樣就生成了訓(xùn)練及驗證用的train.record與test.record。接下來指定標簽名稱,仿照models/ object_detection/data/ pet_label_map.pbtxt,重新創(chuàng)建一個文件,指定標簽名。
item {
id: 1
name: 'sunglasses'
}

tensorflow 訓(xùn)練模型訓(xùn)練教程

根據(jù)自己的需要,選擇一款用coco數(shù)據(jù)集預(yù)訓(xùn)練的模型,把前綴model.ckpt放置在待訓(xùn)練的目錄,這里meta文件保存了graph和metadata,ckpt保存了網(wǎng)絡(luò)的weights,這幾個文件表示預(yù)訓(xùn)練模型的初始狀態(tài)。

打開ssd_mobilenet_v1_pets.config文件,并做如下修改:

num_classes:修改為自己的classes num

將所有PATH_TO_BE_CONFIGURED的地方修改為自己之前設(shè)置的路徑(共5處)

其他參數(shù)均保持默認參數(shù)。

準備好上述文件后就可以直接調(diào)用train文件進行訓(xùn)練。
python object_detection/train.py
--logtostderr
--pipeline_config_path= D:/training-sets /data-translate/training/ssd_mobilenet_v1_pets.config
--train_dir=D:/training-sets/data-translate/training

TensorBoard監(jiān)控:

通過tensorboard工具,可以監(jiān)控訓(xùn)練過程,輸入西面指令后,在瀏覽器輸入localhost:6006(默認)即可。
tensorboard --logdir= D:/training-sets/data-translate/training

本站聲明: 本文章由作者或相關(guān)機構(gòu)授權(quán)發(fā)布,目的在于傳遞更多信息,并不代表本站贊同其觀點,本站亦不保證或承諾內(nèi)容真實性等。需要轉(zhuǎn)載請聯(lián)系該專欄作者,如若文章內(nèi)容侵犯您的權(quán)益,請及時聯(lián)系本站刪除。
換一批
延伸閱讀

9月2日消息,不造車的華為或?qū)⒋呱龈蟮莫毥谦F公司,隨著阿維塔和賽力斯的入局,華為引望愈發(fā)顯得引人矚目。

關(guān)鍵字: 阿維塔 塞力斯 華為

加利福尼亞州圣克拉拉縣2024年8月30日 /美通社/ -- 數(shù)字化轉(zhuǎn)型技術(shù)解決方案公司Trianz今天宣布,該公司與Amazon Web Services (AWS)簽訂了...

關(guān)鍵字: AWS AN BSP 數(shù)字化

倫敦2024年8月29日 /美通社/ -- 英國汽車技術(shù)公司SODA.Auto推出其旗艦產(chǎn)品SODA V,這是全球首款涵蓋汽車工程師從創(chuàng)意到認證的所有需求的工具,可用于創(chuàng)建軟件定義汽車。 SODA V工具的開發(fā)耗時1.5...

關(guān)鍵字: 汽車 人工智能 智能驅(qū)動 BSP

北京2024年8月28日 /美通社/ -- 越來越多用戶希望企業(yè)業(yè)務(wù)能7×24不間斷運行,同時企業(yè)卻面臨越來越多業(yè)務(wù)中斷的風險,如企業(yè)系統(tǒng)復(fù)雜性的增加,頻繁的功能更新和發(fā)布等。如何確保業(yè)務(wù)連續(xù)性,提升韌性,成...

關(guān)鍵字: 亞馬遜 解密 控制平面 BSP

8月30日消息,據(jù)媒體報道,騰訊和網(wǎng)易近期正在縮減他們對日本游戲市場的投資。

關(guān)鍵字: 騰訊 編碼器 CPU

8月28日消息,今天上午,2024中國國際大數(shù)據(jù)產(chǎn)業(yè)博覽會開幕式在貴陽舉行,華為董事、質(zhì)量流程IT總裁陶景文發(fā)表了演講。

關(guān)鍵字: 華為 12nm EDA 半導(dǎo)體

8月28日消息,在2024中國國際大數(shù)據(jù)產(chǎn)業(yè)博覽會上,華為常務(wù)董事、華為云CEO張平安發(fā)表演講稱,數(shù)字世界的話語權(quán)最終是由生態(tài)的繁榮決定的。

關(guān)鍵字: 華為 12nm 手機 衛(wèi)星通信

要點: 有效應(yīng)對環(huán)境變化,經(jīng)營業(yè)績穩(wěn)中有升 落實提質(zhì)增效舉措,毛利潤率延續(xù)升勢 戰(zhàn)略布局成效顯著,戰(zhàn)新業(yè)務(wù)引領(lǐng)增長 以科技創(chuàng)新為引領(lǐng),提升企業(yè)核心競爭力 堅持高質(zhì)量發(fā)展策略,塑強核心競爭優(yōu)勢...

關(guān)鍵字: 通信 BSP 電信運營商 數(shù)字經(jīng)濟

北京2024年8月27日 /美通社/ -- 8月21日,由中央廣播電視總臺與中國電影電視技術(shù)學(xué)會聯(lián)合牽頭組建的NVI技術(shù)創(chuàng)新聯(lián)盟在BIRTV2024超高清全產(chǎn)業(yè)鏈發(fā)展研討會上宣布正式成立。 活動現(xiàn)場 NVI技術(shù)創(chuàng)新聯(lián)...

關(guān)鍵字: VI 傳輸協(xié)議 音頻 BSP

北京2024年8月27日 /美通社/ -- 在8月23日舉辦的2024年長三角生態(tài)綠色一體化發(fā)展示范區(qū)聯(lián)合招商會上,軟通動力信息技術(shù)(集團)股份有限公司(以下簡稱"軟通動力")與長三角投資(上海)有限...

關(guān)鍵字: BSP 信息技術(shù)
關(guān)閉
關(guān)閉