Pytorch 3dunet Versions Save

3D U-Net model for volumetric semantic segmentation written in pytorch

1.8.7

2 weeks ago

1.8.5

3 weeks ago

1.8.4

2 months ago

1.8.3

4 months ago
  • improve training with PixelWiseCrossEntropyLoss
  • add examples for training with PixelWiseCrossEntropyLoss

1.8.2

4 months ago

Allow arbitrary patch sizes to be passed to ResidualUNet

1.8.1

4 months ago

Fix a bug related to --loaders.test.file_paths optional argument

1.8.0

4 months ago

Fixes an issue during a config loading

1.7.0

5 months ago
  • Override config parameters from command-line in prediction
  • Move tensorboard files to separate subfolders in CHECKPOINT_DIR
  • Allow other optimizers to be used apart from Adam

1.3.4

2 years ago