Skip to content

jahongir7174/DeepLab-pt

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Releases

No releases published

Packages

No packages published

Languages