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

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