This is the segmentation template by Code-Generator using deeplabv3_resnet101
and cifar10
dataset from TorchVision and training is powered by PyTorch and PyTorch-Ignite.
Install the dependencies with pip
:
pip install -r requirements.txt --progress-bar off -U
|
|- README.md
|
|- main.py : main script to run
|- data.py : helper module with functions to setup input datasets and create dataloaders
|- models.py : helper module with functions to create a model or multiple models
|- trainers.py : helper module with functions to create trainer and evaluator
|- utils.py : module with various helper functions
|- vis.py : helper module for data visualizations
|
|- requirements.txt : dependencies to install with pip
|
|- config.yaml : global configuration YAML file
python main.py