基于YOLOv5s的自動(dòng)扶梯乘客異常行為實(shí)時(shí)檢測(cè)算法
激光與光電子學(xué)進(jìn)展
頁(yè)數(shù): 8 2023-08-23
摘要: 為了實(shí)時(shí)檢測(cè)乘客的異常行為,提出一種基于YOLOv5s算法的輕量化自動(dòng)扶梯乘客異常行為實(shí)時(shí)檢測(cè)算法YOLO-STE。首先在主干網(wǎng)絡(luò)中引入輕量化ShuffleNetV2網(wǎng)絡(luò),以減少主干網(wǎng)絡(luò)的參數(shù)量和計(jì)算量;其次在骨干網(wǎng)絡(luò)的最后一層引入基于Transformer編碼的C3TR模塊,以更好地提取豐富的全局信息和融合不同尺度的特征;最后在YOLOv5s的特征融合網(wǎng)絡(luò)中嵌入SE(Sque... (共8頁(yè))