在熟练使用labelimg打标及训练识别之后,就可以做一些有趣的事情了,比如说识别下是否戴口罩,如果没带的话提示“请佩戴口罩”。https://download.csdn.net/download/buaaweibin/85802968
这里有训练好的 mask.pt 文件,只需要使用该权重,就可以识别是否佩戴口罩。可以把mask.pt放在weights目录下,这样就可以实时检测戴口罩了。
python detect.py --weights weights/mask.pt --source 0
当然今天的主要内容不在识别,而是track追踪。
使用yolov5+deepsort 是最容易的实现追踪的方法。
下面为具体实现步骤:
选好兼容版本,使用yolov5-5.0和
Yolov5_StrongSORT_OSNet-3.0,python3.6
网站分别为
https://github.com/ultralytics/yolov5/releases
https://github.com/mikel-brostrom/Yolov5_StrongSORT_OSNet/releases
注意要在 releases 找到对应的版本哦。注意yolov5-6.0和5.0的pt权重文件不能混用。
下面我们又要开始构建熟悉的环境了。
conda create --prefix=D:/PycharmProjects/yoloV5_deepsort/yoloV5deepsort_env python=3.6
如果嫌慢,建立环境时先暂时添加清华镜像,后期取消掉
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pk