Giter Club home page Giter Club logo

object-detection's Introduction

프로젝트 개요

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

1

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

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

  • Input : 쓰레기 객체가 담긴 이미지가 모델의 인풋으로 사용된다. 또한 bbox 정보(좌표, 카테고리)는 model 학습 시 사용이 된다. bbox annotation은 COCO format으로 제공된다.
  • Output : 모델은 bbox 좌표, 카테고리, score 값을 리턴한다. 이를 submission 양식에 맞게 csv 파일을 만들어 제출한다.

평가 기준

Test set의 mAP50(Mean Average Precision)로 평가

Object Detection에서 사용하는 대표적인 성능 측정 방법.

Ground Truth 박스와 Prediction 박스간 IoU(Intersection Over Union, Detector의 정확도를 평가하는 지표)가 50이 넘는 예측에 대해 True라고 판단함.

2


프로젝트 팀 구성 및 역할

송영동: 1-stage model

정상헌: Backbone, 1-Stage model, 2-Stage model

최휘준: Backbone, Neck, Scheduler

김형훈: Validation-set search, Preprocessing, Pseudo labeling

김동영 : EDA, Fine-tuning, Postprocessing


프로젝트 수행 과정 및 결과

Model

Detector

1-Stage 실험

2-Stage 실험

Anchor Generator 실험

YOLOv7 실험

Backbone

Faster R-CNN Backbone 실험

Cascade R-CNN Backbone 실험

Backbone 모델 : swin-T, swin-L에 따른 성능 비교 실험

Neck

512사이즈 Neck 실험

1024사이즈 Neck 실험

Preprocessing

Basic Augmentation 실험

Advanced Augmentation 실험

Mixup Augmentation 실험

Postprocessing

Pseudo Detection 실험

NMS 실험

Optimizer

Optimizer 실험

Scheduler

Scheduler 실험

Validation-set Search

Validation-set Search 실험

Pseudo Labeling

Pseudo Labeling 실험

object-detection's People

Contributors

brotherhoon-code avatar qwer55252 avatar ydsong2 avatar chjcode avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.