I have trained a YOLO v3 Object Detection Model. To incorporate into my flutter application I am trying to convert it to .tflite, with .pb needed as intermediate. I am getting this error with every github repo I have tried. (A few linked below)
Error: ValueError: cannot reshape array of size 278540 into shape (256,128,3,3)
Following is what my classes.names file looks like:
I have just 2 classes. I am unable to convert. Can someone please help?
Link to my weights and config file:
- Weights: https://drive.google.com/file/d/1dqKHqzaqhHuxO5wuW9mgIeCbor9NyTMh/view?usp=sharing
- Cfg: https://drive.google.com/file/d/1_e607nkMTObKlYUCQc8IZtXMsUKJ_aab/view?usp=sharing
A few repos that I have tried: