貓狗二分類模型

貓狗二分類模型點擊這裏。 記錄幾點: (1)這是一個在小型數據集上訓練模型的案例(訓練集2000張圖片,驗證集1000張,測試集1000張)。圖片資源點擊這裏。 (2)圖像輸入之前需要預處理,將其轉換爲浮點數張量,使用了keras.preprocessing.image中的ImageDataGenerator類,它可以實現圖像的縮放,旋轉,平移,錯切等操作。 展示幾張變換的圖片: (3)使用數據增
相關文章
相關標籤/搜索