ó  ‰\c@s9ddlZddlZddlmZddlmZddlmZm Z ddlm Z ddlm Z ddlm Z ddl mZdd lmZdd l mZd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zejddefdefdefgƒd„ƒZd„ZdS(iÿÿÿÿN(tndimage(tcolor(tdatatdata_dir(tfeature(t img_as_float(tdraw(tassert_almost_equal(ttesting(texpected_warningsc Csttjƒdd…dd…fjddƒƒ}tj|dddd ddd d ƒ}t|ƒdddks{t‚dS(Nitaxisit orientationsi tpixels_per_cellitcells_per_blockit block_normtL1i(ii(iii i@(RRt astronauttmeanRthogtlentAssertionError(timgtfd((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyttest_hog_output_sizes4cCs}tjtjƒƒ}tjtjjt dƒƒ}t j |dddd dddd d t d t d t ƒ}t||ƒdS(Nsastronaut_GRAY_hog_L1.npyR i R iR iRRtfeature_vectorttransform_sqrtt visualize(ii(ii(Rtrgb2grayRRtnptloadtostpathtjoinRRRtTruetFalseR(Rtcorrect_outputtoutput((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyt#test_hog_output_correctness_l1_norms   cCs}tjtjƒƒ}tjtjjt dƒƒ}t j |dddd dddd d t d t d t ƒ}t||ƒdS(Nsastronaut_GRAY_hog_L2-Hys.npyR i R iR iRsL2-HysRRR(ii(ii(RRRRRRRRR RRRR!R"R(RR#R$((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyt&test_hog_output_correctness_l2hys_norm#s   c Csmtjƒdd…dd…f}tj|dddd dd d d ƒ}t|ƒdd dksit‚dS(Ni–iÈR i R iR iRR(ii(iiii(RtcameraRRRR(timageR((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyt&test_hog_image_size_cell_size_mismatch/s"cCsTd}}tj||fddƒ}d||d)xtdƒD] }tj||ƒ}|jdƒ}tj|ddd dd dd tdtddƒ\}}tj|ddd d d d!d tdtddƒ\}} tj|ddd d"d d#d tdtddƒ\} } tj|ddd d$d d%d tdtddƒ\} } trŠddl j }|j ƒ|j ddd ƒ|j |ƒ|jƒ|jdƒ|j dddƒ|j |ƒ|jƒ|jdƒ|j dddƒ|j | ƒ|jƒ|jdƒ|j dddƒ|j | ƒ|jƒ|jdƒ|j dddƒ|j | ƒ|jƒ|jdƒ|jƒnt||ƒt|| ƒt|| ddƒt|| ddƒtj|jddƒddƒ}|d&kr ddd dg}n-|d'kr-d dddg}n tdƒ‚t||ddƒq@WdS((Ni#tdtypetfloatidiituint8R R iR iRRRRiÿÿÿÿiR(s$HOG result visualisation (float img)is$HOG result visualisation (uint8 img)s5HOG result (transform_sqrt) visualisation (float img)is5HOG result (transform_sqrt) visualisation (uint8 img)tdecimalR is+Result is not determined for this rotation.(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(Rtzerostrangetrot90tastypeRRR!R"tmatplotlib.pyplottpyplottfiguretsubplottimshowtcolorbarttitletshowRtmaxtreshapet Exception(twidththeighttimage0trott image_floatt image_uint8t hog_floatt hog_img_floatt hog_uint8t hog_img_uint8thog_float_normthog_img_float_normthog_uint8_normthog_img_uint8_normtplttactualtdesired((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyt*test_hog_basic_orientations_and_data_types6sr                    !   c Cs¸d}}tj||fƒ}tjt|dƒt|dƒt|dƒƒ\}}d|||fR(trrtccR Rthog_imgRKt hog_matrixRLRM((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyttest_hog_orientations_circle—s8 9         c Csõd}}tj||fddƒ}d||d)tj|ddd||fd dd td d ƒ\}}|d}gt|ƒD]}||krƒ|^qƒ}||d d …fdkjƒsÉt‚||dd…fdkjƒsñt‚d S(szTest that the visualization produces a line with correct orientation The hog visualization is expected to draw line segments perpendicular to the midpoints of orientation bins. This example verifies that when orientations=3 and the gradient is entirely in the middle bin (bisected by the y-axis), the line segment drawn by the visualization is horizontal. i R*R+iiR iR R RRRNiiÿÿÿÿ(ii(RR.RRR!R/tallR(R=R>R(t_t hog_imaget middle_indextxtindices_excluding_middle((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyt"test_hog_visualization_orientationÌs   +(cCs<tjdƒ}tjtƒtj|ddƒWdQXdS(NiRtLinf(RteyeRtraisest ValueErrorRR(R((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyt,test_hog_block_normalization_incorrect_errorêssshape,multichannelic CsBtj|ƒ}tjtƒtj|d|ddƒWdQXdS(Nt multichannelRR(RR.RRaRbRR(tshapeRdR((s=lib/python2.7/site-packages/skimage/feature/tests/test_hog.pyttest_hog_incorrect_dimensionsðscCs•tjƒ}d|dd…dd…dfs.    a 5