树莓派(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.python

1、libtorch

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

2、yolov5s

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

3、版本

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

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