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:
- 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:
- https://github.com/benjamintanweihao/YOLOv3
- https://github.com/hunglc007/tensorflow-yolov4-tflite
- https://github.com/mystic123/tensorflow-yolo-v3?fbclid=IwAR0ojevKaFH2EIQvxmMfSgLPm9Mz9V_e_5E621BEckgGb-FpvuS7ZLcM7NI
- https://github.com/peace195/tensorflow-lite-YOLOv3
- https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects
- https://github.com/zzh8829/yolov3-tf2
- https://github.com/mystic123/tensorflow-yolo-v3
submitted by /u/mishaalnaeem
[visit reddit] [comments]