pytorch-3dunet icon indicating copy to clipboard operation
pytorch-3dunet copied to clipboard

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

Results 4 pytorch-3dunet issues
Sort by recently updated
recently updated
newest added

This PR implements a multicalss/mzltichannel aware `Normalize` transform functionality where `channelwise: true` can be specified so that the Normalize transform is applied to each channel seperately. In addition, `min_value` and...

Hi @wolny I came across this issue where the predictions are overwritten if multiple thresholds are used. A simple intermediate renaming should fix.

Remove redundant final_activation layer in the forward() method of AbstractUNet. The layer is redundant because it is only applied during validation, however the loss function of segmentation models already applies...