树莓派(Raspberry Pi 4 Model B, arrch64)移动端使用libtorch部署yolov5s目标检测测试

记录yolov5s部署在树莓派(Ubuntu, arm64)移动端上操作以及libtorch使用测试

参考链接1: https://github.com/yasenh/libtorch-yolov5.
参考链接2:https://github.com/ljk53/pytorch-rpi.

一、libtorch

官方目前没有正式提供arm64架构的编译好的库,需要自己编译,顶部链接2中提供了部分arm架构预编译库文件,这里使用arm64架构的预编译文件。
下载链接2中arm64预编译库 “libtorch-rpi-aarch64-cxx11-abi-shared-without-deps-1.6.0.zip”
在这里插入图片描述
解压后会得到libtorch文件夹
注意:这里是缺少依赖的,不然最后make的时候会出现错误。
需要运行如下指令:
在这里插入图片描述
这样子libtorch才可以使用。

二、yolov5s

这里主要是用yolov5s来进行环境测试,将yolov5s的pt文件利用torchscript转化为适合部署在移动端的torchscript文件,参考链接1中提供了具体的操作步骤以及注意点,可按步骤进行,注意版本,已经测试实现。
这里的代码原来是提供于PC端测试,在树莓派系统上同理。

三、版本

ubuntu:20.04
opencv:3.4.8
python:3.8.2(这里其实安装树莓派ubuntu20系统时,默认的python版本就是3.8.2)

主要记录移动端部署libtorch使用yolov5s目标检测环境,以便将来查询,供大家参考,感谢顶部参考链接中提供资源的大佬。

  • 0
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值