From f23cad412c4c7bd52c73f9f5847eccdb6d776f03 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=B2=9B=E7=9F=B3?= <54824693+bruce2233@users.noreply.github.com> Date: Wed, 23 Apr 2025 10:19:31 +0800 Subject: [PATCH] Update gs_renderer.py --- LHM/models/rendering/gs_renderer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/LHM/models/rendering/gs_renderer.py b/LHM/models/rendering/gs_renderer.py index caa2df2..32e2edc 100755 --- a/LHM/models/rendering/gs_renderer.py +++ b/LHM/models/rendering/gs_renderer.py @@ -1202,7 +1202,7 @@ def forward_single_batch( import cv2 cv2.imwrite( - "fuck.png", + "comp_rgb_debug.png", (out["comp_rgb"].detach().cpu().numpy()[0, ..., ::-1] * 255).astype( np.uint8 ), @@ -1280,7 +1280,7 @@ def forward_cano_batch( for i in range(4): cv2.imwrite( - f"fuck_{i}.png", + f"comp_rgb_debug_{i}.png", (out["comp_rgb"].detach().cpu().numpy()[i, ..., ::-1] * 255).astype( np.uint8 ),