CalvinBackup/ComfyUI-DeepFuze
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Deepfuze updated

Browse Source
This commit is contained in:
Sam Khoze
2024-07-10 23:13:47 +05:30
committed by GitHub
parent 81508114a1
commit 20ab8904cc
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+9 -8
View File
@@ -515,7 +515,6 @@ class DeepFuzeFaceSwap:
"required": { "required": {
"source_images": ("IMAGE",), "source_images": ("IMAGE",),
"target_images": ("IMAGE",), "target_images": ("IMAGE",),
"audio": ("AUDIO",),
"enhancer": ("None,codeformer,gfpgan_1.2,gfpgan_1.3,gfpgan_1.4,gpen_bfr_256,gpen_bfr_512,gpen_bfr_1024,gpen_bfr_2048,restoreformer_plus_plus".split(","),{"default":'None'}), "enhancer": ("None,codeformer,gfpgan_1.2,gfpgan_1.3,gfpgan_1.4,gpen_bfr_256,gpen_bfr_512,gpen_bfr_1024,gpen_bfr_2048,restoreformer_plus_plus".split(","),{"default":'None'}),
"faceswap_model":("blendswap_256,inswapper_128,inswapper_128_fp16,simswap_256,simswap_512_unofficial,uniface_256".split(","),{"default":"blendswap_256"}), "faceswap_model":("blendswap_256,inswapper_128,inswapper_128_fp16,simswap_256,simswap_512_unofficial,uniface_256".split(","),{"default":"blendswap_256"}),
"frame_enhancer": ("None,clear_reality_x4,lsdir_x4,nomos8k_sc_x4,real_esrgan_x2,real_esrgan_x2_fp16,real_esrgan_x4,real_esrgan_x4_fp16,real_hatgan_x4,span_kendata_x4,ultra_sharp_x4".split(","),{"default":'None'}), "frame_enhancer": ("None,clear_reality_x4,lsdir_x4,nomos8k_sc_x4,real_esrgan_x2,real_esrgan_x2_fp16,real_esrgan_x4,real_esrgan_x4_fp16,real_hatgan_x4,span_kendata_x4,ultra_sharp_x4".split(","),{"default":'None'}),
@@ -533,6 +532,7 @@ class DeepFuzeFaceSwap:
}, },
"optional": { "optional": {
"audio": ("AUDIO",),
"meta_batch": ("VHS_BatchManager",), "meta_batch": ("VHS_BatchManager",),
"loop_count": ("INT", {"default": 0, "min": 0, "max": 100, "step": 1}), "loop_count": ("INT", {"default": 0, "min": 0, "max": 100, "step": 1}),
"filename_prefix": ("STRING", {"default": "deepfuze"}), "filename_prefix": ("STRING", {"default": "deepfuze"}),
@@ -561,7 +561,6 @@ class DeepFuzeFaceSwap:
self, self,
source_images, source_images,
target_images, target_images,
audio,
enhancer, enhancer,
faceswap_model, faceswap_model,
frame_enhancer, frame_enhancer,
@@ -573,7 +572,8 @@ class DeepFuzeFaceSwap:
face_mask_padding_top, face_mask_padding_top,
device, device,
frame_rate: int, frame_rate: int,
loop_count: int, audio="",
loop_count: int = 0,
filename_prefix="deepfuze", filename_prefix="deepfuze",
format="video/h265-mp4", format="video/h265-mp4",
pingpong=False, pingpong=False,
@@ -920,11 +920,12 @@ class DeepFuzeFaceSwap:
# faceswap_filename = temp_file # faceswap_filename = temp_file
print(result.stderr) print(result.stderr)
audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav") if audio:
torchaudio.save(audio_file,audio["waveform"][0],audio["sample_rate"]) audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav")
subprocess.run(f"ffmpeg -i {faceswap_filename} -i {audio_file} -c copy {faceswap_filename.replace('.mp4','_.mp4')} -y".split()) torchaudio.save(audio_file,audio["waveform"][0],audio["sample_rate"])
subprocess.run(f"ffmpeg -i {faceswap_filename} -i {audio_file} -c copy {faceswap_filename.replace('.mp4','_.mp4')} -y".split())
return load_video_cv(faceswap_filename.replace('.mp4','_.mp4'),0,'Disabled',512,512,0,0,1) return load_video_cv(faceswap_filename.replace('.mp4','_.mp4'),0,'Disabled',512,512,0,0,1)
return load_video_cv(faceswap_filename,0,'Disabled',512,512,0,0,1)