基于TensorFlow對自定義數(shù)字圖片搭建BP網絡

繼續(xù)上一篇中的問題2招狸,怎樣對自定義的圖片數(shù)據(jù)集進行訓練呢?在參數(shù)和模型固定的情況下乘凸,增加訓練集,有助于提高模型的泛化能力营勤。在前面壹罚,我們是對下載下來的mnist數(shù)據(jù)集進行的訓練,那么怎樣應用到我們自己制作或采集的圖片上呢猖凛?

如上面的圖片,我們需要借助tfrecords文件,這是一種能將圖像數(shù)據(jù)和標簽放在一起的二進制文件霞幅,可以提高內存效率,實現(xiàn)快速的讀取司恳、存儲等。

首先耍共,新建一個writer用于制作tfreords文件,num_pic用于計數(shù)试读,f 打開的文件是如下的txt文件,每行是圖片文件名和對應的數(shù)字钩骇,用for循環(huán)遍歷每張圖片。

這樣,value[0]就是圖片名傍妒,用于讀圖片,value[1]是對應標簽,用于生成一行10列的一維數(shù)組,對應標簽索引的值是1馏段。

tf.train.Example包含F(xiàn)eatures字段,features里包含feature字典力穗,鍵值對是img_raw和label對應的圖像數(shù)據(jù)和標簽數(shù)據(jù),格式是bytelist和Int64list够坐,這樣就把所有信息存到一個example文件里崖面,方便讀取元咙。

對于tfrecords文件的解析如下:

tf.train.string_input_producer生成一個先入先出的隊列queue巫员,用來讀取數(shù)據(jù),傳入存儲信息的tfrecords文件名列表赶掖,然后新建一個reader七扰,把讀取的每個樣本保存到serialized_example中奢赂,通過tf.parse_single_example將協(xié)議內存塊解析為張量颈走,傳入待解析的內存塊,features字典映射,傳入的字典鍵名要和制作的相同序厉,返回1*784的圖片張量和1*10的標簽毕箍。

get_tfrecords是隨機讀取batch_size個樣本,tf.train.shuffle_batch傳入待亂序處理的tensors[img,label],返回batch_size組的img,label.

這樣霉晕,我們通過tfrecords文件實現(xiàn)了對自定義圖片數(shù)據(jù)集的讀取,要注意的是還有修改反向傳播和test文件中的接口牺堰。

在backward文件中要先導入生成tfrecord的文件,傳入訓練樣本數(shù)恨搓,獲得每輪喂入的batch,在會話中開啟多線程協(xié)調器筏养,提高圖片和標簽批獲取效率。其中渐溶,tf.train.start_queue_runners()將會啟動輸入隊列線程,填入訓練樣本到隊列中茎辐,配合tf.train.Coordinator()在發(fā)生錯誤的情況下關閉線程。

同理拖陆,修改test文件,在獲取batch時get_tfrecord()中的isTrain=False.

這樣乎串,我們就解決了問題2,運行backward文件訓練數(shù)據(jù)叹誉,運行test文件觀察準確度闷旧,達到一定準確度后可以結合上一篇中的appliaction文件對實際圖片進行預測了。

需要注意的一點是鸠匀,在train和test集上準確度符合要求逾柿,但實際預測有錯誤時宅此,關注圖片的預處理過程,如修改閾值等父腕。



新手學習青瀑,歡迎指教!

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末斥难,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子群扶,更是在濱河造成了極大的恐慌,老刑警劉巖竞阐,帶你破解...
    沈念sama閱讀 222,252評論 6 516
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件暑劝,死亡現(xiàn)場離奇詭異,居然都是意外死亡担猛,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,886評論 3 399
  • 文/潘曉璐 我一進店門智嚷,熙熙樓的掌柜王于貴愁眉苦臉地迎上來纺且,“玉大人,你說我怎么就攤上這事载碌。” “怎么了嫁艇?”我有些...
    開封第一講書人閱讀 168,814評論 0 361
  • 文/不壞的土叔 我叫張陵,是天一觀的道長论皆。 經常有香客問我,道長点晴,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 59,869評論 1 299
  • 正文 為了忘掉前任粒督,我火速辦了婚禮陪竿,結果婚禮上屠橄,老公的妹妹穿的比我還像新娘。我一直安慰自己锐墙,他們只是感情好,可當我...
    茶點故事閱讀 68,888評論 6 398
  • 文/花漫 我一把揭開白布姐仅。 她就那樣靜靜地躺著刻盐,像睡著了一般掏膏。 火紅的嫁衣襯著肌膚如雪敦锌。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,475評論 1 312
  • 那天颖变,我揣著相機與錄音听想,去河邊找鬼腥刹。 笑死汉买,一個胖子當著我的面吹牛,可吹牛的內容都是我干的蛙粘。 我是一名探鬼主播,決...
    沈念sama閱讀 41,010評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼穴肘,長吁一口氣:“原來是場噩夢啊……” “哼舔痕!你這毒婦竟也來了评抚?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 39,924評論 0 277
  • 序言:老撾萬榮一對情侶失蹤耿眉,失蹤者是張志新(化名)和其女友劉穎鱼响,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體丈积,經...
    沈念sama閱讀 46,469評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡债鸡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 38,552評論 3 342
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了唬滑。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,680評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡晶密,死狀恐怖模她,靈堂內的尸體忽然破棺而出稻艰,到底是詐尸還是另有隱情侈净,我是刑警寧澤,帶...
    沈念sama閱讀 36,362評論 5 351
  • 正文 年R本政府宣布元扔,位于F島的核電站,受9級特大地震影響澎语,放射性物質發(fā)生泄漏溺忧。R本人自食惡果不足惜咏连,卻給世界環(huán)境...
    茶點故事閱讀 42,037評論 3 335
  • 文/蒙蒙 一鲁森、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧歌溉,春花似錦骑晶、人聲如沸草慧。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,519評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至舔示,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間竖共,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,621評論 1 274
  • 我被黑心中介騙來泰國打工公给, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留蜘渣,地道東北人淌铐。 一個月前我還...
    沈念sama閱讀 49,099評論 3 378
  • 正文 我出身青樓匣沼,卻偏偏與公主長得像,于是被迫代替她去往敵國和親捂龄。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 45,691評論 2 361

推薦閱讀更多精彩內容