(29) 运动目标检测之python多线程调用YOLO检测
(29) 运动目标检测之python多线程调用YOLO检测
-
针对视频检测卡顿的问题,使用多线程是一个很好的解决方案。下面我将详细介绍如何实现一个主线程负责视频读取和显示,另一个线程负责 YOLO 检测,并通过线程安全的方式传递检测结果。
-
实现方案概述
-
主线程:负责读取视频帧、显示结果和控制程序流程
-
检测线程:专门负责运行 YOLO 模型进行目标检测
-
结果传递:使用线程安全的队列来传递帧和检测结果
-
结果获取:可以通过回调函数或直接访问结果队列获取检测结果
-
代码实现:
import cv2
import threading
import queue
import time
from collections import deque