人脸检测与表情推理阶段一:闭环系统验证——环境准备

0 阅读2分钟

0 环境准备

一、操作系统

VirtualBox虚拟机软件+Ubuntu24.04

选择VirtualBox(VB)原因:虽然在各个方面VM都碾压VB,但是VB是开源且免费的,而VM免费版有相关限制,VMpro(要米);针对此项目,这个阶段只是为了验证系统闭环,即:摄像头-->模型-->推理-->输出,这个过程不需要强大的虚拟机软件,所以VB即可。

选择Ubuntu24.04原因:稳定且较新。

二、系统环境配置

首先是检查并更新软件库(强烈建议每次开机先跑一遍)

sudo apt update 
sudo apt upgrate

三、安装依赖

依赖:代码正常运行所要依靠的东西 本验证阶段需要的三个核心依赖:python、onnxruntime、cv2

安装python基础环境

sudo apt install -y python3 python3-pip python3-venv

-y:下载过程中自动帮你补齐yes

pip:用来安装、管理python的各种库

venv:用来创建虚拟环境。为了防止pip下载一些包破坏了系统的Python环境,通常在虚拟环境中使用pip下载和管理包。

验证:

python --version
pip --version

如果有对应版本号,即下载成功。

创建、打开、关闭子环境代码

创建

python3 -m venv ort_env

在终端里输

source ort_env/bin/activate

即可进入。其中,ort_env换成自己的子环境文件夹名。

在虚拟环境下输入

deactivate

即可退出。

安装onnxruntime

子环境激活的情况下运行

pip install onnxruntime 

也可以用镜像

pip install onnxruntime -i https://pypi.tuna.tsinghua.edu.cn/simple

-i 指定下载源(index)

什么是runtime?什么是onnx?

onnx:一种专门用于模型文件的拓展名。正常情况下,在模型训练阶段,模型结构通常以. py文件格式保存。在训练过程中,会产生储存权重的.pth文件。如果想把模型部署到一个算力较小,跑不动训练的地方(某些arm架构的边缘ai产品),就需要一个“静态计算图”。这张图就是onnx。

注:不是所有模型都可以导出成onnx,注意onnx的静态特性。

runtime=把代码语言解释成机器语言+安排计算的执行顺序+管理过程中的内存+调用别人优化好的计算库+安排在哪运行(GPU,CPU,...)

验证onnxruntime正确安装

import onnxruntime as ort
print("ONNX Runtime version:",ort.__version__)
print("Available providers:",ort.get_available_providers())

安装和验证cv库

安装:

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple

验证;

python -c "import cv2; print(cv2.__version__)"

-c:让python执行字符串里的代码

摄像头透传与验证(仅用于笔记本自带摄像头)

VB的设置(非常重要!!!) image.png

image.png

image.png

image.png

终端里输入

ls /dev/video*
sudo apt install cheese
cheese

基础环境配置到此结束,下一步,摄像头的数据流能传入模型,并简单处理每帧图像,敬请期待。