1-Ph`ddZddlZddlmZddlmZddlm Z dZ dZ dd Z dd d ej fd ZdS)z(Common tools to optical flow algorithms.N)ndimage)pyramid_reduce)_convertc||}t|D]\}}||dfxx|z cc<|S)a>Compute warp point coordinates. Parameters ---------- grid : iterable The sparse grid to be warped (obtained using ``np.meshgrid(..., sparse=True)).``) flow : ndarray The warping motion field. Returns ------- out : ndarray The warp point coordinates. .)copy enumerate)gridflowoutidxgs h/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/registration/_optical_flow_utils.py_get_warp_pointsr sK" ))++CD//Q CH  Jcdt||jddD}tj||j}|D]}|dtjf}|t j|dg|zddd z}|S) aRescale the values of the vector field (u, v) to the desired shape. The values of the output vector field are scaled to the new resolution. Parameters ---------- flow : ndarray The motion field to be processed. shape : iterable Couple of integers representing the output shape. Returns ------- rflow : ndarray The resized and rescaled motion field. cg|] \}}||z  Sr).0nos r z _resize_flow..5s : : :tq!QU : : :rNdtype.rnearestF)ordermode prefilter)zipshapenparrayrnewaxisndizoom)r r!scale scale_factor_rflows r _resize_flowr+!s( ; :s5$*QRR.99 : : :E8E444L 55#CO4 38 qcEke E Lr@ c|g}t|j}d}||kr^|||zkrUt|d|d}||t|j}|dz }||kr |||zkU|dddS)aConstruct image pyramid. Parameters ---------- I : ndarray The image to be preprocessed (Grayscale or RGB). downscale : float The pyramid downscale factor. nlevel : int The maximum number of pyramid levels. min_size : int The minimum size for any dimension of the pyramid levels. Returns ------- pyramid : list[ndarray] The coarse to fine images pyramid. rN) channel_axis)minr!rappend)I downscalenlevelmin_sizepyramidsizecountJs r _get_pyramidr<Bs*cG qw<>y8'; ; ; 72;  E E Eq17||   6>>y8'; ; ; 44R4=rc l|j|jkrtdtj|jdvrtdt t tt|||||tt|||||}tj |ddj f|ddjz|}||dd|dd|}|ddD]%\} } || | t|| j}&|S)ayGeneric coarse to fine solver. Parameters ---------- I0 : ndarray The first grayscale image of the sequence. I1 : ndarray The second grayscale image of the sequence. solver : callable The solver applied at each pyramid level. downscale : float The pyramid downscale factor. nlevel : int The maximum number of pyramid levels. min_size : int The minimum size for any dimension of the pyramid levels. dtype : dtype Output data type. Returns ------- flow : ndarray The estimated optical flow components for each axis. z'Input images should have the same shapeefdgz8Only floating point data type are valid for optical flowrrrN) r! ValueErrorr"rcharlistr r<rzerosndimr+) I0I1solverr5r6r7rr8r J0J1s r_coarse_to_finerIds1: x28BCCC x6))VWWW "e,,i J J "e,,i J J  G 8WQZ]')GAJqM,??u M M MD 6'!*Q-A 5 5D!""+<<Bvb"l4::;; Kr)r,r-r.)__doc__numpyr"scipyrr% transformr util.dtyperrr+r<float32rIrrrrPs.. &&&&&&!!!!!!.BF Rrz222222r