def get_dataset_args(parser, args=[]):
    """
    Get all relevant parameters to handle the dataset
    """
    # Dataset:
    # ------------------------------------------------------------------------
    parser.add_argument(
        "--dataset_type",
        type=str,
        default="MRNet",
        help='Dataset: "MRNet, KneeXray, ..."',
    )
    parser.add_argument(
        "--n_gpu", type=int, default=1, help="Number of GPUs on the PC",
    )
    parser.add_argument(
        "--crop_size", type=int, default=16, help="Depth of the MR-Scan",
    )
    parser.add_argument(
        "--rot_max", type=int, default=30, help="Maximum random rotation angle",
    )
    parser.add_argument(
        "--crop_percent",
        type=float,
        default=0.95,
        help="if we use triplenet take percentage of av. layers",
    )
    parser.add_argument(
        "--rand_crop", type=bool, default=True, help="Decide whether to crop randomly",
    )
    parser.add_argument(
        "--pic_size", type=int, default=256, help="Picture size of the MR-Scan",
    )
    parser.add_argument(
        "--batch_size", type=int, default=16, help="Batch size for training",
    )
    parser.add_argument(
        "--num_worker",
        type=int,
        default=0,
        help="Number of workers for loading the dataset",
    )
    # ------------------------------------------------------------------------
    return parser