導入ImageDataGenerator到Keras訓練的val_accuracy提升不上去?
2019/12/23 下午 06:20
電腦視覺深度學習討論版
胡連福
觀看數:55
回答數:2
收藏數:0
cvdl-1
cvdl-1-d16
您好,導入ImageDataGenerator到Keras訓練, 得到val_loss降不下來, val_accuracy提升不上去的結果(如附圖):
我用了train_set*8張圖, test_set*8張圖來做訓練, 使用了多層的Conv2D, MaxPooling2D, BatchNormalization, Dropout, 調整一些參數也都是一樣的結果, 但用此訓練模型去做predict, 卻都能得到正確的機率結果, 請問 :
1. train_loss有下降, train_accuracy也有提升, 但val_loss降不下來, 而且val_accuracy在第2個epoch就一直停留在0.5上不去, 是我的樣本太少造成的嗎 ?
2. 有無其他原因造成 ?
回答列表
-
2019/12/23 下午 06:22楊哲寧贊同數:0不贊同數:0留言數:0
您好,訓練影像太少了!一般來說每個類別沒有百張照片很容易會overfitting ,正常至少要數百張以上。
-
2019/12/23 下午 06:46Jeffrey贊同數:0不贊同數:0留言數:2