MontaEllis / Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.
MIT License
860 stars 196 forks source link

How to train 2D images with multi-classes? #41

Closed zhongqiu1245 closed 2 years ago

zhongqiu1245 commented 2 years ago

Hello, dear author! Thank you for your amazing job! Could you tell me how to train 2D images with multi-classes (assume num_classes=5)? Where should I modify? (expect "mode = '2d', in_class = 3, out_class = 5" in hparam.py) Thank you in advance!