def testdata_matching_affine_inliers():
import vtool.tests.dummy as dummy
import vtool as vt
scale_thresh = 2.0
xy_thresh = ut.get_argval('--xy-thresh', type_=float, default=.01)
dlen_sqrd2 = 447271.015
ori_thresh = 1.57
xy_thresh_sqrd = dlen_sqrd2 * xy_thresh
featkw = ut.argparse_dict(vt.get_extract_features_default_params())
fname1 = ut.get_argval('--fname1', type_=str, default='easy1.png')
fname2 = ut.get_argval('--fname2', type_=str, default='easy2.png')
(kpts1, kpts2, fm, fs, rchip1, rchip2) = dummy.testdata_ratio_matches(fname1, fname2, **featkw)
aff_inliers, aff_errors, Aff = get_best_affine_inliers_(
kpts1, kpts2, fm, fs, xy_thresh_sqrd, scale_thresh, ori_thresh)
return kpts1, kpts2, fm, aff_inliers, rchip1, rchip2, xy_thresh_sqrd
def testdata_matching_affine_inliers():
import vtool.tests.dummy as dummy
import vtool as vt
scale_thresh = 2.0
xy_thresh = ut.get_argval('--xy-thresh', type_=float, default=.01)
dlen_sqrd2 = 447271.015
ori_thresh = 1.57
xy_thresh_sqrd = dlen_sqrd2 * xy_thresh
featkw = ut.argparse_dict(vt.get_extract_features_default_params())
fname1 = ut.get_argval('--fname1', type_=str, default='easy1.png')
fname2 = ut.get_argval('--fname2', type_=str, default='easy2.png')
(kpts1, kpts2, fm, fs, rchip1,
rchip2) = dummy.testdata_ratio_matches(fname1, fname2, **featkw)
aff_inliers, aff_errors, Aff = get_best_affine_inliers_(
kpts1, kpts2, fm, fs, xy_thresh_sqrd, scale_thresh, ori_thresh)
return kpts1, kpts2, fm, aff_inliers, rchip1, rchip2, xy_thresh_sqrd
def test_sver_wrapper():
"""
Test to ensure cpp and python agree and that cpp is faster
CommandLine:
python -m vtool.sver_c_wrapper --test-test_sver_wrapper
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --rebuild-sver
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --dummy
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --fname1=easy1.png --fname2=easy2.png
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --fname1=easy1.png --fname2=hard3.png
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --fname1=carl.jpg --fname2=hard3.png
Example:
>>> # ENABLE_DOCTEST
>>> from vtool.sver_c_wrapper import * # NOQA
>>> test_sver_wrapper()
Ignore:
%timeit call_python_version(*args)
%timeit get_affine_inliers_cpp(*args)
"""
import vtool.spatial_verification as sver
import vtool.tests.dummy as dummy
xy_thresh_sqrd = ktool.KPTS_DTYPE(.4)
scale_thresh_sqrd = ktool.KPTS_DTYPE(2.0)
ori_thresh = ktool.KPTS_DTYPE(TAU / 4.0)
keys = 'xy_thresh_sqrd, scale_thresh_sqrd, ori_thresh'.split(', ')
print(ut.dict_str(ut.dict_subset(locals(), keys)))
def report_errors():
pass
if ut.get_argflag('--dummy'):
testtup = dummy.testdata_dummy_matches()
(kpts1, kpts2, fm_input, fs_input, rchip1, rchip2) = testtup
fm_input = fm_input.astype(fm_dtype)
#fm_input = fm_input[0:10].astype(fm_dtype)
#fs_input = fs_input[0:10].astype(np.float32)
else:
fname1 = ut.get_argval('--fname1', type_=str, default='easy1.png')
fname2 = ut.get_argval('--fname2', type_=str, default='easy2.png')
testtup = dummy.testdata_ratio_matches(fname1, fname2)
(kpts1, kpts2, fm_input, fs_input, rchip1, rchip2) = testtup
# pack up call to aff hypothesis
import vtool as vt
import scipy.stats.mstats
scales1 = vt.get_scales(kpts1.take(fm_input.T[0], axis=0))
scales2 = vt.get_scales(kpts2.take(fm_input.T[1], axis=0))
#fs_input = 1 / scipy.stats.mstats.gmean(np.vstack((scales1, scales2)))
fs_input = scipy.stats.mstats.gmean(np.vstack((scales1, scales2)))
print('fs_input = ' + ut.numpy_str(fs_input))
#fs_input[0:-9] = 0
#fs_input = np.ones(len(fm_input), dtype=fs_dtype)
#ut.embed()
#fs_input = scales1 * scales2
args = (kpts1, kpts2, fm_input, fs_input, xy_thresh_sqrd, scale_thresh_sqrd, ori_thresh)
ex_list = []
try:
with ut.Indenter('[TEST1] '):
inlier_tup = vt.compare_implementations(
sver.get_affine_inliers,
get_affine_inliers_cpp,
args, lbl1='py', lbl2='c',
output_lbl=('aff_inliers_list', 'aff_errors_list', 'Aff_mats')
)
out_inliers, out_errors, out_mats = inlier_tup
except AssertionError as ex:
ex_list.append(ex)
raise
try:
import functools
with ut.Indenter('[TEST2] '):
bestinlier_tup = vt.compare_implementations(
functools.partial(sver.get_best_affine_inliers, forcepy=True),
get_best_affine_inliers_cpp,
args, show_output=True, lbl1='py', lbl2='c',
output_lbl=('bestinliers', 'besterror', 'bestmat')
)
bestinliers, besterror, bestmat = bestinlier_tup
except AssertionError as ex:
ex_list.append(ex)
raise
if len(ex_list) > 0:
raise AssertionError('some tests failed. see previous stdout')
#num_inliers_list = np.array(map(len, out_inliers_c))
#best_argx = num_inliers_list.argmax()
##best_inliers_py = out_inliers_py[best_argx]
#best_inliers_c = out_inliers_c[best_argx]
if ut.show_was_requested():
import plottool as pt
fm_output = fm_input.take(bestinliers, axis=0)
fnum = pt.next_fnum()
pt.figure(fnum=fnum, doclf=True, docla=True)
pt.show_chipmatch2(rchip1, rchip2, kpts1, kpts2, fm_input, ell_linewidth=5, fnum=fnum, pnum=(2, 1, 1))
pt.show_chipmatch2(rchip1, rchip2, kpts1, kpts2, fm_output, ell_linewidth=5, fnum=fnum, pnum=(2, 1, 2))
pt.show_if_requested()
def test_sver_wrapper():
"""
Test to ensure cpp and python agree and that cpp is faster
CommandLine:
python -m vtool.sver_c_wrapper --test-test_sver_wrapper
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --rebuild-sver
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --dummy
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --fname1=easy1.png --fname2=easy2.png
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --fname1=easy1.png --fname2=hard3.png
python -m vtool.sver_c_wrapper --test-test_sver_wrapper --show --fname1=carl.jpg --fname2=hard3.png
Example:
>>> # ENABLE_DOCTEST
>>> from vtool.sver_c_wrapper import * # NOQA
>>> test_sver_wrapper()
Ignore:
%timeit call_python_version(*args)
%timeit get_affine_inliers_cpp(*args)
"""
import vtool.spatial_verification as sver
import vtool.tests.dummy as dummy
xy_thresh_sqrd = ktool.KPTS_DTYPE(.4)
scale_thresh_sqrd = ktool.KPTS_DTYPE(2.0)
ori_thresh = ktool.KPTS_DTYPE(TAU / 4.0)
keys = 'xy_thresh_sqrd, scale_thresh_sqrd, ori_thresh'.split(', ')
print(ut.dict_str(ut.dict_subset(locals(), keys)))
def report_errors():
pass
if ut.get_argflag('--dummy'):
testtup = dummy.testdata_dummy_matches()
(kpts1, kpts2, fm_input, fs_input, rchip1, rchip2) = testtup
fm_input = fm_input.astype(fm_dtype)
#fm_input = fm_input[0:10].astype(fm_dtype)
#fs_input = fs_input[0:10].astype(np.float32)
else:
fname1 = ut.get_argval('--fname1', type_=str, default='easy1.png')
fname2 = ut.get_argval('--fname2', type_=str, default='easy2.png')
testtup = dummy.testdata_ratio_matches(fname1, fname2)
(kpts1, kpts2, fm_input, fs_input, rchip1, rchip2) = testtup
# pack up call to aff hypothesis
import vtool as vt
import scipy.stats.mstats
scales1 = vt.get_scales(kpts1.take(fm_input.T[0], axis=0))
scales2 = vt.get_scales(kpts2.take(fm_input.T[1], axis=0))
#fs_input = 1 / scipy.stats.mstats.gmean(np.vstack((scales1, scales2)))
fs_input = scipy.stats.mstats.gmean(np.vstack((scales1, scales2)))
print('fs_input = ' + ut.numpy_str(fs_input))
#fs_input[0:-9] = 0
#fs_input = np.ones(len(fm_input), dtype=fs_dtype)
#ut.embed()
#fs_input = scales1 * scales2
args = (kpts1, kpts2, fm_input, fs_input, xy_thresh_sqrd,
scale_thresh_sqrd, ori_thresh)
ex_list = []
try:
with ut.Indenter('[TEST1] '):
inlier_tup = vt.compare_implementations(
sver.get_affine_inliers,
get_affine_inliers_cpp,
args,
lbl1='py',
lbl2='c',
output_lbl=('aff_inliers_list', 'aff_errors_list', 'Aff_mats'))
out_inliers, out_errors, out_mats = inlier_tup
except AssertionError as ex:
ex_list.append(ex)
raise
try:
import functools
with ut.Indenter('[TEST2] '):
bestinlier_tup = vt.compare_implementations(
functools.partial(sver.get_best_affine_inliers, forcepy=True),
get_best_affine_inliers_cpp,
args,
show_output=True,
lbl1='py',
lbl2='c',
output_lbl=('bestinliers', 'besterror', 'bestmat'))
bestinliers, besterror, bestmat = bestinlier_tup
except AssertionError as ex:
ex_list.append(ex)
raise
if len(ex_list) > 0:
raise AssertionError('some tests failed. see previous stdout')
#num_inliers_list = np.array(map(len, out_inliers_c))
#best_argx = num_inliers_list.argmax()
##best_inliers_py = out_inliers_py[best_argx]
#best_inliers_c = out_inliers_c[best_argx]
if ut.show_was_requested():
import plottool as pt
fm_output = fm_input.take(bestinliers, axis=0)
fnum = pt.next_fnum()
pt.figure(fnum=fnum, doclf=True, docla=True)
pt.show_chipmatch2(rchip1,
rchip2,
kpts1,
kpts2,
fm_input,
ell_linewidth=5,
fnum=fnum,
pnum=(2, 1, 1))
pt.show_chipmatch2(rchip1,
rchip2,
kpts1,
kpts2,
fm_output,
ell_linewidth=5,
fnum=fnum,
pnum=(2, 1, 2))
pt.show_if_requested()
