lloongx / SFDA-CBMT

Pytorch implementation of MICCAI'23 pape: Source-Free Domain Adaptive Fundus Image Segmentation with Class-Balanced Mean Teacher.
17 stars 0 forks source link

Visualization result implementation #8

Closed H-CODE6 closed 2 months ago

H-CODE6 commented 2 months ago

Dear author, thank you very much for opening up this project. We are very interested in it and have conducted experiments to reproduce it. However, we found that no visualization results were generated in the end. Could you please open source this part of the code in your spare time for us to further learn and refer to? Thank you very much, looking forward to your reply!

lloongx commented 2 months ago

Visualization of segmentation results can be easily generated by torchvision.utils.save_image from model output tensors. You can have a try.

H-CODE6 commented 2 months ago

Thank you very much for your reply. Would it be convenient for you to synchronize and update the source code or operations for visualizing this part? This can help us to have a deeper understanding of learning. Looking forward to your reply

lloongx commented 2 months ago

I didn't keep the original code in my current PC. I think the visualization operation is quite basic. You can even ask ChatGPT to do this. So please do it yourself :)