lhy0718/yolov7

YOLO7 for TFLite

Conversion YOLOv7 from Pytorch to TF Lite

Original YOLOv7 code - https://github.com/WongKinYiu/yolov7

Compatible with TensorFlow ObjectDetector objects

• TensorFlow ObjectDetector API
• TensorFlow ObjectDetector Android Demo App

Coversion Overview

Pytorch → ONNX → TensorFlow → TF Lite

Changes

add torch-to-tflite.py

modify model/experimental.py

Acknowledgements

Expand

• https://medium.com/geekculture/journey-putting-yolo-v7-model-into-tensorflow-lite-object-detection-api-model-running-on-android-e3f746a02fc4
• https://github.com/WongKinYiu/yolov7
• https://github.com/AlexeyAB/darknet
• https://github.com/WongKinYiu/yolor
• https://github.com/WongKinYiu/PyTorch_YOLOv4
• https://github.com/WongKinYiu/ScaledYOLOv4
• https://github.com/Megvii-BaseDetection/YOLOX
• https://github.com/ultralytics/yolov3
• https://github.com/ultralytics/yolov5
• https://github.com/DingXiaoH/RepVGG
• https://github.com/JUGGHM/OREPA_CVPR2022
• https://github.com/TexasInstruments/edgeai-yolov5/tree/yolo-pose