Tensorflow項目實戰-利用CNN實現簡單的旅遊景點識別

完整源代碼請見我的github地址:https://github.com/starmoon0309/Tourist_attraction_identification

本機環境

Win10 + python3.6 + pycharm + tensorflow-gpu 1.3.0 


下面簡單介紹一下整個流程:

首先,從網站上批量下載景點的圖片。我選用4個景點,分別是泰山,大雁塔,兵馬俑,頤和園。一共下載了共計804張圖片(有點少) 

調用create_record.py進行圖片處理,生成Attractions_train.tfrecords。這裏主要是將圖片轉換爲64×64像素的圖片。在這一步中我出現瞭如下的問題()
1.運行時報錯 NewRandomAccessFile failed to Create/Open
解決方法:文件夾沒有創建出來,所以將文件夾在運行的位置創建出來就好了
2.運行時報錯 InvalidArgumentError (see above for traceback): Input to reshape is a tensor with xx values, but the requested shape has xx
解決方法:在create_record.py中的img = Image.open(img_path)後加上了“.convert("RGB")”,強制都轉成RGB格式,就好了(困擾了我很久)
3.特別注意,文件的路徑上不要有中文名!!!忘了報的錯是什麼了

運行train.py,需要注意的是train_dir和logs_train_dir需要改成本機的目錄。大約訓練8~10分鐘左右,可以在/save目錄下得到訓練好的模型,如圖所示:

 

-調用test.py,進行測試。用於測試的是在百度圖片上隨便找的一張,命名爲1.jpg,同樣,文件裏的相關目錄也需要更改。需要注意的是,當程序彈出圖片後,需關掉,才繼續向下運行。。。

運行gui.py文件,彈出對話框,選擇待測試的圖片即可。 


(這裏有一個小小的問題,在程序運行出結果後仍會報錯,如下所示,有知道原因的可以在評論裏告知一下~)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章