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)