From 82dd75844c86b369151ac174d2b5bc479589aa02 Mon Sep 17 00:00:00 2001 From: Hacker 17082006 Date: Thu, 19 Oct 2023 22:48:44 +0700 Subject: [PATCH] Fix bug --- __init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index f9c27a3..ef7ea53 100644 --- a/__init__.py +++ b/__init__.py @@ -340,7 +340,7 @@ class RenderSMPLMesh: #Ref: https://github.com/mmatl/pyrender/issues/10#issuecomment-468995891 normalized_depth = 1 - normalized_depth #https://github.com/Fannovel16/comfyui_controlnet_aux/blob/main/src/controlnet_aux/util.py#L24 - depth_frames = [torch.from_numpy(np.concatenate([x, x, x], axis=2)) for x in normalized_depth[None, ...]] + depth_frames = [torch.from_numpy(np.concatenate([x, x, x], axis=2)) for x in normalized_depth] depth_frames = torch.stack(depth_frames, dim=0) return (color_frames, depth_frames)