0-PhddlmZddlZddlmZddlmZ ddlZdZ n #e $rdZ YnwxYwd d Z ej d dd d Z dS))productN)utils)warnTFct|}tj|r|f||z zd|zz}nM|r)t||dz krt |dz}n"t||krt dtj|r|f||z zd|zz}nM|r)t||dz krt |dz}n"t||krt dt d|Drt d|r|ddkrt d td t||DS) aReturns all combinations of shifts in n dimensions over the specified max_shifts and step sizes. Examples -------- >>> s = list(_generate_shifts(2, False, max_shifts=(1, 2), shift_steps=1)) >>> print(s) [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)] )rrz"max_shifts should have length ndimrc3"K|] }|dkV dS)rN.0ss _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/restoration/_cycle_spin.py z#_generate_shifts..'s& & &Q1q5 & & & & & &zshift_steps must all be >= 1rzGMultichannel cycle spinning should not have shifts along the last axis.c<g|]\}}td|dz|S)rr)range)r rts r z$_generate_shifts../s,QQQDAqU1a!eQ''QQQr) intnpisscalarlentuple ValueErroranyrzip)ndim multichannel max_shifts shift_stepsmcs r_generate_shiftsr$s \  B {:? ]dRi04"9< ?#j//TAX55:&&- ZD =>>> {;?"nr 2TBY> ?#k**dQh66K((4/ [  T ! !=>>> & &+ & & &&&97888 2!++ X    QQC K4P4PQQQ RRr) channel_axisc6 itj|du}tj|||}t |}t t j  fd ts||dkrd}td|dkr? |d} |ddD]} |  | z } | t|z} nC fd|D} t| t| z } | |} | S)a Cycle spinning (repeatedly apply func to shifted versions of x). Parameters ---------- x : array-like Data for input to ``func``. func : function A function to apply to circularly shifted versions of ``x``. Should take ``x`` as its first argument. Any additional arguments can be supplied via ``func_kw``. max_shifts : int or tuple If an integer, shifts in ``range(0, max_shifts+1)`` will be used along each axis of ``x``. If a tuple, ``range(0, max_shifts[i]+1)`` will be along axis i. shift_steps : int or tuple, optional The step size for the shifts applied along axis, i, are:: ``range((0, max_shifts[i]+1, shift_steps[i]))``. If an integer is provided, the same step size is used for all axes. num_workers : int or None, optional The number of parallel threads to use during cycle spinning. If set to ``None``, the full set of available cores are used. func_kw : dict, optional Additional keyword arguments to supply to ``func``. channel_axis : int or None, optional If None, the image is assumed to be a grayscale (single channel) image. Otherwise, this parameter indicates which axis of the array corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- avg_y : np.ndarray The output of ``func(x, **func_kw)`` averaged over all combinations of the specified axis shifts. Notes ----- Cycle spinning was proposed as a way to approach shift-invariance via performing several circular shifts of a shift-variant transform [1]_. For a n-level discrete wavelet transforms, one may wish to perform all shifts up to ``max_shifts = 2**n - 1``. In practice, much of the benefit can often be realized with only a small number of shifts per axis. For transforms such as the blockwise discrete cosine transform, one may wish to evaluate shifts up to the block size used by the transform. References ---------- .. [1] R.R. Coifman and D.L. Donoho. "Translation-Invariant De-Noising". Wavelets and Statistics, Lecture Notes in Statistics, vol.103. Springer, New York, 1995, pp.125-150. :DOI:`10.1007/978-1-4612-2544-7_9` Examples -------- >>> import skimage.data >>> from skimage import img_as_float >>> from skimage.restoration import denoise_tv_chambolle, cycle_spin >>> img = img_as_float(skimage.data.camera()) >>> sigma = 0.1 >>> img = img + sigma * np.random.standard_normal(img.shape) >>> denoised = cycle_spin(img, func=denoise_tv_chambolle, ... max_shifts=3) # doctest: +IGNORE_WARNINGS Nctj|}|fi}tj|td|DS)N)axisc3K|]}| VdS)Nr r s rrz5cycle_spin.._run_one_shift..s$!4!41"!4!4!4!4!4!4r)rrollr)shiftxstmpfuncfunc_kw roll_axesxs r_run_one_shiftz"cycle_spin.._run_one_shifts^ WQI . . .d2!!!!wsE!4!4e!4!4!4449EEEErrzThe optional dask dependency is not installed. The number of workers is set to 1. To silence this warning, install dask or explicitly set `num_workers=1` when calling the `cycle_spin` functionrcJg|]}tj| Sr )daskdelayed)r rr2s rrzcycle_spin..s.GGGq/4<//22GGGr) num_workers) r asanyarrayr$rlistrrdask_availablerrsumcompute)r1r.r!r"r6r/r%r all_shiftsmeanr+futuresr2r0s`` ` @@r cycle_spinr?2sp^ aAt+L!!&, KPPJj!!JeAFmm$$IFFFFFFFF  {2kAoo   5   a~jm,,^ * *E NN5)) )DD JHGGGJGGG7||c'll*|| |44 Krr )rNN) itertoolsrnumpyr_sharedr _shared.utilsrr4r9 ImportErrorr$channel_as_last_axisr?r rrrFs KKKNNNNN!S!S!S!SH   qqqqqqqqs ))