Tensorflow object detection API 将自己的模型迁移到手机并实时检测

本文详细介绍了如何将使用Tensorflow Object Detection API训练的模型部署到Android手机上,实现实时目标检测。步骤包括将.pb模型文件放入Android Studio项目assets目录,编辑标签文件与模型对应,更新DetectorActivity.java中的模型和标签文件路径,以及调整MINIMUM_CONFIDENCE_TF_OD_API以控制检测的置信度阈值。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

本节将上两节内容合并,讨论如何将自己训练的图表检测模型部署到手机并实现实时检测.

1,打开上次的android studio,将上上一节训练的图表检测模型的.pb文件放到本项目的assets目录下.命名任意(本例yjf.pb)

2,复制assets目录下的任意一个.txt文件为yjf.txt,打开后从第二行开始写lable(第一行是???不用动,别删),label的顺利和训练模型时候的标号顺序一致.如:

???

table

graph

3,找到DetectorActivity.java文件,找到并修改TF_OD_API_MODEL_FILE 与 TF_OD_API_LABELS_FILE 变量为自己刚刚修改的yjf.pb和yjf.txt文件(默认前面的file不要动,直接把它的.pb或者.txt的名字改成自己的就行)

4,DetectorActivity.java中MINIMUM_CONFIDENCE_TF_OD_API  变量是检测到的目标的置信度,也即是说模型判定的这个框为该目标的可信程度.该值越高框越准确.如下为置信度为0.1和0.6时的框的效果.

 

至此tensorflow object detect api从数据准备,训练模型,到安卓平台部署完成.

评论 17
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

洛希极限-cz

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

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

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

打赏作者

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

抵扣说明:

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

余额充值