/semantic-segmentation-for-recycling-trash-data

semantic-segmentation-level2-cv-16 created by GitHub Classroom

Primary LanguageJupyter Notebook

Passion-ateπŸ”₯

κ°•μž¬ν˜„ κΉ€λ―Όμ€€ λ°•μƒν˜„ μ„œκ΄‘μ±„ μ˜€ν•˜μ€ 이승우
image image image image image image
Blog Blog Notion Notion

python pytorch ubuntu git github mmcv mmsegmentation smp

Overview

ν”„λ‘œμ νŠΈ λͺ©ν‘œ

  • μž¬ν™œμš© image semantic segmentation λͺ¨λΈ 개발

데이터셋

  • COCO format 이미지 3272μž₯ (512x512 resolution)
    • Class : Background, General trash, Paper, Paper pack, Metal, Glass, Plastic, Styrofoam, Plastic bag, Battery, Clothing
    • Tool: Pytorch, mmsegmentation, Weights and Biases, Github, Notion

Result

  • Public & Private LB 1λ“±: 0. 773(19 teams) leaderboard

Directory

semantic-segmentation-level2-cv-16/  
β”œβ”€β”€ dev/  
└── release/  
    β”œβ”€β”€ data_processing/  
    |   β”œβ”€β”€ EDA_vis_tools/   
    |   β”œβ”€β”€ Passion-ateMix/   
    |   β”œβ”€β”€ Pseudo_labeling/   
    |   β”œβ”€β”€ ensemble/   
    |   └── post_process_vis/   
    └── segmentation_model/  
        β”œβ”€β”€ BEiT/  
        β”œβ”€β”€ HRNetV2_W64_OCR/  
        β”œβ”€β”€ SMP/  
        β”œβ”€β”€ SwinB-UperNet/  
        └── UNet/



Model

BEiT

BEiT

SwinTransformer

SwinB-UperNet

HRNetV2

HRNetV2_W64_OCR

UNet++

SMP

Ensemble

hard_voting.ipynb