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

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

Prize : 100만원
901명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

Private 5위, score 4.43106, Seq2Seq+ ConvLSTM

SZJ

2021.04.20 14:19 9,668 Views language

안녕하세요 VisionAI 팀 입니다.

저희 팀은 NLP에서 많이 사용하는 Sequence to Sequence 모델에 ConvLSTM을 결합한 모델을 사용하였습니다.

입력으로는 해빙농도 이미지만을 사용하였고 출력 결과를 앙상블 하여 최종 결과를 만들었습니다.

3090 GPU를 이용하였으며 Jupyter notebook으로는 따로 돌려보지 않았기 때문에 메모리 부족 현상이 생길 수 있는 점 양해바랍니다.

감사합니다.

Code