Some error in predict~~

FengheTan9 / Medical-Image-Segmentation-Benchmarks

A Pytorch implement of medical image segmentation U-shape architecture benchmarks

MIT License

72 stars 3 forks source link

Some error in predict~~ #3

Open WFLiu0327 opened 10 months ago

WFLiu0327 commented 10 months ago

We are getting error in predicting single images after training these models, the predicted images are black, can you please upload the script for predicting single images as well, thanks a lot!

FengheTan9 commented 10 months ago

mask = output.cpu().detach().numpy()[0][0] * 255