[연구 소개] 문서 이미지 그림자제거 / 문서 OCR 결과를 향상시키기 위해

요즘은 문서를 사진으로 찍어서 회사나 공공 기관에 제출하는 경우가 많다. 이 때 회사는 받은 문서에서 OCR 기술을 사용해서 텍스트를 디지털화시켜서 저장하게 된다. 그런데 휴대폰으로 문서 사진을 찍는 경우 그림자가 많이 생겨서 이미지의 퀄리티가 떨어지는 경우가 많고 이는 텍스트 인식 오류를 발생하게 한다.

 

그런데... 이미지에서 그림자를 제거하는 연구가 존재한다고 한다. 역시 세상 사람들은 참 똑똑하고 없는 게 잘 없다...

 

• Paper : BEDSR-Net A Deep Shadow Removal Network from a Single Document Image / CVPR 2020
• github : https://github.com/IsHYuhi/BEDSR-Net_A_Deep_Shadow_Removal_Network_from_a_Single_Document_Image

그림자 제거 프로젝트는 크게 자연 이미지와 문서 이미지로 나뉘는데, 본 논문은 문서에 특화된 이미지 그림자 제거 연구이다.

• 본 연구는 문서 이미지에서 비정형 형태의 그림자를 제거에 특화된 딥러닝 기반 아키텍처를 사용
• 문서의 background color 를 추정하고 그림자 위치 정보를 attention map 형태(Grad-CAM)로 찾아서 그림자 영역 제거
• Pre-train 된 모델 그대로 사용 가능

 

아래 결과들을 보면 그림자 영역을 꽤 잘 찾아서 제거해 준다. 깃허브 레퍼지토리에서 데모를 사용해볼 수 있으며 pre-trained 된 모델을 그대로 사용할 수 있다는게 큰 장점인 것 같다.