search

ELEKTRONN / elektronn3

A PyTorch-based library for working with 3D and 2D convolutional neural networks, with focus on semantic segmentation of volumetric biomedical image data
MIT License
160 stars 27 forks source link

noise to zero_like, not empty_like #43

Closed riegerfr closed 4 years ago

riegerfr commented 4 years ago

https://github.com/ELEKTRONN/elektronn3/blob/2298c30a1cf23e455732e522c6f3293b4cd96164/elektronn3/data/transforms/transforms.py#L668

noise should be np.zeros_like instead of np.empty_like to add 0 noise (not undefined) for channels which should not be modified.

Similar: https://github.com/ELEKTRONN/elektronn3/blob/2298c30a1cf23e455732e522c6f3293b4cd96164/elektronn3/data/transforms/transforms.py#L884