AGV DL Segmentation Pipeline

Structure

├── config  
│   ├── vanilla_enet.json
├── data
├── dataloader
│   ├── __init__.py
│   ├── cityscapes.py
├── models
│   ├── __init__.py
│   ├── enet.py
├── utils
│   ├── __init__.py
│   ├── augmentation.py
│   ├── config.py
│   ├── device.py
│   ├── losses.py
│   ├── metric.py
│   ├── preprocessing.py
│   ├── saving.py
│   ├── scripts.py
│   ├── tester.py
│   ├── trainer.py
│   ├── wandb_utils.py
├── __init__.py
├── README.md
├── run.py
├── test.py
├── train.py              

Usage

python3 run.py [-h] [--config CONFIG_FILE] [--mode MODE]
                    [--wandb_id "WANDB_API_KEY"]