树莓派4/5:运行Yolov5n模型(文末附镜像文件)

发布于:2024-08-11 ⋅ 阅读:(145) ⋅ 点赞:(0)

〇、前言

因国内网络问题,可直接烧录文末镜像文件,或者按照本教程进行手动操作。

 

一、实验目的

在树莓派4B运行Yolov5n模型。

 

二、实验条件

1、Windows 11计算机:安装了Mobaxterm

2、树莓派4B:64Bit Lite OS,安装了miniforge

如果树莓派未安装系统,请先查看以下链接。

http://t.csdnimg.cn/3erlihttp://t.csdnimg.cn/3erli如果树莓派未安装miniforge或未添加清华镜像源,请先查看以下链接。

http://t.csdnimg.cn/t3cgFhttp://t.csdnimg.cn/t3cgF

 

三、实验准备

1、更新软件

sudo apt update
sudo apt upgrade

2、安装opencv python

sudo apt install python3-opencv

3、安装git

sudo apt install git

 

四、添加多个PIP镜像源(可选)

1、打开pip下载源配置文件

mkdir -p $HOME/.pip && nano $HOME/.pip/pip.conf

2、添加多个pip镜像源

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

[install]
extra-index-url =
    https://mirrors.aliyun.com/pypi/simple/
    https://pypi.org/simple

编辑完成后,使用CTRL+X关闭文件,并键入Y,最后按Enter。

 

五、实验环境

1、创建pytorch环境

conda create -n pytorch python==3.8

2、激活pytorch环境

conda activate pytorch

3、安装pytorch

conda install pytorch torchvision cpuonly -c pytorch

4、安装yolov5

# 官方地址
git clone https://github.com/ultralytics/yolov5  # clone
# 加速地址
git clone https://mirror.ghproxy.com/https://github.com/ultralytics/yolov5  # clone

cd yolov5
pip install -r requirements.txt  # install

因国内网络问题,安装过程可能比较久,耐心等候即可。在安装过程中,如果某个库安装失败,手动使用pip或conda安装后,再安装一遍完整的依赖项即可。

 

六、实验过程

1、将USB摄像头连接到树莓派。

2、在pytorch虚拟环境中,在yolov5文件夹中运行以下代码。

python detect.py --weights yolov5n.pt --source 0

如果成功运行,则弹窗显示当前推理过程,并在Mobaxterm中打印检测结果。

6f16bebda12c4d4a83be362964699f29.png

 

七、镜像下载

根据上述操作,完整备份一个镜像,以下为镜像的基本信息。

f740855cec4642c3bb62486b5e74d5ed.png

文件较大,请从百度网盘下载。需要使用≥16GB的SD卡。

[待添加]

 

如果SD卡>16GB,可利用raspi-config扩容。

sudo raspi-config

依次选择Advanced Options >> Expand Filesystem,完成后退出raspi-config,然后重启系统。

sudo reboot

重启后,树莓派将完成扩容。

 


网站公告

今日签到

点亮在社区的每一天
去签到