level2_semanticsegmentation_cv-level2-cv-07

팀원

블랙박스
🙈 김기용	🐒 김성수	🙉 김주엽	🙊 이 구	🐵 이태희

• 김기용_T4020: model, augmentation, DiceTopk loss, Weighted CE 실험
• 김성수_T4039: MMsegmentation baseline 분석, 모델 조사, augmentation, Combo loss, Weighted CE 실험
• 김주엽_T4048: Git 브랜치 템플릿 구성, SMP 템플릿 작성, augmentation 실험
• 이 구_T4145: model, augmentation, Focal & Dice loss, Weighted CE 실험, Model Soups 적용 및 디버깅
• 이태희_T4172: Confusion Matrix 코드, model, augmentation, Focal & Tversky loss, Weighted CE 실험

프로젝트 개요

바야흐로 대량 생산, 대량 소비의 시대. 우리는 많은 물건이 대량으로 생산되고, 소비되는 시대를 살고 있다. 하지만 이러한 문화는 '쓰레기 대란', '매립지 부족'과 같은 여러 사회 문제를 낳고 있다.

분리수거는 이러한 환경 부담을 줄일 수 있는 방법 중 하나이다. 잘 분리배출 된 쓰레기는 자원으로서 가치를 인정받아 재활용되지만, 잘못 분리배출 되면 그대로 폐기물로 분류되어 매립 또는 소각되기 때문이다.

따라서 우리는 사진에서 쓰레기를 Segmentation 하는 모델을 만들어 이러한 문제점을 해결해보고자 한다. 문제 해결을 위한 데이터셋으로는 일반 쓰레기, 플라스틱, 종이, 유리 등 10 종류의 쓰레기가 찍힌 사진 데이터셋을 사용한다.

이를 이용하여 학습 시킨 모델은 쓰레기장에 설치되어 정확한 분리수거를 돕거나, 어린아이들의 분리수거 교육 등에 사용될 수 있을 것이다.

Dataset

우리는 수많은 쓰레기를 배출하면서 지구의 환경파괴, 야생동물의 생계 위협 등 여러 문제를 겪고 있습니다. 이러한 문제는 쓰레기를 줍는 드론, 쓰레기 배출 방지 비디오 감시, 인간의 쓰레기 분류를 돕는 AR 기술과 같은 여러 기술을 통해서 조금이나마 개선이 가능합니다.

제공되는 이 데이터셋은 위의 기술을 뒷받침하는 쓰레기를 판별하는 모델을 학습할 수 있게 해줍니다.

• 11 Class : Background, General trash, Paper, Paper pack, Metal, Glass, Plastic, Styrofoam, Plastic bag, Battery, Clothing
• images:
  • id: 파일 안에서 image 고유 id, ex) 1
  • height: 512
  • width: 512
  • filename: ex) batch01_vt/002.jpg
• annotations:
  • id: 파일 안에 annotation 고유 id, ex) 1
  • segmentation: masking 되어 있는 고유의 좌표
  • category_id: 객체가 해당하는 class의 id
  • image_id: annotation이 표시된 이미지 고유 id

프로젝트 환경

모든 실험은 아래의 환경에서 진행되었다.

• Ubuntu 18.04.5 LTS
• Intel(R) Xeon(R) Gold 5120 CPU @ 2.20GHz
• NVIDIA Tesla V100-SXM2-32GB

프로젝트 수행 일정

• Course & Mission (12.19-12.21) : 강의 수강 및 Special mission 진행
• Model Selecting (12.22-12.27) : 다양한 모델들 중, 최종적으로 사용할 모델 선정
• 새로고침 데이 (12.28-01.01)
• Model Training & Development (01.02-01.04) : 선정한 모델들의 성능 개선을 위한 다양한 방법 적용
• Ensemble (01.05) : 다양한 방법으로 성능을 향상시킨 모델들 간의 ensemble 진행

Wrap-Up Report

Result

• Public LB mIoU: 74.05
• Private LB mIoU : 72.64
• Rank : 13/19