pytorch-UNet
pytorch-UNet copied to clipboard
cosmic-cortex
2D and 3D UNet implementation in PyTorch.
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...
`File ".../unet/model.py", line 212, in fit_dataset **val_logs, **train_logs} UnboundLocalError: local variable 'val_logs' referenced before assignment` The `val_dataset` argument to `fit_dataset` is optional. However, if none is provided, `val_logs` is not...
Hi, Thank you for your project. It's helpful. I'm tried to launch your train.py script on a new dataset and i have encounter the following error > File "src/train.py", line...
val_loss
I have a new question. In my training results, the value of val_loss is very unstable, but the result of train_loss is good. What is the reason for this? val_loss...
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...
Hello please kindly look at this error. Whenever I try to train my dataset by using train.py I keep getting this error. As I am new to pytorch a little...
**I ran:** ``` python train.py --train_dataset images_folder/ --device 'cuda:0' --epochs 10 --save_freq 1 --save_model 1 --model_name round1 --crop 2560 --learning_rate 0.0001 --checkpoint_path checkpoint ``` **Error output:** ``` Traceback (most recent...
The diagram says images_folder, but there is none. The instructions repeat itself for diff flag. I wish there was standards in docs, you can reduce the amount of stuff in...
 Hello, I have read your code, which is very helpful to me. However, I reported such an error when using my own data set. Is there any solution?Thanks a...