def _exp_train(out_base_dir,
data_const_train,
data_const_val,
data_sign='hico'):
args = parser.parse_args()
not_specified_args = manage_required_args(
args,
parser,
required_args=['imgs_per_batch', 'fp_to_tp_ratio'],
optional_args=[
'verb_given_appearance', 'verb_given_human_appearance',
'verb_given_object_appearance',
'verb_given_boxes_and_object_label', 'verb_given_human_pose',
'rcnn_det_prob'
])
exp_name = 'factors'
if args.rcnn_det_prob:
exp_name += '_rcnn_det_prob'
if args.verb_given_appearance:
exp_name += '_appearance'
if args.verb_given_human_appearance:
exp_name += '_human_appearance'
if args.verb_given_object_appearance:
exp_name += '_object_appearance'
if args.verb_given_boxes_and_object_label:
exp_name += '_boxes_and_object_label'
if args.verb_given_human_pose:
pose = '_human_pose'
exp_name += pose
exp_const = ExpConstants(exp_name=exp_name, out_base_dir=out_base_dir)
exp_const.log_dir = os.path.join(exp_const.exp_dir, 'log')
exp_const.model_dir = os.path.join(exp_const.exp_dir, 'models')
exp_const.num_epochs = 10
exp_const.imgs_per_batch = args.imgs_per_batch
exp_const.lr = 1e-3
model_const = Constants()
model_const.hoi_classifier = HoiClassifierConstants(data_sign)
model_const.hoi_classifier.verb_given_appearance = args.verb_given_appearance
model_const.hoi_classifier.verb_given_human_appearance = args.verb_given_human_appearance
model_const.hoi_classifier.verb_given_object_appearance = args.verb_given_object_appearance
model_const.hoi_classifier.verb_given_boxes_and_object_label = args.verb_given_boxes_and_object_label
model_const.hoi_classifier.verb_given_human_pose = args.verb_given_human_pose
model_const.hoi_classifier.rcnn_det_prob = args.rcnn_det_prob
train.main(exp_const, data_const_train, data_const_val, model_const,
data_sign)
def exp_train():
args = parser.parse_args()
not_specified_args = manage_required_args(
args,
parser,
required_args=['fappend', 'imgs_per_batch', 'fp_to_tp_ratio'],
optional_args=[
'verb_given_appearance', 'verb_given_human_appearance',
'verb_given_object_appearance',
'verb_given_boxes_and_object_label', 'rcnn_det_prob'
])
exp_name = 'factors'
exp_name += '_' + args.fappend
out_base_dir = os.path.join(os.getcwd(),
'data_symlinks/hico_exp/hoi_classifier')
exp_const = ExpConstants(exp_name=exp_name, out_base_dir=out_base_dir)
exp_const.log_dir = os.path.join(exp_const.exp_dir, 'log')
exp_const.model_dir = os.path.join(exp_const.exp_dir, 'models')
exp_const.num_epochs = 10
exp_const.imgs_per_batch = args.imgs_per_batch
exp_const.lr = 1e-3
data_const_train = FeatureConstants(subset='train')
data_const_val = FeatureConstants(subset='val')
model_const = Constants()
model_const.hoi_classifier = HoiClassifierConstants()
model_const.hoi_classifier.verb_given_appearance = args.verb_given_appearance
model_const.hoi_classifier.verb_given_human_appearance = args.verb_given_human_appearance
model_const.hoi_classifier.verb_given_object_appearance = args.verb_given_object_appearance
model_const.hoi_classifier.verb_given_boxes_and_object_label = args.verb_given_boxes_and_object_label
model_const.hoi_classifier.verb_given_human_pose = args.verb_given_human_pose
model_const.hoi_classifier.rcnn_det_prob = args.rcnn_det_prob
train.main(exp_const, data_const_train, data_const_val, model_const)
