jeya-maria-jose / Medical-Transformer

Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021
MIT License
791 stars 176 forks source link

pred images are black #73

Open scwuchung opened 2 years ago

scwuchung commented 2 years ago

i found that the output tensor are <0,so that after tmp[tmp>=0.5] = 1 ,tmp[tmp<0.5] = 0,the saved mask are black..can you give some advice? thanks