TensorFlow學(xué)習(xí)7:輸入圖片,預(yù)測結(jié)果

代碼處理過程

1影所,模型的要求是黑底白字氢哮,但輸入的圖是白底黑字,所以需要對每個像素點(diǎn)的值改為255減去原值以得到互補(bǔ)的反色
2型檀,對圖片做二值化處理
3,把圖片形狀拉成1行784列听盖,并把值變成浮點(diǎn)型(要求像素點(diǎn)是0-1之間的浮點(diǎn)數(shù))
4胀溺,計(jì)算求得輸出y裂七,y的最大值所對應(yīng)的列表索引號就是預(yù)測結(jié)果

示例代碼

#coding:utf-8
#將符合神經(jīng)網(wǎng)絡(luò)輸入要求的圖片喂給復(fù)現(xiàn)的神經(jīng)網(wǎng)絡(luò)模型,輸出預(yù)測值
def restore_model(testPicArr):
    #創(chuàng)建一個默認(rèn)圖仓坞,在該圖中執(zhí)行以下操作
    with tf.Graph().as_default() as tg:
        x=tf.placeholder(tf.float32,[None,mnist_forword.INPUT_NONE])
        y=mnist_forword.mnist_forword(x,None)
        #得到概率最大的預(yù)測值
        preValue=tf.argmax(y,1)

        #實(shí)現(xiàn)滑動平均模型背零,參數(shù)MOVING_AVERAGE_DECAY用于控制模型更新的速度
        #訓(xùn)練過程中會對每一個變量維護(hù)一個影子變量,這個影子變量的初始值
        #就是相應(yīng)變量的初始值无埃,每次變量更新時徙瓶,影子變量就會隨之更新
        variable_averages=tf.train.ExponentialMovingAverage(mnist_backward.MOVING_AVERAGE_DECAY)
        variable_to_restore=variable_averages.variable_to_restore()
        saver=tf.train.Saver(variable_to_restore)

        with tf.session() as sess:
            #通過checkpoint文件定位到最新保存的模型
            ckpt=tf.train.get_checkpoint_state(mnist_backward.MODEL_SAVE_PATH)
            if ckpt and ckpt.model_checkpoint_path:
                saver.restore(sess,ckpt.model_checkpoint_path)

                preValue=sess.run(preValue,feed_dict={x:testPicArr})
                return preValue
            else:
                print("No checkpoint file found")
                return -1

#預(yù)處理函數(shù),包括resize嫉称,轉(zhuǎn)變灰度圖侦镇,二值化操作
def pre_pic(picName):
    img=Image.open(picName)
    prIm=img.resize((28,28),Image.ANTIALIAS)
    im_arr=np.array(reIm.convert('L'))
    #設(shè)定合理的闕值
    threshold=50
    for i in range(28):
        for j in range(28):
            im_arr[i][j]=255-im_arr[i][j]
            if(im_arr[i][j]<threshold):
                im_arr[i][j]=0
            else:
                im_arr[i][j]=255
    nm_arr=im_arr.reshape([1,784])
    nm_arr=nm_arr.astype(np.float32)
    img_ready=np.multiply(nm_arr,1.0/255.0)

    return img_ready

參考:人工智能實(shí)踐:Tensorflow筆記

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市织阅,隨后出現(xiàn)的幾起案子壳繁,更是在濱河造成了極大的恐慌,老刑警劉巖荔棉,帶你破解...
    沈念sama閱讀 212,383評論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件闹炉,死亡現(xiàn)場離奇詭異,居然都是意外死亡润樱,警方通過查閱死者的電腦和手機(jī)渣触,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,522評論 3 385
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來壹若,“玉大人嗅钻,你說我怎么就攤上這事∩嘞。” “怎么了啊犬?”我有些...
    開封第一講書人閱讀 157,852評論 0 348
  • 文/不壞的土叔 我叫張陵,是天一觀的道長壁查。 經(jīng)常有香客問我觉至,道長,這世上最難降的妖魔是什么睡腿? 我笑而不...
    開封第一講書人閱讀 56,621評論 1 284
  • 正文 為了忘掉前任语御,我火速辦了婚禮,結(jié)果婚禮上席怪,老公的妹妹穿的比我還像新娘应闯。我一直安慰自己,他們只是感情好挂捻,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,741評論 6 386
  • 文/花漫 我一把揭開白布碉纺。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪骨田。 梳的紋絲不亂的頭發(fā)上耿导,一...
    開封第一講書人閱讀 49,929評論 1 290
  • 那天,我揣著相機(jī)與錄音态贤,去河邊找鬼舱呻。 笑死,一個胖子當(dāng)著我的面吹牛悠汽,可吹牛的內(nèi)容都是我干的箱吕。 我是一名探鬼主播,決...
    沈念sama閱讀 39,076評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼柿冲,長吁一口氣:“原來是場噩夢啊……” “哼茬高!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起姻采,我...
    開封第一講書人閱讀 37,803評論 0 268
  • 序言:老撾萬榮一對情侶失蹤雅采,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后慨亲,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體婚瓜,經(jīng)...
    沈念sama閱讀 44,265評論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,582評論 2 327
  • 正文 我和宋清朗相戀三年刑棵,在試婚紗的時候發(fā)現(xiàn)自己被綠了巴刻。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,716評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡蛉签,死狀恐怖胡陪,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情碍舍,我是刑警寧澤柠座,帶...
    沈念sama閱讀 34,395評論 4 333
  • 正文 年R本政府宣布,位于F島的核電站片橡,受9級特大地震影響妈经,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜捧书,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 40,039評論 3 316
  • 文/蒙蒙 一吹泡、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧经瓷,春花似錦爆哑、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,798評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽队贱。三九已至,卻和暖如春萝勤,著一層夾襖步出監(jiān)牢的瞬間露筒,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,027評論 1 266
  • 我被黑心中介騙來泰國打工敌卓, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人伶氢。 一個月前我還...
    沈念sama閱讀 46,488評論 2 361
  • 正文 我出身青樓趟径,卻偏偏與公主長得像,于是被迫代替她去往敵國和親癣防。 傳聞我的和親對象是個殘疾皇子蜗巧,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,612評論 2 350

推薦閱讀更多精彩內(nèi)容