qqwweee / keras-yolo3

A Keras implementation of YOLOv3 (Tensorflow backend)
MIT License
7.14k stars 3.45k forks source link

Unable to modify train code to accomodate train_on_batch #760

Closed SidhaantAnand closed 3 years ago

SidhaantAnand commented 3 years ago

Hey, I used the create_tiny_model to create the model body I am trying to use that model to train using the train_on_batch(x=input_batch, y=targets)

What should be the format and shape of input_batch and targets.

When I pass input batch with a shape of (batch_size,416,416,3), I keep seeing "ValueError: Error when checking model input: the list of Numpy arrays that you are passing to your model is not the size the model expected. Expected to see 3 array(s), but instead got the following list of 1 arrays" What are the three arrays