基于改進YOLOv8n-Pose的軌道作業(yè)人員跨軌安全動作識別
國外電子測量技術(shù)
頁數(shù): 8 2024-08-15
摘要: 針對軌道作業(yè)人員跨軌安全動作監(jiān)督方法存在效率低、漏檢率高等問題,引入改進的人體姿態(tài)估計算法YOLOv8n-Pose對跨軌安全動作進行識別和監(jiān)督。對YOLOv8n-Pose算法改進方法為在網(wǎng)絡(luò)中添加注意力機制并輕量化網(wǎng)絡(luò)結(jié)構(gòu),并改進網(wǎng)絡(luò)的bbox損失函數(shù)和關(guān)鍵點損失函數(shù),以提高網(wǎng)絡(luò)的識別精度和速度。使用高斯濾波和ColorJitter算法對自制數(shù)據(jù)集增強。在訓(xùn)練前使用遺傳算法對訓(xùn)... (共8頁)