def get_parser()

in make_data_radioactive.py [0:0]

• 29 lines of code
• 1 McCabe index (conditional complexity)

def get_parser():
    parser = argparse.ArgumentParser()

    # main parameters
    parser.add_argument("--dump_path", type=str, default="")
    parser.add_argument("--exp_name", type=str, default="bypass")
    parser.add_argument("--exp_id", type=str, default="")

    parser.add_argument("--img_size", type=int, default=256)
    parser.add_argument("--crop_size", type=int, default=224)
    parser.add_argument("--data_augmentation", type=str, default="random", choices=["center", "random"])

    parser.add_argument("--radius", type=int, default=10)
    parser.add_argument("--epochs", type=int, default=300)
    parser.add_argument("--lambda_ft_l2", type=float, default=0.5)
    parser.add_argument("--lambda_l2_img", type=float, default=0.05)
    parser.add_argument("--optimizer", type=str, default="sgd,lr=0.1-0.01-0.001,momentum=0.9,weight_decay=0.0001")
    parser.add_argument("--carrier_path", type=str, default="", help="Direction in which to move features")
    parser.add_argument("--carrier_id", type=int, default=0, help="Id of direction in direction array")

    parser.add_argument("--angle", type=float, default=None, help="Angle (if cone)")
    parser.add_argument("--half_cone", type=bool_flag, default=True)

    parser.add_argument("--img_list", type=str, default=None,
                        help="File that contains list of all images")
    parser.add_argument("--img_paths", type=str, default='',
                        help="Path to image to which apply adversarial pattern")

    parser.add_argument("--marking_network", type=str, required=True)
    # parser.add_argument("--image_sizes")

    # debug
    parser.add_argument("--debug_train", type=bool_flag, default=False,
                        help="Use valid sets for train sets (faster loading)")
    parser.add_argument("--debug_slurm", type=bool_flag, default=False,
                        help="Debug from a SLURM job")
    parser.add_argument("--debug", help="Enable all debug flags",
                        action="store_true")

    return parser