search

PingoLH / FCHarDNet

Fully Convolutional HarDNet for Segmentation in Pytorch
MIT License
195 stars 52 forks source link

Merge several classes into one class while training #17

Open ghost opened 4 years ago

ghost commented 4 years ago

Hello, thank you for sharing the great works.

I now stumbled on merging the classes while training. I have original CItyscapes data and during the training, I want to merge some classes into one class. Let's say class labels 2 to 10 merging to class 5 , and class 11, 12 to 15 merging to 11 for example.

Can you tell me how to implement what I want to do?

Thank you