31、TensorFlow训练模型转成tfilte,进行python、Android端进行车辆检测、跟踪、部署

本文介绍了如何将训练好的TensorFlow模型转换为tflite格式,以便在Python和Android平台上进行车辆检测和跟踪。过程中涉及到数据集的收集、YOLO标注、模型训练、数据转换、tflite生成以及Android端的测试。
摘要由CSDN通过智能技术生成

基本思想: 原本想写一个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
  • 3
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

sxj731533730

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值