Looking at the code in Vgg16.py in the create function there is a line :-
model.add(Lambda(vgg_preprocess, input_shape=(3,224,224), output_shape=(3,224,224)))
Suggest the line in the VGG16 model should be changed with the one above.
Tried it my self and it worked
1 Like