Keras: How to use predict_generator with ImageDataGenerator?

前端 未结 3 1553
我寻月下人不归
我寻月下人不归 2020-12-30 01:10

I\'m very new to Keras. I trained a model and would like to predict some images stored in subfolders (like for training). For testing, I want to predict 2 images from 7 clas

3条回答
  •  南笙
    南笙 (楼主)
    2020-12-30 01:12

    The problem is the inclusion of nb_samples in the predict_generator which is creating 14 batches of 14 images

    14*14 = 196
    

提交回复
热议问题