| import argparse |
|
|
|
|
| class SwapOptions(): |
| def __init__(self): |
| self.parser = argparse.ArgumentParser() |
| self.initialized = False |
|
|
| def initialize(self): |
| |
| self.parser.add_argument('--arcface_path', type=str, |
| default="arcface_model/arcface/arc_res50.h5", |
| help='path to arcface model. Used to extract identity from source.') |
|
|
| |
| self.parser.add_argument('--retina_path', type=str, |
| default="retinaface/retinaface_res50.h5", |
| help='path to retinaface model.') |
| self.parser.add_argument('--compare', type=bool, |
| default=True, |
| help='If true, concatenates the frame with the manipulated frame') |
|
|
| self.parser.add_argument('--load', type=int, |
| default=30, |
| help='int of number to load checkpoint weights.') |
| self.parser.add_argument('--device_id', type=int, default=0, |
| help='which device to use') |
|
|
| |
| self.parser.add_argument('--log_dir', type=str, default='logs/runs/', |
| help='logging directory') |
| self.parser.add_argument('--log_name', type=str, default='affa_f', |
| help='name of the run, change this to track several experiments') |
|
|
| self.parser.add_argument('--chkp_dir', type=str, default='checkpoints/', |
| help='checkpoint directory (will use same name as log_name!)') |
| self.initialized = True |
|
|
| def parse(self): |
| if not self.initialized: |
| self.initialize() |
| self.opt = self.parser.parse_args() |
| return self.opt |
