/openSurfaces-semantic-segmentation-pytorch

Material Semantic Segmentation on OpenSurfaces dataset (Pytorch)

Primary LanguagePython

Material Semantic Segmentation on opensurfaces dataset (Pytorch)

  • Dataset : OpenSurfaces
  • Task : Semantic Segmentation

image

Training

python train.py --gpus 0    # single gpu
python train.py --gpus 0,1  # multi gpu : 0,1 or 0-3, ...

Evaluation

python eval_multipro.py