機器學習之貓狗大戰,解決image RGB values must be in the 0..1 range.


 貓狗大戰是比較經典的機器學習案例,前幾天體驗了一番,來記錄一下

1.圖片准備

首先是准備訓練的圖片

鏈接:https://pan.baidu.com/s/1ht1HIuw 密碼:aw9s

 

 

 

2.開始訓練

需要的時間可能比較長

 

 

 

注:可能出現image RGB values must be in the 0..1 range.的錯誤

可以嘗試修改下面的代碼,將img[j,:,:,:]數組的值調到0到1之間,

0.1 * old_img + 0.5 對圖片顯示影響較小,不過可能會偏移;

0.2 * abs(old_img)對圖片位置影響較小,不過可能顯示會失真。

可參考:https://stackoverflow.com/questions/47318871/valueerror-floating-point-image-rgb-values-must-be-in-the-0-1-range-while-usi

 

 

3.測試訓練

再找幾張貓狗圖片進行測試

test.py

 

 

參考:

TensorFlow——訓練自己的數據(一)數據處理:http://blog.csdn.net/xinyu3307/article/details/74643019
TensorFlow——訓練自己的數據(二)模型設計:http://blog.csdn.net/xinyu3307/article/details/74943033
TensorFlow——訓練自己的數據(三)模型訓練:http://blog.csdn.net/xinyu3307/article/details/74979842
TensorFlow——訓練自己的數據(四)模型測試:http://blog.csdn.net/xinyu3307/article/details/75008458

用自己的圖片數據做tensorflow深度學習:http://blog.csdn.net/baidu_23263735/article/details/74177831
github地址:https://github.com/kevin28520/My-TensorFlow-tutorials/tree/master/01%20cats%20vs%20dogs


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM