Skip to content

wxk-cmd/yolov5-mask-detection

Repository files navigation

yolov5-mask-detection

使用yolov5实现对口罩佩戴检测 其中训练好的模型有增加了CBAM注意力机制,修改了损失函数为EIoU。 运行只需要直接启动detect.py,训练代码在train.py.

About

使用yolov5实现对口罩佩戴检测

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages