Updated depthmap seamless and normalize image color

akatz-ai · Nov 20, 2024 · d9f5a30 · d9f5a30
1 parent 385077a
commit d9f5a30
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/src/ak_make_depthmap_seamless.py b/src/ak_make_depthmap_seamless.py
@@ -71,10 +71,6 @@ def make_depthmap_seamless(self, depthmap_batch):
             # Replace in batch
             depthmap_np[i] = depthmap_seamless_np
 
-        # Remove batch dimension if input was a single image
-        if depthmap_np.shape[0] == 1:
-            depthmap_np = depthmap_np[0]
-
         # Convert back to PyTorch tensor
         depthmap_seamless = torch.from_numpy(depthmap_np).to(depthmap_batch.device).type_as(depthmap_batch)
 

diff --git a/src/ak_normalize_image_color.py b/src/ak_normalize_image_color.py
@@ -14,8 +14,7 @@ def INPUT_TYPES(s):
                     "default": 0.2,
                     "min": 0.0,
                     "max": 1.0,
-                    "step": 0.1,
-                    "round": 0.1, #The value representing the precision to round to, will be set to the step value by default. Can be set to False to disable rounding.
+                    "step": 0.001,
                     "display": "number"}),
                 "red": ("INT",{
                     "default": 255,