参考连接1: https://github.com/yasenh/libtorch-yolov5.
参考连接2:https://github.com/ljk53/pytorch-rpi.python
官方目前没有正式提供arm64架构的编译好的库,须要本身编译,顶部连接2中提供了部分arm架构预编译库文件,这里使用arm64架构的预编译文件。
下载连接2中arm64预编译库 “libtorch-rpi-aarch64-cxx11-abi-shared-without-deps-1.6.0.zip”
解压后会获得libtorch文件夹
注意:这里是缺乏依赖的,否则最后make的时候会出现错误。
须要运行以下指令:
这样子libtorch才能够使用。git
这里主要是用yolov5s来进行环境测试,将yolov5s的pt文件利用torchscript转化为适合部署在移动端的torchscript文件,参考连接1中提供了具体的操做步骤以及注意点,可按步骤进行,注意版本,已经测试实现。
这里的代码原来是提供于PC端测试,在树莓派系统上同理。github
ubuntu:20.04
opencv:3.4.8
python:3.8.2(这里其实安装树莓派ubuntu20系统时,默认的python版本就是3.8.2)web
主要记录移动端部署libtorch使用yolov5s目标检测环境,以便未来查询,供你们参考,感谢顶部参考连接中提供资源的大佬。ubuntu