Medical-Transformer
Medical-Transformer copied to clipboard
pred images are black
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