Tensorflow lite在树莓派4B上进行图像分类和目标检测

接着前面系列博客来讲,这里来实现下官方例子

一.在Raspberry Pi 4B上进行图像分类

examples/lite/examples/image_classification/raspberry_pi at master · tensorflow/examples · GitHubTensorFlow examples. Contribute to tensorflow/examples development by creating an account on GitHub.https://github.com/tensorflow/examples/tree/master/lite/examples/image_classification/raspberry_pi下载完毕后,将examples-master整个文件夹上传到树莓派上,然后终端cd到如下目录下(结合自己的路径)

 这边博主修改了下setup.sh中的路径,不然tflite模型按照默认路径下载会失败,红框内的路径可以修改为如下:

https://storage.googleapis.com/tfhub-lite-models/tensorflow/lite-model/efficientnet/lite0/uint8/2.tflite

 完毕后,终端执行,安装一些必要库,同时下载两个tflite模型

sh setup.sh

可以看到下载的两个模型已经在当前文件夹下了

博主这边在pycharm中运行下classify.py文件

 连上一个usb的摄像头就可以取像了,实时去分类了,博主拿了一把汤勺做实验

二. 在Raspberry Pi 4B上进行目标检测

examples/lite/examples/object_detection/raspberry_pi at master · tensorflow/examples · GitHubhttps://github.com/tensorflow/examples/tree/master/lite/examples/object_detection/raspberry_pi参考上面官网进行配置,cd到目标检测目录下

 同上,这里也要修改下setup.sh中的路径

 红框内的路径修改为如下:

https://storage.googleapis.com/tfhub-lite-models/tensorflow/lite-model/efficientdet/lite0/detection/metadata/1.tflite

完毕后,运行安装必要库及下载检测模型

sh setup.sh

完毕后,可以看到目录下已经有了需要的模型

终端可直接如下命令,运行下检测demo程序

python3 detect.py --model efficientdet_lite0.tflite

 我家的小白猫被检测出来了

到此,demo结束。

仔细看下里面的代码,其实主要实现代码,在博主前面几篇博客中也用到了

Tensorflow Lite Model Maker实现图像分类和目标检测迁移学习_竹叶青lvye的博客-CSDN博客

Tensorflow Lite使用介绍_竹叶青lvye的博客-CSDN博客

基于树莓派4B的YOLOv5-Lite目标检测的移植与部署可以参考以下步骤: 1. 硬件准备:准备一台树莓派4B主机,具备至少4GB的内存和足够的存储空间。 2. 安装操作系统:在树莓派4B上安装支持深度学习的操作系统,如Raspberry Pi OS(之前的Raspbian)。 3. 安装深度学习框架:在树莓派上安装适用于ARM架构的深度学习框架,如PyTorch。 4. 下载YOLOv5-Lite模型:从YOLOv5的官方GitHub仓库下载YOLOv5-Lite模型的源代码和预训练权重。 5. 数据集准备:根据目标检测任务的需求,准备适合的训练数据集,并将其转换为YOLOv5可训练的格式。 6. 训练模型(可选):如果没有现成的适用于目标的预训练权重,可以选择在树莓派上训练YOLOv5-Lite模型。 7. 模型移植:将训练好的YOLOv5-Lite模型移植到树莓派4B上,确保模型能够在树莓派上运行。 8. 部署与测试:使用树莓派4B的摄像头或者外部摄像头采集图像或视频,并将其输入到模型中进行目标检测,观察模型在树莓派上的检测性能。 9. 优化与调整:根据实际情况,进行模型的优化和调整,如调整阈值、模型量化等,以提高模型在树莓派上的运行速度和准确性。 10. 应用开发:根据需求,将目标检测模型移植到实际应用中,如智能监控系统、机器人等,以实现对目标的实时检测和跟踪。 总之,基于树莓派4B的YOLOv5-Lite目标检测的移植与部署需要在树莓派上安装深度学习框架、准备训练数据集、移植模型、进行性能测试和优化,最终将模型应用到实际场景中。这需要对深度学习和树莓派4B有一定的了解和实践经验。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

竹叶青lvye

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

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

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

打赏作者

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

抵扣说明:

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

余额充值