Real-Time Object Detection for the Running Train Based on the Improved YOLO V4 Neural Network

Table 5

Ablation results.

Object detection algorithm	Track (%)	Signal (%)	Tunnel (%)	Train (%)	mAP (%)	FPS
YOLO V4	97.99	89.78	89.03	92.77	92.40	12.8
YOLO V4-tiny	97.58	86.34	87.29	90.16	90.34	21.37
MV3-YOLO-lite (NMS)	99.81	91.59	96.26	95.30	95.74	42.04
Soft-NMS MV3-YOLO-lite	99.70	92.79	96.04	95.71	96.06	41.73