SegmentationBLWX / sssegmentation

SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch.
https://sssegmentation.readthedocs.io/en/latest/
Apache License 2.0
775 stars 107 forks source link

how to train other datasets #20

Closed ASguoyue closed 2 years ago

ASguoyue commented 2 years ago

train the datasets by mask and imgs . how to finish it

CharlesPikachu commented 2 years ago

First you need to add the py file (e.g., yourdatasetname.py) in ./ssseg/modules/datasets according to your dataset format, and then add it in the corresponding builder.py.

You can refer to ./ssseg/modules/datasets/ade20k.py to write your own dataset class.

shituo123456 commented 2 years ago

ValueError: Error initializing torch.distributed using env:// rendezvous: environment variable RANK expected, but not set

how to deal with it