search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.23k stars 1.24k forks source link

Deploy Yolov3 on android #429

Open teguh161402025 opened 2 years ago

teguh161402025 commented 2 years ago

I convert my yolov3 model to tflite and its succes and then i want deploy my yolo v3 tflite model to android its run succesfull but the camera is freezing and laggy , I tried to use yolov4 and its run normaly. I want ask how can this happening? Is that cause yolov3 is too heavy for mobile?

Pratyaksha-047 commented 2 years ago

Hi, I am trying to convert my yolov3 custom model to tflite but I am getting an error: conv_weights = conv_weights.reshape(conv_shape).transpose([2, 3, 1, 0]) ValueError: cannot reshape array of size 2318619 into shape (1024,512,3,3)

on running line: !python save_model.py --weights ./data/yolov3_training_last.weights --output ./checkpoints/yolov3-416 --input_size 416 --model yolov4 --framework tflite

I have already changed the path of __C.YOLO.CLASSES to my classes file in config.py

I have used AlexeyAB/darknet for yolov3 model. Can you please help me resolve this issue. thanks!