imcui/third_party/DarkFeat/datasets/InvISP/config/config.py

import argparse

BATCH_SIZE = 1

DATA_PATH = "./data/"



def get_arguments():
    parser = argparse.ArgumentParser(description="training codes")
    
    parser.add_argument("--task", type=str, help="Name of this training")
    parser.add_argument("--data_path", type=str, default=DATA_PATH, help="Dataset root path.")
    parser.add_argument("--batch_size", type=int, default=BATCH_SIZE, help="Batch size for training. ")       
    parser.add_argument("--debug_mode", dest='debug_mode', action='store_true',  help="If debug mode, load less data.")    
    parser.add_argument("--gamma", dest='gamma', action='store_true', help="Use gamma compression for raw data.")     
    parser.add_argument("--camera", type=str, default="NIKON_D700", choices=["NIKON_D700", "Canon_EOS_5D"], help="Choose which camera to use. ")    
    parser.add_argument("--rgb_weight", type=float, default=1, help="Weight for rgb loss. ")                 
    
    
    return parser