Expert Python developer to fix an issue in one object detection code.

Job ID: 35890705

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

I need to update an existing file that I use to export the YOLOv7 model into ONNX format. The script by default saves the input shape in (batch size, channel, width, height) format. I want to save it in (batch size, width, height, channel) format so that I can later use it in tensorflow lite model in android studio.
Current problem and possible solution is mentioned in this blog-
https://medium.com/geekculture/journey-putting-yolo-v7-model-into-tensorflow-lite-object-detection-api-model-running-on-android-e3f746a02fc4
And export.py file can be found here-
https://github.com/WongKinYiu/yolov7

So if you can make these changes and can share working files then let me know.
Related categories: Python Software Architecture Computer Vision