twni2016 / OrganSegRSTN_PyTorch

PyTorch implementation of OrganSegRSTN - CVPR 2018
http://ccvl.jhu.edu/
MIT License
107 stars 33 forks source link

multi-organ segmentation supporting #8

Open shiyu2011 opened 4 years ago

shiyu2011 commented 4 years ago

Is this course-fine model supporting multi-organ segmentation simultaneously?

twni2016 commented 4 years ago

Sorry, not.

shiyu2011 commented 4 years ago

Thanks @twni2016 for sharing this good work. To support multi-organ segmentation simultaneously, could I apply saliency transformation over each organ, crop it and pass it through fine_model and return the probability map belongs to each class for joint optimization? please advise.

twni2016 commented 4 years ago

Yes, you may try this idea to implement multi-organ segmentation. I would suggest a global coarse seg model to output the prob maps of multiple organs, and crop each channel and feed them into the corresponding fine seg models.