nodes/MrForExample-ComfyUI-3D-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #374 from kampelmuehler/fix_rotate_normals_horizontally

Browse Source 
fix Rotate_Normal_Maps_Horizontally node
This commit is contained in:
Mr. For Example
2024-12-10 18:17:06 -06:00
committed by GitHub
parent add829643c 24a876add6
commit bfefa53b7e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+2 -1
View File
@@ -559,10 +559,11 @@ class Rotate_Normal_Maps_Horizontally:
CATEGORY = "Comfy3D/Preprocessor"
def make_image_grid(self, normal_maps, normal_masks, clockwise):
rotate_direction = 1 if clockwise is True else -1
if normal_maps.shape[0] > 1:
from Unique3D.scripts.utils import rotate_normals_torch
pil_image_list = torch_imgs_to_pils(normal_maps, normal_masks)
pil_image_list = rotate_normals_torch(pil_image_list, return_types='pil', rotate_direction=int(clockwise))
pil_image_list = rotate_normals_torch(pil_image_list, return_types='pil', rotate_direction=rotate_direction)
normal_maps = pils_to_torch_imgs(pil_image_list, normal_maps.dtype, normal_maps.device)
return (normal_maps,)