i`ldZddlZddlmZddlmZddlm Z dZ dZ d d Z dd d ejfd Zy)z(Common tools to optical flow algorithms.N)ndimage)pyramid_reduce)_convertcj|j}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 /var/www/ITT-ServiceAPI-Services/ITT-Document-Scan/mrzscanner/mrzenv/lib/python3.12/site-packages/skimage/registration/_optical_flow_utils.py_get_warp_pointsr s=" ))+CD/Q CH   Jc*t||jddDcgc] \}}||z  }}}tj||j}|D]}|dtj f}|t j|dg|zdddz}|Scc}}w) 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. Ndtype.rnearestF)ordermode prefilter)zipshapenparrayrnewaxisndizoom)r rnoscale scale_factor_rflows r _resize_flowr'!s( #5$**QR.9 :tq!QU :E :88E4L 5#CO4 5 388 qcEke E L ;sB c|g}t|j}d}||krR|||zkDrJt|d|d}|j|t|j}|dz }||kr |||zkDrJ|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)minrrappend)I downscalenlevelmin_sizepyramidsizecountJs r _get_pyramidr7Bs*cG qwwy8'; ; 72;  Eq177|   6>y8'; ; 4R4=rc  |j|jk7r tdtj|jdvr tdt t tt|||||tt|||||}tj|ddjf|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 ValueErrorrrcharlistrr7rzerosndimr') I0I1solverr0r1r2rr3r J0J1s r_coarse_to_finerDds : xx288BCC xx6)VWW "e,i J "e,i J G 88WQZ]'')GAJqM,?,??u MD '!*Q-A 5D!"+<Bb"l4:;< Kr)g@r(r))__doc__numpyrscipyrr transformr util.dtyperrr'r7float32rDrrrLs9. &!.BF Rrzz2r