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

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

  • moneyIcon Prize : 100만원
  • 890명 마감

 

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

2021.03.09 00:37 7,780 Views language

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

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

Code
로그인이 필요합니다
0 / 1000
당쇠
2021.03.09 07:50

IMJI님 좋은 내용 감사합니다.