【教学】yolov5_deepsort目标跟踪行人统计数量,可鼠标点击设置统计区域

直接上效果图:

所有代码显示如下:

只要运行01main.py即可。会弹出视频的第一帧,然后我们对第一帧点击鼠标,选个碰撞区域即可,就能用来统计通过的行人了。

其他文件,如weights下存放的模型文件,这里直接用原始的coco数据集模型即可。

然后video文件夹下是待检测的视频文件,同时处理后的结果视频也会保存在这个文件夹下。

主函数代码如下:

import numpy as np
import os
import tracker
from detector import Detector
import cv2

#定义一个函数用于在本地生成视频其中一张的截图,便于后续的手动获取坐标点
def getone(root):
    vc = cv2.VideoCapture(root)  # 读取需要处理的视频
    if vc.isOpened():  # 判读视频是否正常打开
        print("打开ok")
    else:
        print("打开失败,程序退出")
        exit(-1)  # 

猜你喜欢

转载自blog.csdn.net/qq_34904125/article/details/124906085