-
Notifications
You must be signed in to change notification settings - Fork 19
YOLOv5 TFLite
Aditya Lohia edited this page Jun 8, 2022
·
1 revision
This guide explains how to run YOLOv5 with tflite backend.
(Supports CPU on every platform, Full Dynamic Support)
You can use TFLite powered detector by specifying the backend parameter.
from cvu.detector import Detector
detector = Detector(classes="coco", backend = "tflite"))
Internally, the Detector will load TFLite pretrained Yolov5s weight model.
We will update dynamic export info soon, please check back again.
Notes
- We currently use
tf.lite
for Interpreter creation. In next update, we'll provide option oftflite_runtime
andpycoral