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.24k stars 1.24k forks source link

Is it possible to deploy on this app yolov5 model ? #473

Open HripsimeS opened 1 year ago

HripsimeS commented 1 year ago

@hunglc007 Hello. Thanks for this repository, it works perfect on initial yolov4-416-fp32.tflite model. I have a model trained on yolov5s and I converted pt to tflite with the following script https://github.com/ultralytics/yolov5/blob/master/export.py

I get an issue and my app does not work while I am trying to deploy my yolov5s.tflite model. The app depends only yolov4 or it should work also with yolov5 normally?