U ?h@sfddlZddlmZddlmZmZGdddZGdddZdd d Z ddZ ddZ ddZ dS)N) img_as_float)_supported_float_typecheck_nDc@seZdZddZddZdS)FeatureDetectorcCstg|_dSN)nparrayZ keypoints_selfr F/var/www/html/venv/lib/python3.8/site-packages/skimage/feature/util.py__init__ szFeatureDetector.__init__cCs tdS)z}Detect keypoints in image. Parameters ---------- image : 2D array Input image. NNotImplementedError)r imager r r detect s zFeatureDetector.detectN)__name__ __module__ __qualname__rrr r r r rsrc@seZdZddZddZdS)DescriptorExtractorcCstg|_dSr)rr Z descriptors_r r r r rszDescriptorExtractor.__init__cCs tdS)zExtract feature descriptors in image for given keypoints. Parameters ---------- image : 2D array Input image. keypoints : (N, 2) array Keypoint locations as ``(row, col)``. Nr)r r keypointsr r r extracts zDescriptorExtractor.extractN)rrrrrr r r r rsrkF horizontalc Cst|}t|}t|j} t|j} |jd|jdkrH|jd| d<n"|jd|jdkrj|jd| d<|jd|jdkr|jd| d<n"|jd|jdkr|jd| d<| |jkrtj| |jd} || d|jdd|jdf<| }| |jkr.tj| |jd} || d|jdd|jdf<| }t|j}| dkr`tj||gdd}d|d<n:| dkrtj||gdd}d|d<nd| d }t||s|j |dddf|dddfd |d |j |dddf|d|dddf|dd |d |j |d d | d|jd|d|jd|ddftj }t|jdD]}||df}||df}|dkr| d}n|}|j||df||df|df||df||df|dfd|dqXdS)aPlot 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)axisverticalzKplot_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 edgecolorsgray)Zcmap-)color)rlistshaperZzerosrr Z concatenate ValueErrorZscatterZimshowrrandomZ default_rngrangeZplot)ZaxZimage1Zimage2Z keypoints1Z keypoints2matchesZkeypoints_colorZ matches_colorZ only_matchesZ alignmentZ new_shape1Z new_shape2Z new_image1Z new_image2offsetrZmesgrngiZidx1Zidx2r#r r r plot_matches+sh!             0.     "r-cCs4t|}t|dt|}t|j}|j|ddS)NrFcopy)rsqueezerrrrastyperZ float_dtyper r r _prepare_grayscale_input_2Ds    r3cCs:t|}t|tddt|}t|j}|j|ddS)NrFr.)rr0rr(rrrr1r2r r r _prepare_grayscale_input_nDs   r5cCs||d}|d}|d|dddfk|dddf||dk@|d|dddfk@|dddf||dk@}|S)aMask 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``). rrNr )Z image_shaperZdistancerowscolsmaskr r r _mask_border_keypointssr9)rNFr) numpyrutilrZ _shared.utilsrrrrr-r3r5r9r r r r s  b