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
