charlieWyatt

Thanks @sovit-123 I appreciate the response. I am printing out the image.shape here in onnx_inference_image.py-  And am getting a value of - `torch.Size([1, 3, 640, 640])` Here is an...

inference on same image folder path using inference.py - `python inference.py --data data_configs/iNaturalist.yaml --weights outputs/training/fasterrcnn_mobilenetv3_large_fpn_iNaturalist/best_model.pth --input ../input/iNaturalist/inference_images/`  Old versions - onnxruntime version - 1.16.3 onnx version - 1.15.0 Tried...

I have a suspicion it is because of the redefinition of the roi heads in the mobilenet model on line 16 -  Also the inference error seems to indicate...

No worries, thanks @sovit-123!