args = parser.parse_args() config, exp_identifier, save_path = load_config(args.config, args.exp_dir) logger = logging.getLogger() ############################### # LOAD DATA ############################# image_builder, crop_builder = None, None # Load image logger.info('Loading images..') use_resnet = False if 'image' in config['model']: logger.info('Loading images..') image_builder = get_img_builder(config['model']['image'], args.img_dir) use_resnet = image_builder.is_raw_image() assert False, "Guesser + Image is not yet available" # Load data logger.info('Loading data..') trainset = Dataset(args.data_dir, "train", image_builder, crop_builder) validset = Dataset(args.data_dir, "valid", image_builder, crop_builder) testset = Dataset(args.data_dir, "test", image_builder, crop_builder) # Load dictionary logger.info('Loading dictionary..') tokenizer = GWTokenizer(os.path.join(args.data_dir, args.dict_file)) # Build Network
############################# # LOAD DATA ############################# # Load image image_builder, crop_builder = None, None use_resnet = False logger.info("Loading ") t_begin = time.time() if config['inputs'].get('image', False): logger.info('Loading images..') image_builder = get_img_builder(config['model']['image'], args.img_dir) use_resnet = image_builder.is_raw_image() if config['inputs'].get('crop', False): logger.info('Loading crops..') crop_builder = get_img_builder(config['model']['crop'], args.crop_dir, is_crop=True) use_resnet = crop_builder.is_raw_image() # Load data logger.info('Loading data..') all_img_bbox = {} all_img_describtion = []
args = parser.parse_args() eval_config, exp_identifier, save_path = load_config( args.config, args.exp_dir) # Load all networks configs logger = logging.getLogger() ############################### # LOAD DATA ############################# # Load image logger.info('Loading images..') image_builder = get_img_builder(eval_config['image'], args.img_dir) crop_builder = None if eval_config.get('crop', False): logger.info('Loading crops..') crop_builder = get_img_builder(eval_config['crop'], args.crop_dir, is_crop=True) # Load data logger.info('Loading data..') trainset = Dataset(args.data_dir, "train", image_builder, crop_builder) validset = Dataset(args.data_dir, "valid", image_builder, crop_builder) testset = Dataset(args.data_dir, "test", image_builder, crop_builder) dataset, dummy_dataset = trainset, validset
loop_config, exp_identifier, save_path = load_config(args.config, args.exp_dir) # Load all networks configs oracle_config = get_config_from_xp(os.path.join(args.networks_dir, "oracle"), args.oracle_identifier) guesser_config = get_config_from_xp(os.path.join(args.networks_dir, "guesser"), args.guesser_identifier) qgen_config = get_config_from_xp(os.path.join(args.networks_dir, "qgen"), args.qgen_identifier) logger = logging.getLogger() ############################### # LOAD DATA ############################# # Load image logger.info('Loading images..') image_loader = get_img_builder(qgen_config['model']['image'], args.image_dir) crop_loader = None # get_img_builder(guesser_config['model']['crop'], args.image_dir) # Load data logger.info('Loading data..') trainset = Dataset(args.data_dir, "train", image_loader, crop_loader) validset = Dataset(args.data_dir, "valid", image_loader, crop_loader) testset = Dataset(args.data_dir, "test", image_loader, crop_loader) # Load dictionary logger.info('Loading dictionary..') tokenizer = GWTokenizer(os.path.join(args.data_dir, args.dict_file)) ############################### # LOAD NETWORKS #############################
os.path.join(args.networks_dir, "oracle"), args.oracle_identifier) guesser_config = get_config_from_xp( os.path.join(args.networks_dir, "guesser"), args.guesser_identifier) qgen_config = get_config_from_xp(os.path.join(args.networks_dir, "qgen"), args.qgen_identifier) logger = logging.getLogger() ############################### # LOAD DATA ############################# # Load image logger.info('Loading images..') image_builder = get_img_builder(qgen_config['model']['image'], args.img_dir, custom_features=True) crop_builder = None if oracle_config['inputs'].get('crop', False): logger.info('Loading crops..') crop_builder = get_img_builder(oracle_config['model']['crop'], args.crop_dir, is_crop=True) # Load dictionary logger.info('Loading dictionary..') tokenizer = GWTokenizer(os.path.join(args.data_dir, args.dict_file)) ############################### # LOAD NETWORKS