When uploading my TF web model to cloud annotations, I get the following error in console when trying to run auto label on an image:

Unhandled Promise Rejection: Error: The shape of dict['x'] provided in model.execute(dict) must be [1,320,320,3], but was [1,1,1,3].

Any idea how to fix this? I can provide additional info if needed.

My model is a custom trained YOLOv5s (converted to a tfjs web model) with the image size set to 320x320 during training and export.