CSCYQJ / LOCATION-SENSITIVE-LOCAL-PROTOTYPE-NETWORK

For ISBI 2021 paper "A LOCATION-SENSITIVE LOCAL PROTOTYPE NETWORK FOR FEW-SHOT MEDICAL IMAGE SEGMENTATION"
9 stars 1 forks source link

Visualization code of the final output of this network #4

Closed smallkaka closed 1 year ago

smallkaka commented 1 year ago

Hello, thank you very much for your work.I have successfully run it on my own dataset, can you tell me the visualization code of the final output of this network? Thank you so much!

CSCYQJ commented 1 year ago

You can first choose a support volume and some query volumes and then run the retrieval function in eval.py to get the corresponding segmentation mask for the given organ class. After obtaining the mask, you can plot the segmentation visualizations as you want.

smallkaka commented 1 year ago

OK. Thank you very much for your reply.