• 분리수거는 환경부담을 줄일 수 있는 방법 중 하나이므로 사진에서 쓰레기를 Detection 하는 모델을 만들어 일반 쓰레기, 플라스틱, 종이 등 10종류의 쓰레기를 분리할 수 있는 모델 설계

• 전체 이미지 개수 : 9754장
• 10 class : General trash, Paper, Paper pack, Metal, Glass, Plastic, Styrofoam, Plastic bag, Battery, Clothing
• 이미지 크기 : (1024, 1024)

• 이미지 내 bbox의 개수, size, aspect ratio, class 분포, 중심점 좌표 위치, 통계치 등을 분석
• 이미지 별 bbox 개수 및 크기 상이, class 불균형을 인지
• image data를 bbox, class와 함께 시각화함으로써 불필요한 bbox 를 인식
• 이미지들의 평균값, 중간값을 이용하여 Anchor box 비율 조정
• EDA를 통한 데이터 기반 의사결정

Faster RCNN baseline/ Adam optimizer/ lr 1e-4

• RandomRotate90, RandomResizedCrop, CLAHE, GaussNoise는 첫번째 실험에서 성능향상을 보여서 후보군에 포함
• Mosaic과 Mixup은 epcoh을 늘리면 성능향상으로 이어질 수도 있다는 가정 하에 2x로 재실험한 결과 Mosaic은 성능향상을 보여 후보군에 포함, mixup은 최종탈락

Faster RCNN / backbone Resnext / Adam optimizer / lr1e-4

• 개별 실험에서 성능이 확실히 입증되지는 않은 기법들을 빼가면서 성능 실험을 진행
• 시각화를 통해, 이미지를 직접 눈으로 확인해가며 파라미터 값 설정
• Cutout의 경우, 작은 size의 hole을 더 많이 생성했을 때 성능 향상
• Multi scale 의 경우, 512 ~ 1024까지 64 간격의 scale로 학습 시켰을 때 성능 향상
• Mixup과 Gaussian noise를 제외하고 나머지 모든 augmentation들을 사용했을 때, 가장 성능이 좋아, 최종 모델에 적용

• State-of-the-art를 이용하여 모델 선택
• 1-stage모델로 yolov5, v7을 사용
• 2-stage모델로 mmdetection을 이용하여 사용

• confusion matrix/ yolov7

TTA

• Multiscale : 512-1024 64간격으로 진행
• Horizontal Flip

Ensemble (Weighted-Boxes-Fusion를 이용)

• Cascade Swin_large, Cascade Swin_base, Yolo V7 ensemble를 이용
• IoU threshold : 0.55
• Cascade Swin_Large가 가장 좋은 성능을 냈기에 가중치 부여