-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
23 lines (21 loc) · 984 Bytes
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import argparse
from modules.addSticker import AddSticker
import time
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument("--input", type=str, default="tests/persons.jpg", help="Path to input image")
parser.add_argument("--style", type=str, default="tests/uv_face_sticker.png", help="Path to square sticker in Face Texture")
parser.add_argument("--savedir", type=str, default="output/")
args = parser.parse_args()
print(" ⊱ ──────ஓ๑♡๑ஓ ────── ⊰")
print("🎵 hhey, arguments are here if you need to check 🎵")
for arg in vars(args):
print("{:>15}: {:>30}".format(str(arg), str(getattr(args, arg))))
print()
return args
if __name__ == "__main__":
args = get_args()
add_sticker = AddSticker()
start = time.time()
output = add_sticker.run(args.input, args.style, savedir=args.savedir)
print("Time inference: ", round((time.time() - start), 2), 's')