- Predict Trash Objects
- Input: 1024 x 1024 Image
- Output: Object annotations
- Class(10): General trash, Paper, Paper pack, Metal, Glass, Plastic, Styrofoam, Plastic bag, Battery, Clothing
- Confidence score: 0 ~ 1
- Object coordinate: (xmin, ymin, xmax, ymax)
image-classification-level1-02/
βββ dataset/
β βββ train/ (4883 images)
β βββ test/ (4871 images)
β βββ train.json
β βββ test.json
βββ mmdetection/
β βββ configs/
β βββ mmdetection library files
β βββ train.py
β βββ inference.py
βββ util/
dataset/
: download from https://stages.ai/
cd mmdetection
python train.py
python inference.py
cd util
jupyter notebook Visualize.ipynb
set result csv in second shell
it also has Visualize_val_gt -> Visualize ground truth of train data & valid
it also has Visualize_val -> Visualize result of valid
- Ubuntu 18.04.5
- Python 3.8.5
- pytorch 1.7.1
- torchvision 0.8.2
Install packages : pip install -r requirements.txt
- CPU: 8 x Intel(R) Xeon(R) Gold 5220 CPU
- GPU: V100
- RAM: 88GB
Name @github |
---|
κ³ μ¬μ± @κ³ μ¬μ± |
κΉμ±λ―Ό @ksm0517 |
λ°μ§λ―Ό @λ°μ§λ―Ό |
λ°μ§ν @ppjh8263 |
μ¬μΈλ Ή @Seryoung Shim |
μ€νμ @Yoon Hajung |