Yolov5+DeepSort 统计数量

GitHub - mikel-brostrom/Yolov5_DeepSort_Pytorch: Real-time multi-object tracker using YOLO v5 and deep sorthttps://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch

1、安装python3.8

1	apt-get install python3.8-distutils -y
2	apt-get install python3.8-tk -y
3	apt-get install curl -y
4	curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
5	python3.8 get-pip.py

2、下载代码

git clone --recurse-submodules https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch.git

3、安装依赖

pip install -r requirements.txt

这里的torch需要针对你的显卡和CUDA版本来进行安装。

python3.8 -m pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tun
a.tsinghua.edu.cn/simple

我的是3070显卡。

4、开始检测视频

python3.8 track.py --source 1.mp4  --save-vid --save-txt --classes 0

5、分析结果

id_dict = {}
for line in open('../exp11/1.txt'):
    id = line.split(' ')[1]
    if id not in id_dict:
        id_dict[id] = 1
    else:
        id_dict[id] += 1
print(len(id_dict))

{'1': 468,
 '2': 5,
 '3': 768,
 '7': 3,
 '9': 6,
 '10': 110,
 '11': 7,
 '19': 2,
 '21': 2,
 '23': 86,
 '27': 46,
 '29': 378,
 '34': 222,
 '35': 194,
 '43': 120}

 

很明显，出现少于10次的数据可以舍弃掉。

最终检测到9个人。

与人眼分析的结果一致！！