def main(): main_arg_parser = argparse.ArgumentParser(description="parser for RAVEN") main_arg_parser.add_argument( "--num-samples", type=int, default=20000, help="number of samples for each component configuration") main_arg_parser.add_argument( "--save-dir", type=str, default="~/Datasets/", help="path to folder where the generated dataset will be saved.") main_arg_parser.add_argument("--seed", type=int, default=1234, help="random seed for dataset generation") main_arg_parser.add_argument( "--fuse", type=int, default=0, help="whether to fuse different configurations") main_arg_parser.add_argument( "--val", type=float, default=2, help="the proportion of the size of validation set") main_arg_parser.add_argument("--test", type=float, default=2, help="the proportion of the size of test set") args = main_arg_parser.parse_args() all_configs = { "center_single": build_center_single(), "distribute_four": build_distribute_four(), "distribute_nine": build_distribute_nine(), "left_center_single_right_center_single": build_left_center_single_right_center_single(), "up_center_single_down_center_single": build_up_center_single_down_center_single(), "in_center_single_out_center_single": build_in_center_single_out_center_single(), "in_distribute_four_out_center_single": build_in_distribute_four_out_center_single() } if not os.path.exists(args.save_dir): os.mkdir(args.save_dir) if args.fuse: if not os.path.exists(os.path.join(args.save_dir, "fuse")): os.mkdir(os.path.join(args.save_dir, "fuse")) fuse(args, all_configs) else: for key in all_configs.keys(): if not os.path.exists(os.path.join(args.save_dir, key)): os.mkdir(os.path.join(args.save_dir, key)) separate(args, all_configs)
def main(): main_arg_parser = argparse.ArgumentParser(description="parser for RAVEN") main_arg_parser.add_argument("--num-samples", type=int, default=10000, help="number of samples for each component configuration") main_arg_parser.add_argument("--save-dir", type=str, default="./Datasets", help="path to folder where the generated dataset will be saved.") main_arg_parser.add_argument("--seed", type=int, default=1234, help="random seed for dataset generation") main_arg_parser.add_argument("--fair", type=int, default=0, help="whether to create FAIR or ORIG dataset") main_arg_parser.add_argument("--val", type=float, default=2, help="the proportion of the size of validation set") main_arg_parser.add_argument("--test", type=float, default=2, help="the proportion of the size of test set") main_arg_parser.add_argument("--save", type=int, default=1, help="save the dataset") args = main_arg_parser.parse_args() args.save_dir = os.path.join(args.save_dir, 'RAVEN' + ('-F' if args.fair else '')) all_configs = {"center_single": build_center_single(), "distribute_four": build_distribute_four(), "distribute_nine": build_distribute_nine(), "left_center_single_right_center_single": build_left_center_single_right_center_single(), "up_center_single_down_center_single": build_up_center_single_down_center_single(), "in_center_single_out_center_single": build_in_center_single_out_center_single(), "in_distribute_four_out_center_single": build_in_distribute_four_out_center_single()} separate(args, all_configs)