Some error in predict~~
Opened this issue · 1 comments
WFLiu0327 commented
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
mask = output.cpu().detach().numpy()[0][0] * 255