anthonyweidai / SvANet

Official implementation of "Exploiting Scale-Variant Attention for Segmenting Small Medical Objects"
MIT License
37 stars 2 forks source link

inference masks #8

Closed GOfBlue closed 1 month ago

GOfBlue commented 1 month ago

During inference, only files such as "tiny metrics avg.csv" will be generated. Is there any relevant code for generating inference masks that can be provided? There is a display of inference masks in the paper. I am very grateful to the author for your work.Hope you have more achievements!

anthonyweidai commented 1 month ago

Please refer to https://pytorch.org/vision/main/auto_examples/others/plot_visualization_utils.html.