본 연구는 악천후 조건에서 자율주행을 위한 의미론적 분할 모델의 일반화 성능을 향상시키는 것을 목표로 합니다. 특히, 현실적이고 다양한 날씨 환경을 반영하는 합성 데이터를 생성하여 기존 모델의 성능 저하 문제를 해결하고자 합니다.
본 논문에서는 Stable Diffusion (SD)과 Large Language Model (LLM)을 결합한 새로운 데이터 증강 프레임워크인 WeatherDG를 제안합니다.
먼저, 소스 도메인 데이터로 SD를 미세 조정하여 생성된 이미지의 내용과 레이아웃을 실제 자율주행 시나리오에 맞춥니다.
LLM을 기반으로 하는 프롬프트 생성 방법을 사용하여 다양하고 세부적인 이미지를 생성합니다. LLM 에이전트를 활용하여 시나리오 설명(프롬프트)을 점진적으로 풍부하게 만들고, 다양한 날씨 및 조명 효과를 생성합니다. 또한, 균형 잡힌 생성 전략을 통해 악천후 조건에서 드물게 나타나는 객체(예: 오토바이 운전자)를 풍부하게 생성합니다.
미세 조정된 SD와 생성된 프롬프트를 사용하여 새롭고 다양한 샘플을 생성하고, 이를 소스 데이터와 함께 모델 학습에 사용합니다.
본 논문에서 제안하는 WeatherDG는 세 가지 주요 데이터 세트(ACDC, BDD100k, DarkZurich)에서 최첨단 성능을 달성했습니다. 특히, Cityscapes에서 ACDC로의 도메인 일반화 설정에서 WeatherDG는 기준 모델인 HRDA보다 mIoU가 13.9% 향상되었습니다.
WeatherDG는 LLM과 SD를 결합하여 사실적이고 다양한 날씨 환경의 자율주행 이미지를 생성하는 효과적인 방법입니다. 본 연구는 악천후 조건에서 자율주행 시스템의 안전성과 신뢰성을 향상시키는 데 기여할 수 있습니다.
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Chenghao Qia... a las arxiv.org 10-17-2024
https://arxiv.org/pdf/2410.12075.pdfConsultas más profundas