royerlab / ultrack

Cell tracking and segmentation software
https://royerlab.github.io/ultrack
BSD 3-Clause "New" or "Revised" License
64 stars 7 forks source link

warning during edge detection #70

Open tischi opened 4 months ago

tischi commented 4 months ago

Hi,

I am getting this:

/g/cba/exchange/erk-signalling-dynamics/code/python/ultrack_tracking.py:49: RuntimeWarning: invalid value encountered in divide
  dist = dist / dist.max(axis=(1, 2, 3), keepdims=True)

Within this function:

def _edt(mask: np.ndarray, voxel_size=(3, 1, 1)) -> np.ndarray:
    dist = np.stack([edt.edt(m, voxel_size) for m in mask])
    dist = dist / dist.max(axis=(1, 2, 3), keepdims=True)
    dist = 1.0 - dist
    dist[~mask] = 1
    return dist

Any ideas?