카메라 이미지 품질 향상 AI 경진대회

알고리즘 | CV | 이미지 변환 | 제어 | PSNR

상금 : 총 1,000만원
1,285명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

albumentations 을 이용한 패치 추출

JGHuh

2021.07.30 17:17 4,186 조회 language

학습데이터를 증가시키려는 목적으로 albumentations 의 rotation 을 이용해서 patch generation 했습니다.
horizontal, vertical flip 사용을 가정하여 0~45도까지 5도 단위로 transform 하도록 구현했고, standard deviation (9)를 이용해서 flat 한 영역을 제거했습니다.
또한 rotation 해서 발생하는 blank 영역(회전해서 픽셀이 없는 영역) 을 10 픽셀 단위로 탐색하여 patch를 최대한 추출하도록 했습니다.

코드