10일 전

SDWNet: 이미지 흐림 제거를 위한 웨이블릿 변환을 갖춘 직선 확장 네트워크

Wenbin Zou, Mingchao Jiang, Yunchen Zhang, Liang Chen, Zhiyong Lu, Yi Wu
SDWNet: 이미지 흐림 제거를 위한 웨이블릿 변환을 갖춘 직선 확장 네트워크
초록

이미지 흐림 제거는 흐린 이미지로부터 선명한 이미지를 복원하는 전통적인 컴퓨터 비전 문제이다. 이 문제를 해결하기 위해 기존의 방법들은 복잡한 네트워크 구조를 설계하기 위해 인코더-디코더 아키텍처를 활용해 뛰어난 성능을 달성하고자 한다. 그러나 대부분의 이러한 방법들은 수차례의 업샘플링 및 다운샘플링 구조를 반복하여 수용 영역(Receptive Field)을 확장하는 방식을 사용하는데, 이 과정에서 텍스처 정보가 손실되는 문제가 발생한다. 또한 일부 방법은 다단계 구조를 설계함으로써 수렴의 어려움을 야기한다. 이에 따라 본 연구에서는 확장된 컨볼루션(dilated convolution)을 활용하여 높은 공간 해상도를 유지하면서도 넓은 수용 영역을 확보하는 방식을 제안한다. 다양한 수용 영역을 효과적으로 활용함으로써, 보다 우수한 성능을 달성할 수 있다. 이와 함께 업샘플링 및 다운샘플링의 횟수를 줄이고 단순한 네트워크 구조를 설계하였다. 더불어 웨이블릿 변환(Wavelet Transform)을 활용한 새로운 모듈을 제안하여, 네트워크가 고주파 텍스처 세부 정보를 효과적으로 복원할 수 있도록 했다. 실제 및 합성 데이터셋에 대한 정성적 및 정량적 평가 결과, 제안하는 흐림 제거 방법은 기존 알고리즘과 비견되는 성능을 보이면서도 훈련 요구 사항이 훨씬 낮음을 입증하였다. 소스 코드 및 사전 훈련 모델은 다음 링크에서 제공된다: https://github.com/FlyEgle/SDWNet.