File size: 1,349 Bytes
2b534de | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | TASK :int = 0
import argparse
parser = argparse.ArgumentParser(description="Custom inference for tgt/ref image pairs.")
parser.add_argument("--task-name", type=str,
default=['face','hair','motion','head'][TASK],
help="face|hair|motion|head")
parser.add_argument("--out-dir", type=str, default='examples/outputs', help="Output directory")
# option 1: pass 2 paths
parser.add_argument("--tgt", type=str, default=None, help="Path to target image. if None, will use paths read from --pair-list")
parser.add_argument("--ref", type=str, default=None, help="Path to reference image")
# option 2: pass a txt containing paths
parser.add_argument("--pair-list", type=str, default='examples/inputs.txt', help="white-space-separated list file: tgt_path ref_path")
args = parser.parse_args()
#-----------------------------------------set TASK--------------------------------------------------------------------------
task_name :str = args.task_name
if task_name is None:
pass
else:
TASK :int = {
'face': 0,
'hair': 1,
'motion': 2,
'head': 3,
}[task_name]
print(f'task: {task_name} transfer (ID: {TASK})')
#---------------------------------------------------------------------------------------------------------------------
from util_and_constant import *
from get_mask import *
from util_cv2 import *
|