문서 이미지 노이즈 제거

무엇을 하는 미션인가

노이즈가 낀 문서 이미지를 입력받아 깨끗한 문서 이미지로 복원하는 오토인코더 미션입니다. 입력과 정답이 같은 크기의 이미지라, 분류와 달리 픽셀 단위로 맞춰야 합니다.

구조와 학습

Conv 인코더로 해상도를 줄이고 ConvTranspose2d로 되돌리는 대칭 구조입니다. 입력이 420×540처럼 큰 문서 이미지라 첫 Conv부터 stride=2로 크기를 줄여 나갑니다. 크기가 제각각인 원본을 맞추려고 PadToSize 전처리 클래스를 따로 만들었습니다.

손실은 MSE와 L1을 함께 정의해 사용했고, 옵티마이저는 Adam(lr=0.0005), 학습은 20 에폭입니다.

결과

노트북 출력에 남은 마지막 에폭과 검증 지표입니다.

train과 val loss가 거의 같은 값으로 끝나 과적합 징후는 보이지 않습니다. 다만 PSNR 14.45 dB는 문서 복원 기준으로 높은 값이 아니고, 노트북에도 그 이상 개선한 기록은 없습니다.

한계

• 20 에폭에서 loss가 아직 내려가는 중이라 학습을 더 돌릴 여지가 있었습니다
• 비교 대상(다른 구조·손실)이 없어 이 구조가 적절했는지는 판단할 수 없습니다
• 복원 결과 이미지는 노트북 안에만 있고 사이트로 옮기지 않았습니다