數據集
? ? ? ? 自建鐵路障礙數據集-包含路障,人等少數標簽。其中百分之八十作為訓練集,百分之二十作為測試集
????????
第一次部署
? ? ? ? 版本:YOLOv5
?
????????訓練50epoch后精度可達0.94 mAP可達0.95.此時未包含任何改進操作
第二次部署
? ? ? ? 版本:YOLOv8改進版本
? ? ? ? 首先對訓練數據進行隨即裁剪+椒鹽加噪處理。
? ? ? ? 將c2f模塊替換為DSConv
????????加EMA注意力機制
????????更換EfficientViT主干網絡
?
?
?
?
?