基于YOLOv5s的自動扶梯乘客異常行為實時檢測算法

摘要： 為了實時檢測乘客的異常行為，提出一種基于YOLOv5s算法的輕量化自動扶梯乘客異常行為實時檢測算法YOLO-STE。首先在主干網絡中引入輕量化ShuffleNetV2網絡，以減少主干網絡的參數量和計算量；其次在骨干網絡的最后一層引入基于Transformer編碼的C3TR模塊，以更好地提取豐富的全局信息和融合不同尺度的特征；最后在YOLOv5s的特征融合網絡中嵌入SE(Sque...