From f22e40674c8444feb1dd24392a3010b03a472dfe Mon Sep 17 00:00:00 2001 From: Your Name Date: Fri, 26 Dec 2025 23:39:03 +0700 Subject: [PATCH] I've optimized the image loading logic --- load_image_list.py | 4 ++-- load_image_sequence.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/load_image_list.py b/load_image_list.py index cd09319..0dfe960 100644 --- a/load_image_list.py +++ b/load_image_list.py @@ -10,8 +10,8 @@ def load_image(path: str) -> torch.Tensor: Shape: [1, H, W, 3], float32 0..1 """ img = Image.open(path).convert("RGB") - arr = np.array(img).astype(np.float32) / 255.0 - return torch.from_numpy(arr)[None,] + # Optimize: Direct to tensor -> float conversion (avoids intermediate float64 numpy array) + return torch.from_numpy(np.array(img)).float().div_(255.0)[None,] class BatchLoadImageList: diff --git a/load_image_sequence.py b/load_image_sequence.py index 6a47cd6..b6e7418 100644 --- a/load_image_sequence.py +++ b/load_image_sequence.py @@ -10,8 +10,8 @@ def load_image(path: str) -> torch.Tensor: Shape: [1, H, W, 3], float32 0..1 """ img = Image.open(path).convert("RGB") - arr = np.array(img).astype(np.float32) / 255.0 - return torch.from_numpy(arr)[None,] + # Optimize: Direct to tensor -> float conversion (avoids intermediate float64 numpy array) + return torch.from_numpy(np.array(img)).float().div_(255.0)[None,] class BatchLoadImageSequence: