preprocessing images generated using keras function ImageDataGenerator() to train resnet50 model

前端 未结 1 1235
情歌与酒
情歌与酒 2021-01-03 15:34

I am trying to train resnet50 model for image classification problem.I have loaded the \'imagenet\' pretrained weights before training the model on the image dataset I have.

1条回答
  •  粉色の甜心
    2021-01-03 16:01

    As an argument when creating ImageDataGenerator:

    train_datagen = ImageDataGenerator(preprocessing_function=preprocess_input)
    

    0 讨论(0)
提交回复
热议问题