Add generate_type as argument to training/example/create_unified_dataset.py as...

Add generate_type as argument to training/example/create_unified_dataset.py as this argument is critical

training/example/create_unified_dataset.py

@@ -20,10 +20,11 @@ parser.add_argument("--components", action="store", nargs='?', type=str, help="c
 parser.add_argument("--nb_patches", action="store", default=1, nargs='?', type=int, help="nb patches to extract, nb_patches=-1 means extracting all patches")
 parser.add_argument("--output_file", action="store", nargs='?', type=str, help="output binary file (patches in float format)")
 parser.add_argument('--random_sample', type=int, default=1, help='whether to sample randomly')
+parser.add_argument('--generate_type', type=int, default=0, help='0 ~ using some specifications for the data loading of JVET-AA0088; 1 ~ using some specifications for the data loading of JVET-A0111')
 args = parser.parse_args()
 
 
-dl=data_loader.DataLoader(args.input_json,args.patch_size,args.poc_list)
+dl=data_loader.DataLoader(args.input_json,args.patch_size,args.poc_list,generate_type=args.generate_type)
 
 print("Nb samples available: {}".format(dl.nb_patches()))
 print("Available components: {}".format(dl.components))