def frame_indexer(self): if self._frame_indexer is None: if 'ALEX_ENABLE' in os.environ: self._frame_indexer = { 'alex': indexer.AlexnetIndexer(), 'inception': indexer.InceptionIndexer(), } else: self._frame_indexer = { 'inception': indexer.InceptionIndexer(), } return self._frame_indexer
def visual_indexer(self): if self._visual_indexer is None: self._visual_indexer = {'inception': indexer.InceptionIndexer(), 'facenet': indexer.FacenetIndexer(), 'alexnet': indexer.AlexnetIndexer() } return self._visual_indexer
def visual_indexer(self): if IndexerTask._visual_indexer is None: IndexerTask._visual_indexer = { 'inception': indexer.InceptionIndexer(), 'facenet': indexer.FacenetIndexer() } return IndexerTask._visual_indexer
def visual_indexer(self): if IndexerTask._visual_indexer is None: # if IndexerTask._session is None: # logging.info("Creating a global shared session") # config = indexer.tf.ConfigProto() # config.gpu_options.per_process_gpu_memory_fraction = 0.2 # IndexerTask._session = indexer.tf.Session() IndexerTask._visual_indexer = { 'inception': indexer.InceptionIndexer(), 'facenet': indexer.FacenetIndexer(), 'vgg': indexer.VGGIndexer() } return IndexerTask._visual_indexer
def get_index(cls,di): di.ensure() if di.pk not in Indexers._visual_indexer: iroot = "{}/models/".format(settings.MEDIA_ROOT) if di.name == 'inception': Indexers._visual_indexer[di.pk] = indexer.InceptionIndexer(iroot + "{}/network.pb".format(di.uuid)) elif di.name == 'facenet': Indexers._visual_indexer[di.pk] = indexer.FacenetIndexer(iroot + "{}/facenet.pb".format(di.uuid)) elif di.algorithm == 'vgg': Indexers._visual_indexer[di.pk] = indexer.VGGIndexer(iroot + "{}/{}".format(di.uuid,di.files[0]['filename'])) else: raise ValueError,"unregistered indexer with id {}".format(di.pk) return Indexers._visual_indexer[di.pk]
def get_index(self, di): if di.pk not in IndexerTask._visual_indexer: iroot = "{}/models/".format(settings.MEDIA_ROOT) if di.name == 'inception': IndexerTask._visual_indexer[di.pk] = indexer.InceptionIndexer( iroot + "{}/network.pb".format(di.pk)) elif di.name == 'facenet': IndexerTask._visual_indexer[di.pk] = indexer.FacenetIndexer( iroot + "{}/facenet.pb".format(di.pk)) elif di.name == 'vgg': IndexerTask._visual_indexer[di.pk] = indexer.VGGIndexer( iroot + "{}/vgg.pb".format(di.pk)) else: raise ValueError, "unregistered indexer with id {}".format( di.pk) return IndexerTask._visual_indexer[di.pk]
def visual_indexer(self): if IndexerTask._visual_indexer is None: indexer_root_dir = "{}/indexers/".format(settings.MEDIA_ROOT) # if IndexerTask._session is None: # logging.info("Creating a global shared session") # config = indexer.tf.ConfigProto() # config.gpu_options.per_process_gpu_memory_fraction = 0.2 # IndexerTask._session = indexer.tf.Session() IndexerTask._visual_indexer = { 'inception': indexer.InceptionIndexer(indexer_root_dir + "inception/network.pb"), 'facenet': indexer.FacenetIndexer(indexer_root_dir + "facenet/facenet.pb"), 'vgg': indexer.VGGIndexer(indexer_root_dir + "vgg/vgg.pb") } return IndexerTask._visual_indexer