Multi-class segmentation

[xiaona123456]

Thank you for your excellent work, can this network do multi-category segmentation?

[YashSharma]

It's a great question. It can be extended to multi-category segmentation. However, the design choice will be empirical - do you need a separate network for each category, or all categories can be tackled by a single network? Here I extended it for multi-category classification - https://github.com/YashSharma/MNIST-DomainAdaptation