SimCLR-YOLO: 낮은 주석 의존성을 위한 도로 손상 감지 알고리즘

XIANG Tianming ,  

YAN Fang ,  

LI Jiahui ,  

SUN Rui ,  

摘要

도로 손상 감지에서 샘플 부족, 약한 텍스처 미검출률 증가 및 복잡한 배경 간섭 문제를 해결하기 위해 비지도 사전학습 SimCLR과 YOLOv8n을 융합한 도로 손상 감지 알고리즘 SimCLR-YOLO를 제안한다. 먼저, SimCLR 백본을 YOLOv8n의 CSPDarknet으로 대체하여 구조 정렬과 특징의 무결한 이전을 구현함으로써 네트워크 구조 불일치로 인한 특징 손실 문제를 해결한다. 둘째, 마스크 재구성을 대비 학습에 통합하여 인코더가 균열과 포트홀의 텍스처 윤곽에 집중하도록 유도하여 약한 텍스처 특징에 대한 대비 학습의 미적합 문제를 보완한다. 마지막으로 CBAM 주의 모듈을 삽입하여 사전학습 단계에서 도로 텍스처, 그림자 등의 배경 노이즈를 억제하도록 모델을 유도하여 주요 영역의 특징 반응을 강화한다. RDD2022 공개 데이터셋에서의 실험 결과 SimCLR-YOLO 알고리즘은 mAP@0.5에서 89.4%를 달성하여 기준 모델 대비 1.7% 향상되었으며, mAP@0.5:0.95는 1.1%, F1-score는 1.8% 향상되었다. 다양한 주석 비율에서의 검증 결과는 사전학습 가중치 도입이 모든 그라디언트에서 처음부터 학습하는 것보다 우수함을 보여주며, 50% 주석 비율에서도 SimCLR-YOLO 알고리즘의 mAP@0.5는 87%에 달해 전체 주석 하의 기준 모델에 근접하며, 주석 데이터 의존도를 크게 줄였다. 또한 개선은 사전학습 단계에서만 유효하며, 추론 매개변수(3.01M)에 추가 비용이 없어 샘플 부족 조건에서 도로 손상 실시간 감지를 위한 효과적인 기술 솔루션을 제공한다.

关键词

도로 손상;YOLOv8n;SimCLR;비지도;낮은 주석 의존성;사전학습

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website