fix(gpu): 修复MuseTalk蓝色块 — Step5删除多余BGR2RGB转换，直接传BGR给VAE

VAE内部preprocess_img已做BGR→RGB，外层再cvtColor导致双重转换、R/B通道互换（蓝脸根因）。该修复此前由运维直接改在GPU机器上验证有效，现补提PR入develop。 GFPGAN链路色彩空间核对：修复后主管线为BGR约定（encode BGR、decode输出按BGR融合）， GFPGAN段BGR→RGB输入/return_rgb=True后RGB→BGR回写，通道一致，无需改动；当前MUSE_USE_GFPGAN=0禁用，待基础功能确认后开启做运行时验证。
2026-09-22 23:51:05 +08:00
parent 90e1460a1b
commit ff2ff863a2
1 changed files with 3 additions and 2 deletions
@@ -805,8 +805,9 @@ def _run_inference(
            crop = frame[y1:y2_eff, x1:x2]
            if crop.size == 0:
                continue
-            crop_rgb = cv2.cvtColor(crop, cv2.COLOR_BGR2RGB)
-            crop_resized = cv2.resize(crop_rgb, (256, 256), interpolation=cv2.INTER_LANCZOS4)
+            # 直接传 BGR 给 VAE：VAE 内部 preprocess_img 已做 BGR→RGB 转换，
+            # 此处再 cvtColor 会造成双重转换、R/B 通道互换（蓝色块根因）。
+            crop_resized = cv2.resize(crop, (256, 256), interpolation=cv2.INTER_LANCZOS4)
            # 使用 VAE 的 get_latents_for_unet 得到 8 通道输入
            # get_latents_for_unet 内部: preprocess(half_mask=True) encode + preprocess(half_mask=False) encode → cat → [1,8,32,32]
            latents = vae.get_latents_for_unet(crop_resized).detach().cpu()