Upload 2 files

.gitignore.txt

@@ -0,0 +1,4 @@
+g_model
+flagged
+arcface_model
+retina_model

options/swap_options.py

@@ -0,0 +1,43 @@
+import argparse
+
+
+class SwapOptions():
+    def __init__(self):
+        self.parser = argparse.ArgumentParser()
+        self.initialized = False
+
+    def initialize(self):
+        # paths (data, models, etc...)
+        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.')
+
+        # Video/Image necessary models
+        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')
+
+        # logging and checkpointing
+        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