DeepLabV3+
DeepLabV3+ implementation for Semantic Segmentation using PyTorch
Train
- Run
python main.py
for training
Dataset structure (similar to CamVid dataset)
├── Dataset folder
├── train
├── 1111.png
├── 2222.png
├── train_labels
├── 1111_L.png
├── 2222_L.png
├── class_dict.csv
Note
- default feature extractor is EfficientNetV2-S
- changing configuration of training, change parameters in
utils/config.py
- default loss function is
weighted cross entropy