基本思想: 原本想写一个YOLO5+Android的车辆检测和跟踪,但是发现无法达成实时检测效果,于是打算在tensorflow的tflite使用和探索一下;
(竟然发现一个哥们写的很YOLO5+Android版的项目,就复述一下https://github.com/zldrobit/yolov5/tree/tf-android)
一、git一下代码吧(复述)
ubuntu@ubuntu:~$git clone https://github.com/zldrobit/yolov5.git
ubuntu@ubuntu:~$ cd yolo5
ubuntu@ubuntu:~/yolo5$ git checkout tf-android
ubuntu@ubuntu:~/yolo5$ bash weights/download_weights.sh
ubuntu@ubuntu:~/yolo5$ python models/tf.py --weights weights/yolov5s.pt --cfg models/yolov5s.yaml --img 320
二、然后查看一下转换的文件及生成文件
ubuntu@ubuntu:~/yolov5/weights$ tree
.
├── download_weights.sh
├── yolov5s-fp16.tflite
├── yolov5s.pb
├── yolov5s.pt
└── yolov5s_saved_model
├── assets
├── saved_model.pb
└── variables
├── variables.data-00000-o