이미지 색상화 및 손실 부분 복원 AI 경진대회

안녕하세요
사용했던 코드와 PPT 자료 첨부합니다.

코드는 "파이썬초보만" 님이 공유해주신 2-Stage 학습 템플릿을 참고하였습니다.
U-Net++ 모델을 활용해 손실된 영역 복원과 색상 복원을 두 단계로 나누어 처리하였으며,
전체 학습 과정을 4개의 쿼터로 나누고 각 쿼터별로 학습률과 마스크 크기를 점진적으로 조정하며 단계적으로 학습을 진행하였습니다.
이와 같은 학습 전략이 모델 성능을 높이는 데 도움이 되었던 것 같습니다.

한 달 동안 고생 많으셨습니다!

Code : https://github.com/Alexaqua7/Image_inpainting_restoration
PPT : 첨부파일