xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-13 11:07:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[feat]tested portrait node

Browse Source
This commit is contained in:
zhu yu 2024-11-29 11:20:10 +08:00
parent 55a156d35c
commit 107df30a51
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
custom_functions/AI-portrait-wf-py.py → custom_functions/AI_portrait_wf_py.py
View File

@ -1,5 +1,6 @@
import os import os
import random import random
import numpy as np
import sys import sys
from typing import Sequence, Mapping, Any, Union from typing import Sequence, Mapping, Any, Union
import torch import torch
@ -86,7 +87,7 @@ def add_extra_model_paths() -> None:
add_comfyui_directory_to_sys_path() add_comfyui_directory_to_sys_path()
add_extra_model_paths() # add_extra_model_paths()
def import_custom_nodes() -> None: def import_custom_nodes() -> None:
@ -115,7 +116,7 @@ def import_custom_nodes() -> None:
from nodes import LoadImage, NODE_CLASS_MAPPINGS from nodes import LoadImage, NODE_CLASS_MAPPINGS
def ai_portrait_infer(image_path): def ai_portrait_infer(image_path, abs_path=True, is_ndarray=False):
import_custom_nodes() import_custom_nodes()
with torch.inference_mode(): with torch.inference_mode():
preprocbuildpipe = NODE_CLASS_MAPPINGS["PreprocBuildPipe"]() preprocbuildpipe = NODE_CLASS_MAPPINGS["PreprocBuildPipe"]()
@ -127,7 +128,7 @@ def ai_portrait_infer(image_path):
) )
loadimage = LoadImage() loadimage = LoadImage()
loadimage_24 = loadimage.load_image(image=image_path) loadimage_24 = loadimage.load_image(image=image_path, abs_path=abs_path, is_ndarray=is_ndarray)
facewarppipebuilder = NODE_CLASS_MAPPINGS["FaceWarpPipeBuilder"]() facewarppipebuilder = NODE_CLASS_MAPPINGS["FaceWarpPipeBuilder"]()
facewarppipebuilder_31 = facewarppipebuilder.load_models( facewarppipebuilder_31 = facewarppipebuilder.load_models(
@ -300,13 +301,14 @@ def ai_portrait_infer(image_path):
conditions=get_value_at_index(preprocgetconds_23, 0), conditions=get_value_at_index(preprocgetconds_23, 0),
src_img=get_value_at_index(faceswapmethod_53, 0), src_img=get_value_at_index(faceswapmethod_53, 0),
) )
outimg = get_value_at_index(final_img, 0) * 255.
return outimg.cpu().numpy().astype(np.uint8)
return final_img from PIL import Image
import cv2 import cv2
if __name__ == "__main__": if __name__ == "__main__":
img_path = '/home/user/works/projs/WebServer/distributed-server-node/submodules/VisualForge/ComfyUI/input/portraitInput.jpg' img_path = '/home/user/works/projs/WebServer/distributed-server-node/submodules/VisualForge/ComfyUI/input/portraitInput.jpg'
image = np.array(Image.open(img_path))[..., :3]
final_img = ai_portrait_infer(image, is_ndarray=True)
final_img = ai_portrait_infer(img_path)
cv2.imwrite('fuckaigc.png', final_img) cv2.imwrite('fuckaigc.png', final_img)