Categories
Misc

“ValueError: cannot reshape array of size 278540 into shape (256,128,3,3)” Conversion YOLOv3 .weights to .pb

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:

upstairs

downstairs

I have just 2 classes. I am unable to convert. Can someone please help?

Link to my weights and config file:

A few repos that I have tried:

submitted by /u/mishaalnaeem
[visit reddit] [comments]

Leave a Reply

Your email address will not be published.