From 6bf5975ada290fed96f918680fd5a56058fe1a17 Mon Sep 17 00:00:00 2001 From: pphuc25 Date: Mon, 16 Oct 2023 13:18:59 +0700 Subject: [PATCH] reformat: change init transform to out of loop --- imagebind/data.py | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/imagebind/data.py b/imagebind/data.py index 70a7a03b..3b826198 100644 --- a/imagebind/data.py +++ b/imagebind/data.py @@ -80,20 +80,22 @@ def load_and_transform_vision_data(image_paths, device): return None image_outputs = [] + + data_transform = transforms.Compose( + [ + transforms.Resize( + 224, interpolation=transforms.InterpolationMode.BICUBIC + ), + transforms.CenterCrop(224), + transforms.ToTensor(), + transforms.Normalize( + mean=(0.48145466, 0.4578275, 0.40821073), + std=(0.26862954, 0.26130258, 0.27577711), + ), + ] + ) + for image_path in image_paths: - data_transform = transforms.Compose( - [ - transforms.Resize( - 224, interpolation=transforms.InterpolationMode.BICUBIC - ), - transforms.CenterCrop(224), - transforms.ToTensor(), - transforms.Normalize( - mean=(0.48145466, 0.4578275, 0.40821073), - std=(0.26862954, 0.26130258, 0.27577711), - ), - ] - ) with open(image_path, "rb") as fopen: image = Image.open(fopen).convert("RGB")