Tensorflow實現圖像分類

Tensorflow實現圖像分類 利用tensorflow做圖像分類的訓練,主要分爲讀取圖像,訓練批次,定義模型,訓練,保存模型,模型調用幾步。 首先是讀取圖像,利用python中os模塊的listdir方法來讀取圖像。在這裏可以把每一類的圖像放到同一文件夾中,文件夾的名字可以直接作爲圖像的標籤,最好是用數字代替,圖像的名字最好也是數字,這樣寫代碼比較方便,如圖: 讀取圖像的代碼如下: 接下來要做
相關文章
相關標籤/搜索