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)
logger.info(ut.get_object_size_str(dstcnvs_normer,
'dstcnvs_normer = '))
logger.info('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 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 wbia.algo.hots.distinctiveness_normalizer --test-view_distinctiveness_model_dir
Example:
>>> # DISABLE_DOCTEST
>>> from wbia.algo.hots.distinctiveness_normalizer import * # NOQA
>>> view_distinctiveness_model_dir()
"""
global_distinctdir = sysres.get_global_distinctiveness_modeldir()
ut.vd(global_distinctdir)
def list_distinctivness_cache():
global_distinctdir = sysres.get_global_distinctiveness_modeldir()
logger.info(ut.repr2(ut.ls(global_distinctdir)))
def clear_distinctivness_cache(j):
global_distinctdir = sysres.get_global_distinctiveness_modeldir()
ut.remove_files_in_dir(global_distinctdir)