a6d|ddlZddlmZddlmZmZGddZGddZ dd Z d Z dZ dZ dS)N) img_as_float)_supported_float_typecheck_nDceZdZdZdZdS)FeatureDetectorc8tjg|_dSN)nparray keypoints_selfs 4lib/python3.11/site-packages/skimage/feature/util.py__init__zFeatureDetector.__init__ s(2,,ct)z}Detect keypoints in image. Parameters ---------- image : 2D array Input image. NotImplementedError)rimages rdetectzFeatureDetector.detect s"###rN)__name__ __module__ __qualname__rrrrrrs2''' $ $ $ $ $rrceZdZdZdZdS)DescriptorExtractorc8tjg|_dSr )r r descriptors_rs rrzDescriptorExtractor.__init__sHRLLrct)zExtract feature descriptors in image for given keypoints. Parameters ---------- image : 2D array Input image. keypoints : (N, 2) array Keypoint locations as ``(row, col)``. r)rr keypointss rextractzDescriptorExtractor.extracts"###rN)rrrrr"rrrrrs2))) $ $ $ $ $rrkF horizontalc t|}t|}t|j} t|j} |jd|jdkr|jd| d<n,|jd|jdkr|jd| d<|jd|jdkr|jd| d<n,|jd|jdkr|jd| d<| |jkr>tj| |j} || d|jdd|jdf<| }| |jkr>tj| |j} || d|jdd|jdf<| }tj|j}| dkrtj||gd}d|d<n9| dkrtj||gd}d|d<nd| d }t||sl| |dddf|dddfd | | |dddf|dz|dddf|dzd | | |d | d|jd|dz|jd|dzdftj }t|jdD]}||df}||df}|| d}n|}|||df||df|dzf||df||df|dzfd|dS)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) facecolors edgecolorsgray)cmap-)color)rlistshaper zerosr'r concatenate ValueErrorscatterimshowr(random default_rngrangeplot)aximage1image2 keypoints1 keypoints2matcheskeypoints_color matches_color only_matches alignment new_shape1 new_shape2 new_image1 new_image2offsetrmesgrngiidx1idx2r1s r plot_matchesrQ+sB& ! !F & ! !Ffl##Jfl##J |Aa(( Q 1 a6<? * * Q 1  |Aa(( Q 1 a6<? * * Q 1 V\!!Xj === 9? #FLO#%5fl1o%556V\!!Xj === 9? #FLO#%5fl1o%556 Xfl # #FL  /a888q j /a888q  #,    B :aaad#Z1%5$  B B B :aaad#fQi/AAAqD1AF1I1M$  B B BIIe&I!!!GGQ Q&)+V\!_vay-H! LMMM )   ! !C 7=# $ $ " "q!t}q!t}  JJqMMEE!E D!G$jq&9F1I&EFD!G$jq&9F1I&EF5  " " " " " "rctj|}t|dt|}t |j}||dS)NrFcopy)r squeezerrrr'astyper float_dtypes r_prepare_grayscale_input_2DrYsS Ju  E UA   E' 44K << %< 0 00rctj|}t|tddt |}t |j}||dS)NrFrS)r rUrr;rrr'rVrWs r_prepare_grayscale_input_nDr\s[ Ju  E UE!QKK     E' 44K << %< 0 00rc|d}|d}|dz |dddfk|dddf||z dzkz|dz |dddfkz|dddf||z dzkz}|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``). rr&Nr) image_shaper!distancerowscolsmasks r_mask_border_keypointsrcs( q>D q>Dli1o -A$/A"56 81 !!!Q$/ 1A$/A"56 8D Kr)r#NFr$) numpyr utilr _shared.utilsrrrrrQrYr\rcrrrrgs;;;;;;;;$$$$$$$$"$$$$$$$$(HM'_"_"_"_"D111111r