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

Table 1

Experiment results of the comparison between original YOLOv5-v6 and our proposed YOLOv5s-v6-HDCN with PIS loss on the driver mask 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	Imgsz	Bs	Epoch	Epoch	Class	mAP@.5	mAP@.5:.95
YOLOv5s-v6	224	64	300	151	All	0.877	0.639
					No_mask	0.905	0.631
					Mask	0.992	0.726
					Mask_incorrect	0.734	0.56
Yolov5s-v6-HDCN-PIS	224	64	300	242	All	0.944 ↑6.7%	0.722 ↑8.3%
					No_mask	0.919 ↑1.4%	0.671 ↑4.0%
					Mask	0.994 ↑0.2%	0.756 ↑3.0%
					Mask_incorrect	0.920 ↑18.6%	0.738 ↑17.8%