当前位置：首页 > news >正文

TensorFlow Lite + OpenCV：在树莓派上实现实时图像识别

news 来源：原创 2025/4/29 13:10:16

1. 引言

在嵌入式 AI 领域，TensorFlow Lite（TFLite） 和 OpenCV 是两大核心工具：

TensorFlow Lite：适用于低功耗设备的深度学习推理框架。
OpenCV：用于图像处理、特征提取和人脸检测。

在本篇文章中，我们将使用 TensorFlow Lite 和 OpenCV，在 树莓派（Raspberry Pi） 上实现 实时图像识别（Object Detection） 和 人脸识别（Face Recognition），并提供详细的代码与优化方案。
在这里插入图片描述

2. TensorFlow Lite 与 OpenCV 的关系

2.1 TensorFlow Lite 的作用

TensorFlow Lite（TFLite）是 TensorFlow 的轻量级版本，专为移动设备和嵌入式设备设计，主要用于：

运行预训练的深度学习模型（如 MobileNet、YOLO、SSD）。
在低功耗设备上进行实时推理（如树莓派、Jetson Nano）。
支持模型量化（Quantization） 以减少计算量。

2.2 OpenCV 的作用

OpenCV 是一个计算机视觉库，主要用于：

图像处理（灰度化、边缘检测、特征点提取等）
人脸检测（基于 Haar 级联分类器或 DNN）
目标跟踪（Tracking）
摄像头操作（读取帧、绘制检测结果）

2.3 结合 TensorFlow Lite 和 OpenCV

TensorFlow Lite 负责深度学习推理（如分类、目标检测）。
OpenCV 负责前后处理（如摄像头读取、图像预处理、绘制识别结果）。

这种组合可以在 树莓派 上实现高效、低延迟的图像识别系统。

3. 环境搭建

3.1 硬件要求

树莓派 4（推荐 4GB/8GB 版本）
官方树莓派摄像头（或 USB 摄像头）
microSD 卡（推荐 32GB+）
电源适配器

3.2 安装 TensorFlow Lite 和 OpenCV

1. 更新系统

sudo apt update && sudo apt upgrade -y

2. 安装 OpenCV

sudo apt install python3-opencv

3. 安装 TensorFlow Lite

pip3 install tflite-runtime

4. 验证安装

python3 -c "import cv2; print(cv2.__version__)"
python3 -c "import tflite_runtime.interpreter as tflite; print('TFLite OK')"

4. 代码实现：实时图像识别（物体检测 + 人脸检测）

4.1 下载 TensorFlow Lite 预训练模型

mkdir -p ~/tflite_models && cd ~/tflite_models
wget https://storage.googleapis.com/download.tensorflow.org/models/tflite/coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.zip
unzip coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.zip

4.2 运行摄像头并进行目标检测

import cv2
import numpy as np
import tflite_runtime.interpreter as tflite

# 加载模型
model_path = "~/tflite_models/detect.tflite"
interpreter = tflite.Interpreter(model_path=model_path)
interpreter.allocate_tensors()

# 获取输入输出张量索引
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

# 打开摄像头
cap = cv2.VideoCapture(0)

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break
    
    # 预处理输入图像
    input_data = cv2.resize(frame, (300, 300))
    input_data = np.expand_dims(input_data, axis=0).astype(np.uint8)
    
    # 运行模型
    interpreter.set_tensor(input_details[0]['index'], input_data)
    interpreter.invoke()
    output_data = interpreter.get_tensor(output_details[0]['index'])
    
    # 绘制检测框
    for obj in output_data[0]:
        if obj[1] > 0.5:  # 置信度阈值
            ymin, xmin, ymax, xmax = obj[2:]
            cv2.rectangle(frame, (int(xmin), int(ymin)), (int(xmax), int(ymax)), (0, 255, 0), 2)
    
    cv2.imshow("Object Detection", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

5. 代码实现：人脸识别（Haar 级联分类器）

import cv2

# 加载人脸检测模型
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')

cap = cv2.VideoCapture(0)
while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break
    
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    faces = face_cascade.detectMultiScale(gray, 1.1, 4)
    
    for (x, y, w, h) in faces:
        cv2.rectangle(frame, (x, y), (x + w, y + h), (255, 0, 0), 2)
    
    cv2.imshow("Face Detection", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()