pip install -r requirements.txt
数据集为收集的不同的数据集统一整理后得到的
源码包含了yolov5的部分使用到的源码
visual.py为GUI文件，直接运行即可
使用之前可以改动是否启用GPU进行预测，
将self.device = 'cpu'中'cpu'改成'0' 即可使用GPU进行预测。
训练好的权重文件位于 runs/train/exp3/weights/best.pt
如果不想使用GUI可以直接运行detect.py文件。
-
Notifications
You must be signed in to change notification settings - Fork 0
WangHaoyuuu/yolov5-mask
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
YOLOv5是一种基于PyTorch的目标检测模型，可以快速准确地识别图像中的不同类别的物体。我们可以利用yolov5来实现一个是否佩戴口罩的预测系统。
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published