steps_batch#

omnipose.core.steps_batch(p, dP, niter, omni=True, suppress=True, interp=True, calc_trace=False, calc_bd=False, verbose=False)[source]#

Euler integration of pixel locations p subject to flow dP for niter steps in N dimensions.

Parameters

  • p (float32, tensor) -- pixel locations [axis x Lz x Ly x Lx] (start at initial meshgrid)

  • dP (float32, ND array) -- flows [axis x Lz x Ly x Lx]

  • niter (int32) -- number of iterations of dynamics to run

Returns

p -- final locations of each pixel after dynamics

Return type

float32, ND array