diff --git a/nilornodes.py b/nilornodes.py index 9036eea..f1bad56 100644 --- a/nilornodes.py +++ b/nilornodes.py @@ -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") diff --git a/utils.py b/utils.py index f5b854c..f6c2818 100644 --- a/utils.py +++ b/utils.py @@ -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)