直接上效果图:
所有代码显示如下:
只要运行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) #