search

cosmic-cortex / pytorch-UNet

2D and 3D UNet implementation in PyTorch.
MIT License
147 stars 40 forks source link

test_dice #13

Open 15793723081 opened 4 years ago

15793723081 commented 4 years ago

After the prediction code is run, only the output image is obtained. How to use these outputs to obtain the dice coefficient? Did you calculate the dice coefficient of the prediction set?How do you calculate that?

fu123456 commented 3 years ago

I run predict.py file with my own configuration arguments, but it outputs error: NameEooor: name 'unet' is not defined. If I write "import unet', it also outputs error: model = Model(unet, checkpoint_folder=args.results_path, device=args.device) TypeError: init() missing 2 required positional arguments: 'loss' and 'optimizer'

shanpriya3 commented 3 years ago

Hi, I am getting the same error. Did you @fu123456 manage to solve this?