PIS-YOLO: Real-Time Detection for Medical Mask Specification in an Edge Device

Table 2

Ablation experiment on the driver mask dataset and safety helmet dataset. “Epoch” denotes the real num of epoch using early stopping. “Epoch” is the num of epoch manual setting. “↑” is used to show the improvement rate over YOLOv5s-v6.

Model	Dataset	Imgsz	Bs	Epoch	Epoch	Class	mAP@.5	mAP@.5:.95
Yolov5s-v6-hdcn	driver_mask	224	64	300	182	All	0.910 ↑3.3%	0.686 ↑4.7%
						No_mask	0.897 ↓0.8%	0.638 ↑0.7%
						Mask	0.994 ↑0.2%	0.740 ↑1.4%
						Mask_incorrect	0.839 ↑10.5%	0.680 ↑12.0%
Yolov5s-v6-pis		224	64	300	217	All	0.897 ↑2.0%	0.688 ↑4.9%
						No_mask	0.899 ↓0.6%	0.651 ↑2.0%
						Mask	0.993 ↑0.1%	0.744 ↑1.8%
						Mask_incorrect	0.798 ↑6.4%	0.668 ↑12.8%
Yolov5s-v6	Safety_helmet	224	512	300	300	All	0.691	0.432
Yolov5s-v6-pis		224	512	300	300	All	0.704 ↑1.3%	0.442 ↑1.0%