Carica un nuovo file

Reference Software/UFV1.0-Pruning/src/arguments.py

+import argparse
+ 
+def str2bool(v):
+    if isinstance(v, bool):
+        return v
+    if v.lower() in ('yes', 'true', 't', 'y', '1'):
+        return True
+    elif v.lower() in ('no', 'false', 'f', 'n', '0'):
+        return False
+    else:
+        raise argparse.ArgumentTypeError('Boolean value expected.')
+ 
+def get_arguments():
+    parser = argparse.ArgumentParser(description='Train Dansley Residual Super Resolution Model',
+                                     formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+    parser.add_argument('--pruning_method', default='growing_reg', type=str, help='The pruning Algoritm To use (random, growing_reg)')
+    parser.add_argument('--weigths', type=str, help='Path to the pretrained weigths')
+    parser.add_argument('-e', '--epochs', type=int, default=50, help='Number of training epochs between pruning steps')
+    parser.add_argument('-ei', '--epochs_iterative', type=int, default=1, help='Number of training epochs for learning sparsity')
+    parser.add_argument('-b', '--batch', type=int, default=2, help='Batch size')
+    parser.add_argument('--lr', type=float, default=10e-5, help='Learning rate')
+    parser.add_argument('-p','--patience', type=float, default=15, help='The amount of epoch that the plateaou scheduler have to wait(whitout changes) before changing the lr')
+    parser.add_argument('-f', '--filter', type=str, default=None, help='A filter to use to get only wanted imagase from the one present in the path')
+    parser.add_argument('-r', '--runs', type=str, default='runs/', help='Base dir for runs outputs and artifacts')
+    parser.add_argument('-s', '--scale', type=int, default=2, help='The super resolution scaling factor')
+    parser.add_argument('-c', '--crop', type=int, default=None, help='The size of the crop to extract from the image')
+    parser.add_argument('--precision', type=str, default='single', choices=('single', 'half'), help='FP precision for test (single(32bit) | half(16bit))')
+    parser.add_argument('--chop', type=str2bool, default=False,help='enable memory-efficient forward, needed for train, or forward to big resolution')
+    parser.add_argument('--self_ensemble', type=str2bool, default=False,help='Needed if memory issue arise')
+    parser.add_argument('--loader', type=str, default='yuv', help='The loader to use (folder_yuv, sequence_yuv, div2k_rgb)')
+    parser.add_argument('--seed',  type=int,default=0, help='The random seed to use')
+    parser.add_argument('--data_range', type=int, default=255, help='Size of the input Data')
+    parser.add_argument('--test_mode', type=str2bool, default=False, help='If True the prune will be done in test mode, without oploading the data on WandB, and with reduced dataset')
+    parser.add_argument('--loss_epsylon', type=float, default=10e-5, help='The value of epsylon for the Charbonnier Loss (default 10e-5)')
+    # YUV - DataLoader
+    parser.add_argument('--low_res_data', type=str, help='Path to low-res data dir')
+    parser.add_argument('--high_res_data', type=str, help='Path to high-res data dir')
+    parser.add_argument('-yf', '--yuv_format', default="yuv420p10le", help='The yuv format of the imput and output images. (note only 400 and 444 are supported)')
+    parser.add_argument('-ys', '--yuv_size', default="128x128", help='The size in format wxh of the low res images.')
+    parser.add_argument('-plr', '--prefix_lowres',  default='', help='The prefix of low res files')
+    parser.add_argument('-phr', '--prefix_highres',  default='', help='The prefix of high res files')
+    parser.add_argument('--only_y_channel',  default=False, help='If to use only the Y channel')
+    parser.add_argument('--yuv_testset',  default='EVCIntra', help='Which Raw Dataset To use: EVCIntra, EVCSDHD, EVCHD4K, VVCSDHD, VVCHD4K.')
+    
+    # Pruning
+    parser.add_argument('--pruning_steps',type=int, default=25, help='Number of pruning stpes to reach the wanted rateo')
+    parser.add_argument('--pruning_target_ratio', type=float, default=0.5, help='The wanted sparsity for each step')
+    parser.add_argument('--max-pruning-ratio', type=float, default=1.0, help='Maximum pruning percentual (to limit the max pruning possible in some corner scenarios)')
+    parser.add_argument('--reg', type=float, default=5e-4,  help='Starting Multiplier for the regularization factor in the Grow Regularization Pruner')
+    parser.add_argument('--delta_reg', type=float, default=1e-4, help='The update delta to perform on the growing regularization factor after each pruning step')
+    parser.add_argument('--eval_each', type=int, default=1, help='How many pruning step to wait before tesing the model')
+    
+    # Pruning-Growing Reg Specific
+    parser.add_argument('--update_reg_interval', type=int, default=5, help='The amount of mini-batch to wait before increase the regularization')
+    parser.add_argument('--target_regularization', type=float, default=1.0, help='The regularization to meet to stop the learning sparsity step in Growing Reg')
+    
+    # Infer
+    parser.add_argument('--infer_count', type=int, default=-1, help='The amount of images to process in the inference.')
+    parser.add_argument('--experiment_path', type=str, default=None, help='The path to the experiment folder from where to get wights and results')
+
+    args = parser.parse_args()
+    return args
\ No newline at end of file