Android developer + Python + YOLO

Job ID: 35922673

Budget: ₹1,500 – ₹12,500 INR

The export.py file mentioned in the below link exports any YOLOv7-based model to ONNX format that we later export to TensorFlow and TFLite-
https://github.com/WongKinYiu/yolov7

The issue with this script is that the exported model gives the following error in the android studio while loading-
The input tensor should have dimensions 1 x height x width x 3. Got 1 x 3 x 1280 x 1280.
Caused by: java.lang.IllegalArgumentException: Error occurred when initializing ObjectDetector: The input tensor should have dimensions 1 x height x width x 3. Got 1 x 3 x 1280 x 1280.

So developer need to fix this issue. For reproducibility, I can share the exported TFLite model and you can load it in the android studio where you can see no bounding boxes on an image. Although the exported model with Python works fine but with the TFLite file issue occurs.

Let me know if you have experience in Python with android studio and can fix the export model issue.