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

前端 未结 1 1234
情歌与酒
情歌与酒 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)
提交回复
热议问题