7
Export YOLOv5 models for mobile device
source link: http://www.donghao.org/2020/11/13/export-yolov5-models-for-mobile-device/
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
Export YOLOv5 models for mobile device
Somebody has finished the work about exporting YOLOv5 models to tflite model. To use it, we only need to:
git clone --single-branch --branch tf-export https://github.com/zldrobit/yolov5.git cd yolov5 # it will download all pytorch models sh -x weights/download_weights.sh # export a tflite model from yolov5l PYTHONPATH=. python3 models/tf.py --weights yolov5l.pt --cfg models/yolov5l.yaml --img 640 # there will be a tflite model file ls yolov5l-fp16.tflite
Python
git clone --single-branch --branch tf-export https://github.com/zldrobit/yolov5.git
cd yolov5
# it will download all pytorch models
sh -x weights/download_weights.sh
# export a tflite model from yolov5l
PYTHONPATH=. python3 models/tf.py --weights yolov5l.pt --cfg models/yolov5l.yaml --img 640
# there will be a tflite model file
ls yolov5l-fp16.tflite
The model file yolov5l-fp16.tflite
is 91MB, which is a little too big but still could be put into a mobile phone.
Like this:
Loading...
tflite, YOLO
Leave a comment
Recommend
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK