Image Colorization and Inpainting AI Challenge

Algorithm | Montly Dacon | Vision | Image Inpainting | Image Colorization | SSIM

  • moneyIcon Prize : DASCHOOL Pro Subscription
  • 2024.11.12 ~ 2024.12.09 09:59 + Google Calendar
  • 770 Users Completed

 

public 0.50486) 2 stage U-net 학습 템플릿

2024.11.16 20:01 3,508 Views language

코드는 다음을 포함하고 있습니다.

1. 이미지 임베딩을 통한 train-valid셋 구성
2. 랜덤 polygon 마스킹
3. grayscale mask predict -> image coloring
4. 평가 산식과 동일한 방식의 validation

다양한 모델/손실함수, augmentation, ensemble등은 포함되어있지 않습니다.
이를 변경 하거나 추가하면서 실험하셔도 좋을 것 같습니다.

Code
Login Required
0 / 1000
도비콘
2024.11.16 20:51

주말에도 화이팅입니다.👍👍

파이썬초보만
2024.11.16 21:11

감사합니다!

co1dtype
2024.11.17 00:57

저도 데이터셋 보자마자 train/valid 구축 전략이 저게 맞다고 생각했는데 코드로 보니 반갑네요. 잘 읽었습니다!

파이썬초보만
2024.11.17 10:44

co1dtype님 안녕하세요. 잘 읽어주셔서 감사합니다 

DoorWH
2024.12.02 02:38

마스크를 어떻게 구성하신건가요? 방법을 딱 모르겠네요ㅠ

파이썬초보만
2024.12.02 09:29

DoorWH님 안녕하세요.
마스크생성은 get_input_image 함수에서 확인할 수 있습니다.
1. 랜덤으로 박스영역을 생성 -> 2. 해당박스안에서 랜덤 폴리곤 생성 -> 3.해당 폴리곤의 영역(마스크)을 구함