树莓派(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目标检测环境,以便将来查询,供大家参考,感谢顶部参考链接中提供资源的大佬。

你可能感兴趣的:(C++,pytorch,github,ubuntu)