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

알고리즘 | 월간 데이콘 | 비전 | 이미지 복원 | 이미지 색상화 | SSIM

  • moneyIcon 상금 : 데이스쿨 프로 구독권
  • 2024.11.12 ~ 2024.12.09 09:59 + Google Calendar
  • 766명 마감

 

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

2024.11.16 20:01 3,445 조회 language

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

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

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

코드
로그인이 필요합니다
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.해당 폴리곤의 영역(마스크)을 구함