moved old pil2tensor helper function code
This commit is contained in:
@@ -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")
|
||||
|
||||
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user