def get_global_distinctiveness_modeldir(ibs, ensure=True):
"""
Returns:
global_distinctdir (str): ibs internal directory
"""
global_distinctdir = sysres.get_global_distinctiveness_modeldir(ensure=ensure)
return global_distinctdir
def view_distinctiveness_model_dir():
r"""
CommandLine:
python -m ibeis.algo.hots.distinctiveness_normalizer --test-view_distinctiveness_model_dir
Example:
>>> # DISABLE_DOCTEST
>>> from ibeis.algo.hots.distinctiveness_normalizer import * # NOQA
>>> view_distinctiveness_model_dir()
"""
global_distinctdir = sysres.get_global_distinctiveness_modeldir()
ut.vd(global_distinctdir)
def request_species_distinctiveness_normalizer(species, cachedir=None, verbose=False):
"""
helper function to get distinctivness model independent of IBEIS.
"""
if species in DISTINCTIVENESS_NORMALIZER_CACHE:
dstcnvs_normer = DISTINCTIVENESS_NORMALIZER_CACHE[species]
else:
if cachedir is None:
cachedir = sysres.get_global_distinctiveness_modeldir(ensure=True)
dstcnvs_normer = DistinctivnessNormalizer(species, cachedir=cachedir)
if not dstcnvs_normer.exists(cachedir):
# download normalizer if it doesn't exist
download_baseline_distinctiveness_normalizer(cachedir, species)
dstcnvs_normer.load(cachedir)
print(ut.get_object_size_str(dstcnvs_normer, 'dstcnvs_normer = '))
print('Loaded distinctivness normalizer')
#dstcnvs_normer.ensure_flann(cachedir)
assert dstcnvs_normer.exists(cachedir, need_flann=True), (
'normalizer should have been downloaded, but it doesnt exist')
DISTINCTIVENESS_NORMALIZER_CACHE[species] = dstcnvs_normer
return dstcnvs_normer
def list_distinctivness_cache():
global_distinctdir = sysres.get_global_distinctiveness_modeldir()
print(ut.list_str(ut.ls(global_distinctdir)))
def clear_distinctivness_cache(j):
global_distinctdir = sysres.get_global_distinctiveness_modeldir()
ut.remove_files_in_dir(global_distinctdir)
