Train with a custom image_size (non-square)

[afazel]

Is it possible to run the training script with a custom image size? e.g 608 x 152?
From train.py command line option we have --img_size IMG_SIZE available, but it seems that it does not accept non-square image sizes? Is there any way to train with the custom image size?

[Petros626]

It should be possible, maybe not recommended. I would also use the standard input format of the model, e.g. 640x640.

You need to change the code for accepting custom image resolutions.