Edge Intelligence-Based OCS Fault Detection in Rail Transit Systems

Table 6

Comparison of results of the three algorithms.

	YOLOv5s	YOLOv5m	YOLOv5l
Model size	28 MB	84 MB	346 MB
mAP@0.5	95%	92%	91%
300 epochs; TrainingTime	11.220 h	14.590 h	14.990 h
291 images; DetectTime	154.825 s	149.068 s	153.242 s
The resulting; WeightModel	14.4 MB	42.5 MB	93.8 MB