Closed liuyx599 closed 1 year ago
Hello, I also encountered this problem, please do you solve it?
Hello, I also encountered the same problem, could you please solve the code?
I did not use the postprocess.py; It was from the UNET++ code base upon which this code is built on.
In post_process.py
I finished the training on BUSI and no error message was reported; then I verified in _postprocess.py and an error was reported when count exceeded 5. I found that output is only within the range of if, which means that when count exceeds 5, it does not feed input into the model to get output, so when the statement runs to
output = torch.sigmoid(output).cpu().numpy()
it generates an error?