Temel Kavramlar
본 논문에서는 Restormer-Plus라는 새로운 단일 이미지 디레이닝 방법을 제안하며, 이는 Restormer 아키텍처를 기반으로 하며 GT-RAIN 챌린지에서 뛰어난 성능을 달성했습니다. Restormer-Plus는 단일 이미지 디레이닝 모듈, 중간 필터링 모듈, 가중 평균 모듈 및 후처리 모듈의 네 가지 주요 모듈로 구성됩니다. 저자는 각 모듈의 기능과 상호 작용을 자세히 설명하고 Restormer-Plus가 Restormer의 세 가지 주요 단점, 즉 빗물 이미지의 디테일 및 텍스처 복원 능력 저하, 훈련 중 과적합 취약성, 빗물 이미지의 밝기 복원 미흡을 해결하는 방법을 보여줍니다.
Özet
Restormer-Plus: 실제 이미지 디레이닝을 위한 향상된 Restormer 접근 방식
본 연구 논문에서는 CVPR 2023 UG2+ Track 3의 GT-RAIN 챌린지에 제출된 Restormer-Plus라는 새로운 단일 이미지 디레이닝 방법을 소개합니다. Restormer-Plus는 Restormer 아키텍처를 기반으로 하며, 챌린지에서 최첨단 성능을 달성하여 PSNR(Peak Signal-to-Noise Ratio)에서 1위, SSIM(Structural Similarity)에서 4위를 기록했습니다.
Restormer-Plus는 다음 네 가지 핵심 모듈로 구성됩니다.
1. 단일 이미지 디레이닝 모듈 (Restormer-X)
이 모듈은 Restormer의 두 가지 버전, 즉 기본 버전(Restormer)과 수정된 버전(Restormer†)을 활용합니다. Restormer†는 출력 레이어를 수정하여 Restormer의 표현 능력을 향상시켜 어려운 장면에 더 적합합니다. Restormer-X는 각 장면에 대해 두 버전 중 하나를 선택적으로 적용하여 성능을 최적화합니다.
2. 중간 필터링 모듈
이 모듈은 시간적 중간 필터링을 적용하여 시간이 지남에 따라 서로 다른 공간적 위치에 나타나는 빗방울을 제거하는 동시에 장면 배경을 유지합니다.
3. 가중 평균 모듈
과적합을 완화하기 위해 Restormer-X의 출력과 중간 필터링 모듈의 출력을 가중 평균합니다.
4. 후처리 모듈
복구된 이미지의 밝기를 조정하기 위해 선형 변환을 사용하여 후처리를 수행합니다.
저자는 GT-Rain 데이터 세트에 대한 광범위한 실험을 통해 Restormer-Plus의 효과를 입증했습니다. 결과는 Restormer-Plus가 최첨단 성능을 달성했으며 각 모듈이 전반적인 성능 향상에 기여했음을 보여줍니다.