核心概念
建設現場におけるヘルメット検出において、SE注意機構とC2fCIBブロックを組み込んだ新しいCIB-SE-YOLOv8モデルは、既存のYOLOモデルと比較して精度と効率が向上しており、リアルタイム安全管理に効果的なソリューションとなる。
本論文は、建設現場における安全装置、特にヘルメットのリアルタイム検出のための最適化されたYOLOv8モデルであるCIB-SE-YOLOv8を提案する研究論文である。
研究背景
建設現場における安全は非常に重要であり、ヘルメットなどの安全装置の着用は負傷や死亡のリスクを大幅に減らす。従来の監督者による目視確認は非効率的であるため、コンピュータビジョンと深層学習を用いたリアルタイム検出技術が注目されている。
研究目的
本研究は、公開されているヘルメット検出データセットSHEL5Kを用いて、YOLOv8nをベースに、SE注意機構とC2fCIBブロックを組み込んだ新しいモデルCIB-SE-YOLOv8を開発し、その性能を評価することを目的とする。
方法
データセット
SHEL5Kデータセットは、多様な背景画像を含む5,000枚の画像で構成され、訓練、検証、テストに7:2:1の比率で使用される。
モデルの開発
ベースラインモデルとしてYOLOv8nを採用し、バックボーンの特定の層のC2fモジュールをC2fCIBモジュールに置き換え、SE注意層を追加することでCIB-SE-YOLOv8を開発した。
評価指標
モデルの性能評価には、Mean Average Precision (mAP)、Precision、Recallを用いた。
結果
CIB-SE-YOLOv8は、ベースラインモデルと比較して、Precisionで0.5%、Recallで3.9%、mAP50で3.2%、mAP50-95で3.9%の改善を示した。また、パラメータ数とGFLOPsも削減され、効率性も向上した。
結論
CIB-SE-YOLOv8は、既存のYOLOモデルと比較して、ヘルメット検出において精度と効率が向上しており、建設現場におけるリアルタイム安全管理に効果的なソリューションとなる。
意義と今後の展望
本研究は、建設現場における安全性の向上に貢献するだけでなく、他の分野におけるオブジェクト検出タスクへの応用も期待される。今後は、より複雑な環境下での性能評価や、他の安全装置の検出への応用が課題となる。
統計
CIB-SE-YOLOv8は、YOLOv8nと比較して、mAP50で3.2%向上し、88.4%を達成した。
CIB-SE-YOLOv8は、YOLOv8nと比較して、Precisionで0.5%、Recallで3.9%向上した。
CIB-SE-YOLOv8は、2,683,222のパラメータと7.6 GFLOPsで動作する。
YOLOv8nは、3,006,038のパラメータと8.1 GFLOPsで動作する。