args.py

import argparse

def get_args_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument('--workers', type=int, help='number of data loading workers', default=0)
    parser.add_argument('--models', help='target classifiers,Resnet50 or Inception_v3 or Densenet121',
                        default='Resnet50')
    parser.add_argument('--lamda_adv', type=int, help='weight of adv_loss', default=3)
    parser.add_argument('--eps', type=float,help=' budget', default=8/255)
    parser.add_argument('--inputpath', help='', default=r'./val_rgb_randomselect/')
    parser.add_argument('--outputpath', help='', default=r'./recover/')
    parser.add_argument('--pre_model', help='Init INN_model', default=r'./pretrained/model_final.pt')
    parser.add_argument('--pass_num', type=int, help='If you stop without finishing all images, you can restart at the stopped index', default=0)
    args = parser.parse_args()
    return args