월간 데이콘 위성 영상을 활용한 북극 해빙 예측 AI 경진대회

알고리즘 | CV | Semantic segmentation | 환경 | MAE & F1

상금 : 100만원
901명 마감

연습

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

제출

Large size image 처리 : Encoder <----> Decoder

IMJI

2021.03.09 00:37 8,845 조회 language

저처럼 무작정 공유 코드를 따라하는 분들에게 코드 공부를 하는데 조금 도움이 되었으면 하는 마음에 올려봅니다.
이전에 공유해주신 코드를 보면 Encoder 와 Decoder를 사용하셨는데 아직 초보라서 그게 왜 꼭 필요한 것인지 몰랐거든요.
원래 그대로 데이터를 사용하여 ConvLSTM2D를 활용한 모델에 학습을 시도했는데
데이터의 사이즈가 너무 커서 OOM 문제가 발생하면서 이유를 알았어요.

글 올리기 전 먼저 올려주신 코드를 보니 Encoder + Decoder 방식을 사용하지 않고 이미지를 분할하여 학습에 시킬 수도 있네요!!

코드