config.py

import argparse
from asyncio import FastChildWatcher
from logging import root

__all__ = ['parse_arguments_main']

def parse_arguments_main():
    parser = argparse.ArgumentParser()
    ## learning paradigm 
    parser.add_argument('--paradigm', '-p', type=str, default='c2d', choices=['c2d', 'c3d', 'f2d'])

    # ----------------------------- centralized learning ----------------------------- #
    parser.add_argument('--dataset', '-d', type=str, default='mvtec2d', choices=['_example', 'mvtec2d', 'mvtec3d', 'mpdd', 'mvtecloco', 'mtd', 
                                                                              'btad', 'mvtec2df3d', 'visa', 'dagm', 'coad'])
    parser.add_argument('--model', '-m', type=str, default='softpatch', choices=['_example', '_patchcore', 'patchcore', 'csflow', 'dne', 
        'draem', 'igd', 'cutpaste', 'devnet', 'dra', 'favae', 'padim', 'reverse', 'spade', 'fastflow', 'softpatch', 'cfa', 'stpm',
        'simplenet', 'softpatch'])
    parser.add_argument('--net', '-n', type=str, default='wide_resnet50', choices=['net_example', 'wide_resnet50', 'resnet18', 'net_csflow',
        'vit_b_16', 'net_draem', 'net_dra', 'net_igd', 'net_reverse', 'net_favae', 'net_fastflow', 'net_cfa', 'net_devnet', 
        'vig_ti_224_gelu'])

    parser.add_argument('--root-path', '-rp', type=str, default=None)
    parser.add_argument('--data-path', '-dp', type=str, default=None)

    parser.add_argument('--train-task-id', '-tid', type=int, default=[0], nargs='+')
    parser.add_argument('--valid-task-id', '-vid', type=int, default=[0], nargs='+')
    parser.add_argument('--sampler-percentage', '-sp', type=float, default=None)

    # vanilla 
    parser.add_argument('--vanilla', '-v', action='store_true', default=False)
    
    # semi-supervised 
    parser.add_argument('--semi', '-s', action='store_true', default=False)
    parser.add_argument('--semi-anomaly-num', '-san', type=int, default=None)
    parser.add_argument('--semi-overlap', '-so', action='store_true', default=False)
    
    # continual 
    parser.add_argument('--continual', '-c', action='store_true', default=False)

    # fewshot 
    parser.add_argument('--fewshot', '-f', action='store_true', default=False)
    parser.add_argument('--fewshot-exm', '-fe', type=int, default=None)
    parser.add_argument('--fewshot-data-aug', '-fda', action='store_true', default=False)
    parser.add_argument('--fewshot-feat-aug', '-ffa', action='store_true', default=False)
    parser.add_argument('--fewshot-num-dg', '-fnd', type=int, default=None)
    parser.add_argument('--fewshot-aug-type', '-fat', default=None, nargs='+', 
                        choices=['normal', 'rotation', 'scale', 'translate', 'flip', 'color_jitter', 'perspective'])

    # noisy label
    parser.add_argument('--noisy', '-z', action='store_true', default=False)
    parser.add_argument('--noisy-overlap', '-no', action='store_true', default=False)
    parser.add_argument('--noisy-ratio', '-nr', type=float, default=None)

    # transfer 
    parser.add_argument('--transfer', '-t', action='store_true', default=False)
    parser.add_argument('--transfer-target-sample-num', '-ttn', type=int, default=None)

    # data augmentation type
    parser.add_argument('--train-aug-type', '-tag', default=None, choices=['normal', 'cutpaste'], help='data augmentation type')
    parser.add_argument('--valid-aug-type', '-vag', default=None, choices=['normal', 'cutpaste'], help='data augmentation type')

    # universal
    parser.add_argument('--gpu-id', '-g', type=str, default=None)
    parser.add_argument('--server-moda', '-sm', type=str, default=None, choices=['eno1', 'lo'])
    parser.add_argument('--num-epochs', '-ne', type=int, default=None)
    parser.add_argument('--seed', type=int, default=None)
    parser.add_argument('--debug', action='store_true', default=False)
    parser.add_argument('--vis', '-vis', action='store_true', default=True)
    parser.add_argument('--vis-em', action='store_true', default=False)

    parser.add_argument('--save-model', action='store_true', default=False)
    parser.add_argument('--load-model', action='store_true', default=False)
    parser.add_argument('--load-model-dir', type=str, default=None)

    # ----------------------------- federated learning ----------------------------- #
    parser.add_argument('--fed-aggregate-method', '-fam', type=str, default=None)
    parser.add_argument('--num-round', type=int, default=None)


    args = parser.parse_args()
    return args