Unverified Commit fab8066a authored by AK391's avatar AK391 Committed by GitHub

add side by side option

parent 6c2de966
...@@ -33,7 +33,7 @@ def generator(pretrained=True, device="cpu", progress=True, check_hash=True): ...@@ -33,7 +33,7 @@ def generator(pretrained=True, device="cpu", progress=True, check_hash=True):
return model return model
def face2paint(device="cpu", size=512): def face2paint(device="cpu", size=512, side_by_side=True):
from PIL import Image from PIL import Image
from torchvision.transforms.functional import to_tensor, to_pil_image from torchvision.transforms.functional import to_tensor, to_pil_image
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment