def make_element(uuid): return PostgresDescriptorElement.from_config(psql_element_config, element_type_str, uuid)
from smqtk.utils.bin_utils import logging, initialize_logging if not logging.getLogger().handlers: initialize_logging(logging.getLogger(), logging.DEBUG) log = logging.getLogger(__name__) log.info("Loading descriptor elements") import cPickle import json from smqtk.representation.descriptor_element.postgres_element import PostgresDescriptorElement d_config = json.load(open('/data/kitware/smqtk/image_cache_cnn_compute/psql_descriptor_config.localhost.json')) d_type_str = open("/data/kitware/smqtk/image_cache_cnn_compute/descriptor_type_name.txt").read().strip() d_elements = [] with open("/data/kitware/smqtk/image_cache_cnn_compute/descriptor_uuid_set.1m_train_sample.pickle") as f: for uuid in cPickle.load(f): d_elements.append( PostgresDescriptorElement.from_config(d_config, d_type_str, uuid) ) log.info("Loading ITQ index algo") from load_algo import load_algo itq_index = load_algo() log.info("Building index") itq_index.build_index(d_elements)