In the file vgg16.py, there are two functions called predict()
and test()
. The function predict()
calls self.model.predict(imgs)
, whereas test() calls self.model.predict_generator()
. Could anyone explain what exactly the differences are?
1 Like
It seems that model#predict takes in a numpy array and returns an array of predictions and model#predict_generator takes in a data generator and returns a predict generator.
You can find more on this here