【詳解】keras + predict_generator + ImageDataGenerator.flow_from_ditectory 進行預測

本文主要講述在利用keras進行預測分類(不是訓練)時,從文件夾中直接讀取圖片(直接利用flow_from_direction),並經過 ImageDataGenerator進行增強。 可以講,這裏面的坑有點多,而且很微小。 先說說有哪些坑: 注意測試的文件夾名稱及結構,否則程序將讀取不到圖片; 設置好flow_from_directory的batch_size,否則預測的結果與圖片不對應; 如何
相關文章
相關標籤/搜索