a d@sddlZddlZddlmZmZddlmZddlm Z m Z m Z ddl m Z ddlmZeZe ejdgdd d Zejdgdd d Ze e d dddZddZddZdS)N)assert_almost_equal assert_equal)data)run_in_parallelxfailarch32)ORB)_convertdtype)float32float64Zuint8Zuint16int64cCstt|}tdddd}||tgd}tgd}tgd}tgd}tgd }t|tjkr|jjtjksJ|j jtjksJ|j jtjksJn6|jjtj ksJ|j jtj ksJ|j jtj ksJt ||j ddd ft ||j ddd ft ||jt ||j d t |t|j d |tt ||j ddd ft ||j ddd fdS)N 皙?) n_keypointsfast_nfast_threshold) a@[@gRj@g``@gbX9j@gP@gi@g f@rr) g0t@gt@gp= ףq@g@r@gvq@g@U@g@p@gq@ǧt@p@) rg ףp= ?rgS?rrrg333333?r) gG'JgYk(M@g͊n4XgfvbgBWg^nFgkxIgZ.d\@gTO@gS) gp/?gv?g/C?gU#K?g?gUk/?g_vj.?gjK>ރ?g ?gt(?rr)r imgrdetectnparrayr r scales responses orientationsr r keypointsrad2degdetect_and_extractr Z_imgdetector_extractorZexp_rowsZexp_colsZ exp_scalesZexp_orientationsZ exp_responser&g/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/feature/tests/test_orb.py*test_keypoints_orb_desired_no_of_keypointss2     r(cCstt|}tdddddd}||tgd}tgd}tgd}tgd }tgd }t||jddd ft||jddd ft||jt||j t|t |j d | tt||jddd ft||jddd fdS)NrgQ?)rrr downscalen_scales)rg`i@ga@g@P@gM@)gPr@rg@i@g@`@g0r@)?r-r-r-r-)gk b@glF=DsLgjSgIKMgfc)g7Ԕƿg"-D>?g܌:g㳆?gO?rr) r rrrrrrr!rrr"r r#r$r&r&r'4test_keypoints_orb_less_than_desired_no_of_keypoints;s(      r/zKnown test failure on 32-bit platforms. See links for details: https://github.com/scikit-image/scikit-image/issues/3091 https://github.com/scikit-image/scikit-image/issues/2529) conditionreasoncCs\tddd}tjgdgdgdgdgdgd gd gd gd gd gdgdgdgdgdgdgdgdgdgdgtd}|t|t|j|j|j t ||j ddddf| tt ||j ddddf|jj d}||j j dksJ||j j dks,J||jj dksBJ||jj dksXJdS)Nrr)rr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr)r dxrr)rrrboolrrextractr!rr rZ descriptorsr#shaper)r%Zexp_descriptorsZkeypoints_countr&r&r'test_descriptor_orb[sR      r8cCsHtd}t}tt||Wdn1s:0YdS)N)r9)rZonesrpytestZraises RuntimeErrorr#rr%r&r&r'!test_no_descriptors_extracted_orbs  r=cCs<tddddf}tddd}||||dS)N@r*)r+r,)rZbrickrrr#r<r&r&r'test_img_too_small_orbs  r@)numpyrr:Z numpy.testingrrZskimagerZskimage._shared.testingrrrZskimage.featurerZskimage.util.dtyper ZcoinsrmarkZ parametrizer(r/r8r=r@r&r&r&r's,    )  )