moved old pil2tensor helper function code

This commit is contained in:
Sebastian Monroy
2025-05-20 15:44:53 +01:00
parent 4f45e0130e
commit 6786d94d44
2 changed files with 4 additions and 4 deletions
-4
View File
@@ -999,10 +999,6 @@ class NilorLoadImageByIndex:
def IS_CHANGED(s, image_directory, seed, sort_mode, reverse_sort):
return seed
## Helper function equivalent to Mikey's pil2tensor
#def pil2tensor(self, image):
# return torch.from_numpy(np.array(image).astype(np.float32) / 255.0).unsqueeze(0)
def load_image_by_index(self, image_directory, seed, sort_mode, reverse_sort):
if not os.path.exists(image_directory):
raise FileNotFoundError(f"Image directory {image_directory} does not exist")
+4
View File
@@ -14,6 +14,10 @@ def numpy2pil(image: np.ndarray, mode=None):
return Image.fromarray(np.clip(255.0 * image, 0, 255).astype(np.uint8), mode)
## Helper function equivalent to Mikey's pil2tensor
#def pil2tensor(self, image):
# return torch.from_numpy(np.array(image).astype(np.float32) / 255.0).unsqueeze(0)
def pil2tensor(image: Image.Image):
return torch.from_numpy(pil2numpy(image)).unsqueeze(0)