pytorch-UNet icon indicating copy to clipboard operation
pytorch-UNet copied to clipboard

2D and 3D UNet implementation in PyTorch.

Results 12 pytorch-UNet issues
Sort by recently updated
recently updated
newest added

Hello, I got val_loss, f1 and train_loss by using my own data training volume networkIs it convenient to provide your training results?I want to make a reference.

Softmax was being applied by default to the output of the Last2D block, and the `softmax` argument in the forward method was unused. Fixed it to use the argument to...