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
