Unet for Multi-class Segmentation

Опубликовано: 13 Май 2026
на канале: AI with Sohini
13,827
224

Here is the codebase and Blog on how to modify U-net for Multi-class semantic segmentation
Blog: https://towardsdatascience.com/a-mach...

Github: https://github.com/sohiniroych/U-net-...

For beginners in Medical Image Processing here is a list of data sets and papers:
https://sites.google.com/site/sohinir...

Code for binary segmentation: https://github.com/sohiniroych/Unet-u...
Previous U-net videos:    • Image Segmentation with U-net and Keras!  
   • How to Create a Successful Github Repo for...  
#######################################################################
Debugging instructions:
1. The code https://github.com/sohiniroych/U-net-... performs BINARY SEMANTIC SEGMENTATION only.
2. The code https://github.com/sohiniroych/U-net-... performs MULTI-CLASS SEMANTIC SEGMENTATION. Please ensure ALL code is located in the same location for this to work. The model file is model_depth_3.py or model_depth_4.py.

Please see section on "How to avoid Errors in Code"
https://github.com/sohiniroych/U-net-...