atria_segmentation_2018 icon indicating copy to clipboard operation
atria_segmentation_2018 copied to clipboard

predict_nrrd.py do error

Open wangchenguang123 opened this issue 1 year ago • 2 comments

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

wangchenguang123 avatar Oct 06 '23 12:10 wangchenguang123

@cherise215

wangchenguang123 avatar Oct 06 '23 12:10 wangchenguang123

Can you check your input dim? thanks.

cherise215 avatar Oct 11 '23 13:10 cherise215