def reset_testdbs(**kwargs):
# Step 0) Parse Args
default_args = {'reset_' + key: False
for key in six.iterkeys(TEST_DBNAMES_MAP)}
default_args['reset_all'] = False
default_args.update(kwargs)
argdict = ut.parse_dict_from_argv(default_args)
if not any(list(six.itervalues(argdict))):
# Default behavior is to reset the small dbs
argdict['reset_testdb0'] = True
argdict['reset_testdb1'] = True
argdict['reset_testdb_guiall'] = True
# Step 1) Delete DBs to be Reset
for key, dbname in six.iteritems(TEST_DBNAMES_MAP):
if argdict.get('reset_' + key, False) or argdict['reset_all']:
delete_dbdir(dbname)
# Step 3) Ensure DBs that dont exist
ensure_smaller_testingdbs()
workdir = ibeis.sysres.get_workdir()
if not ut.checkpath(join(workdir, 'PZ_MTEST'), verbose=True):
ibeis.ensure_pz_mtest()
if not ut.checkpath(join(workdir, 'NAUT_test'), verbose=True):
ibeis.ensure_nauts()
if not ut.checkpath(join(workdir, 'testdb2'), verbose=True):
ibeis.init.sysres.ensure_testdb2()
# Step 4) testdb1 becomes the main database
workdir = ibeis.sysres.get_workdir()
TESTDB1 = join(workdir, 'testdb1')
sysres.set_default_dbdir(TESTDB1)
def from_argv_dict(cls, **kwargs):
"""
handy command line tool
ut.parse_argv_cfg
"""
cfg = cls(**kwargs)
new_vals = ut.parse_dict_from_argv(cfg)
cfg.update(**new_vals)
return cfg
def testdata_kpts():
import utool as ut
import vtool as vt
import pyhesaff
img_fpath = ut.grab_test_imgpath(ut.get_argval('--fname', default='star.png'))
kwargs = ut.parse_dict_from_argv(pyhesaff.get_hesaff_default_params())
(kpts, vecs) = pyhesaff.detect_feats(img_fpath, **kwargs)
imgBGR = vt.imread(img_fpath)
return kpts, vecs, imgBGR
def from_argv_dict(cls, **kwargs):
"""
handy command line tool
ut.parse_argv_cfg
"""
cfg = cls(**kwargs)
new_vals = ut.parse_dict_from_argv(cfg)
cfg.update(**new_vals)
return cfg
def reset_testdbs(**kwargs):
# Step 0) Parse Args
import ibeis
ibeis.ENABLE_WILDBOOK_SIGNAL = False
default_args = {
'reset_' + key: False
for key in six.iterkeys(TEST_DBNAMES_MAP)
}
default_args['reset_all'] = False
default_args.update(kwargs)
argdict = ut.parse_dict_from_argv(default_args)
if not any(list(six.itervalues(argdict))):
# Default behavior is to reset the small dbs
argdict['reset_testdb0'] = True
argdict['reset_testdb1'] = True
argdict['reset_testdb_guiall'] = True
# Step 1) Delete DBs to be Reset
for key, dbname in six.iteritems(TEST_DBNAMES_MAP):
if argdict.get('reset_' + key, False) or argdict['reset_all']:
delete_dbdir(dbname)
# Step 3) Ensure DBs that dont exist
ensure_smaller_testingdbs()
workdir = ibeis.sysres.get_workdir()
if not ut.checkpath(join(workdir, 'PZ_MTEST'), verbose=True):
ibeis.ensure_pz_mtest()
if not ut.checkpath(join(workdir, 'NAUT_test'), verbose=True):
ibeis.ensure_nauts()
if not ut.checkpath(join(workdir, 'wd_peter2'), verbose=True):
ibeis.ensure_wilddogs()
if not ut.checkpath(join(workdir, 'testdb2'), verbose=True):
ibeis.init.sysres.ensure_testdb2()
# Step 4) testdb1 becomes the main database
workdir = ibeis.sysres.get_workdir()
TESTDB1 = join(workdir, 'testdb1')
sysres.set_default_dbdir(TESTDB1)
