o ¥&?eÞã@sjddlZddlmZddlmZmZGdd„dƒZGdd„dƒZ dd d „Z dd„Z dd„Z dd„Z dS)éNé)Ú img_as_float)Ú_supported_float_typeÚcheck_nDc@óeZdZdd„Zdd„ZdS)ÚFeatureDetectorcCót g¡|_dS©N)ÚnpÚarrayZ keypoints_©Úself©rúU/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/skimage/feature/util.pyÚ__init__ ózFeatureDetector.__init__cCótƒ‚)z}Detect keypoints in image. Parameters ---------- image : 2D array Input image. ©ÚNotImplementedError)r ÚimagerrrÚdetect s zFeatureDetector.detectN)Ú__name__Ú __module__Ú __qualname__rrrrrrró rc@r)ÚDescriptorExtractorcCrr )r r Z descriptors_r rrrrrzDescriptorExtractor.__init__cCr)zïExtract feature descriptors in image for given keypoints. Parameters ---------- image : 2D array Input image. keypoints : (N, 2) array Keypoint locations as ``(row, col)``. r)r rÚ keypointsrrrÚextracts zDescriptorExtractor.extractN)rrrrrrrrrrrrÚkFÚ horizontalc CsØt|ƒ}t|ƒ}t|jƒ} t|jƒ} |jd|jdkr$|jd| d<n|jd|jdkr5|jd| d<|jd|jdkrG|jd| d<n|jd|jdkrX|jd| d<| |jkrwtj| |jd} || d|jd…d|jd…f<| }| |jkr–tj| |jd} || d|jd…d|jd…f<| }t |j¡}| dkr®tj||gdd}d|d<n| dkrÀtj||gdd}d|d<n d| ›d }t|ƒ‚|sþ|j |dd…df|dd…dfd |d |j |dd…df|d|dd…df|dd |d |j |d d |  d|jd|d|jd|ddf¡tj   ¡}t|jdƒD]A}||df}||df}|durA|  d¡}n|}|j||df||df|df||df||df|dfd|dq(dS)a¼Plot matched features. Parameters ---------- ax : matplotlib.axes.Axes Matches and image are drawn in this ax. image1 : (N, M [, 3]) array First grayscale or color image. image2 : (N, M [, 3]) array Second grayscale or color image. keypoints1 : (K1, 2) array First keypoint coordinates as ``(row, col)``. keypoints2 : (K2, 2) array Second keypoint coordinates as ``(row, col)``. matches : (Q, 2) array Indices of corresponding matches in first and second set of descriptors, where ``matches[:, 0]`` denote the indices in the first and ``matches[:, 1]`` the indices in the second set of descriptors. keypoints_color : matplotlib color, optional Color for keypoint locations. matches_color : matplotlib color, optional Color for lines which connect keypoint matches. By default the color is chosen randomly. only_matches : bool, optional Whether to only plot matches and not plot the keypoint locations. alignment : {'horizontal', 'vertical'}, optional Whether to show images side by side, ``'horizontal'``, or one above the other, ``'vertical'``. ré)ÚdtypeNr)ÚaxisÚverticalzKplot_matches accepts either 'horizontal' or 'vertical' for alignment, but 'zu' was given. See https://scikit-image.org/docs/dev/api/skimage.feature.html#skimage.feature.plot_matches for details.Únone)Z facecolorsZ edgecolorsÚgray)Zcmapéú-)Úcolor)rÚlistÚshaper Zzerosr!r Z concatenateÚ ValueErrorZscatterZimshowr"ÚrandomZ default_rngÚrangeZplot)ÚaxZimage1Zimage2Z keypoints1Z keypoints2ÚmatchesZkeypoints_colorZ matches_colorZ only_matchesZ alignmentZ new_shape1Z new_shape2Z new_image1Z new_image2ÚoffsetrZmesgÚrngÚiZidx1Zidx2r(rrrÚ plot_matches+sh!         ÿ ÿ0ÿ.     " þ÷r3cCs4t |¡}t|dƒt|ƒ}t|jƒ}|j|ddS)NrF©Úcopy)r Úsqueezerrrr!Úastype©rZ float_dtyperrrÚ_prepare_grayscale_input_2Ds   r9cCs:t |¡}t|tddƒƒt|ƒ}t|jƒ}|j|ddS)NréFr4)r r6rr-rrr!r7r8rrrÚ_prepare_grayscale_input_nD•s  r;cCs||d}|d}|d|dd…dfk|dd…df||dk@|d|dd…dfk@|dd…df||dk@}|S)aòMask coordinates that are within certain distance from the image border. Parameters ---------- image_shape : (2, ) array_like Shape of the image as ``(rows, cols)``. keypoints : (N, 2) array Keypoint coordinates as ``(rows, cols)``. distance : int Image border distance. Returns ------- mask : (N, ) bool array Mask indicating if pixels are within the image (``True``) or in the border region of the image (``False``). rr Nr)Z image_shaperZdistanceÚrowsÚcolsÚmaskrrrÚ_mask_border_keypointssÿþýr?)rNFr) Únumpyr ÚutilrZ _shared.utilsrrrrr3r9r;r?rrrrÚs  þb