atria_segmentation_2018
atria_segmentation_2018 copied to clipboard
predict_nrrd.py do error
when i finshed training this model,i want to do some predict. but when i do this py ,i find i canot get the pred
pred = outputs.data.max(1)[1].cpu().numpy()
this code result show is empty ,so i want to ask why about it, And how can i do to solve this problem
Thank you
@cherise215
Can you check your input dim? thanks.