NVIDIA / semantic-segmentation

Nvidia Semantic Segmentation monorepo
BSD 3-Clause "New" or "Revised" License
1.76k stars 388 forks source link

Changing the number of channels #145

Closed AxMM closed 3 years ago

AxMM commented 3 years ago

Hello! First of all thanks for this amazing work. I am trying to use fine tuning in your framework. I want to use a reduced size dataset (<1k), from a different domain and grayscale images. I already create my data loader (similar to your cityscapes) where I create labelids masks with 2 classes (out of roi, roi, class1 and class2). I want to use your framework for classification and segmentation tasks.