赞
踩
参考链接1: https://github.com/yasenh/libtorch-yolov5.
参考链接2:https://github.com/ljk53/pytorch-rpi.
官方目前没有正式提供arm64架构的编译好的库,需要自己编译,顶部链接2中提供了部分arm架构预编译库文件,这里使用arm64架构的预编译文件。
下载链接2中arm64预编译库 “libtorch-rpi-aarch64-cxx11-abi-shared-without-deps-1.6.0.zip”
解压后会得到libtorch文件夹
注意:这里是缺少依赖的,不然最后make的时候会出现错误。
需要运行如下指令:
这样子libtorch才可以使用。
这里主要是用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目标检测环境,以便将来查询,供大家参考,感谢顶部参考链接中提供资源的大佬。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。