Allow images to be resized to specific resolution

Author: anmarques

src/guidellm/core/request.py

@@ -1,5 +1,5 @@
 import uuid
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Tuple
 
 from pydantic import Field
 
@@ -41,6 +41,14 @@ def number_images(self) -> int:
         else:
             return len(self.images)
 
+    @property
+    def image_resolution(self) -> Tuple[int]:
+        if self.images is None:
+            return None
+        else:
+            return [im.size for im in self.images]
+
+
     def __str__(self) -> str:
         prompt_short = (
             self.prompt[:32] + "..."
@@ -53,5 +61,5 @@ def __str__(self) -> str:
             f"prompt={prompt_short}, prompt_token_count={self.prompt_token_count}, "
             f"output_token_count={self.output_token_count}, "
             f"params={self.params})"
-            f"images={self.number_images}"
+            f"image_resolution={self.image_resolution}"
         )

src/guidellm/request/emulated.py

@@ -30,7 +30,8 @@ class EmulatedConfig:
         generated_tokens_variance (Optional[int]): Variance for generated tokens.
         generated_tokens_min (Optional[int]): Minimum number of generated tokens.
         generated_tokens_max (Optional[int]): Maximum number of generated tokens.
-        images (Optional[int]): Number of input images.
+        images (Optional[int]): Number of images.
+        image_resultion (Optional[List[int]]): Resolution of images.
     """
 
     @staticmethod
@@ -107,6 +108,11 @@ def create_config(config: Optional[Union[str, Path, Dict]]) -> "EmulatedConfig":
     generated_tokens_max: Optional[int] = None
 
     images: int = 0
+    image_resolution = None
+
+    def __post_init__(self):
+        if self.images is not None and self.image_resultion is not None and self.images > 0:
+            assert len(self.image_resolution) == 2
 
     @property
     def prompt_tokens_range(self) -> Tuple[int, int]:
@@ -331,7 +337,7 @@ def __init__(
             settings.emulated_data.filter_end,
         )
         if self._config.images > 0:
-            self._images = load_images(settings.emulated_data.image_source)
+            self._images = load_images(settings.emulated_data.image_source, self._config.image_resolution)
         self._rng = np.random.default_rng(random_seed)
 
         # NOTE: Must be after all the parameters since the queue population

src/guidellm/utils/images.py

@@ -27,7 +27,7 @@ class ImageDescriptor(Serializable):
     )
 
 
-def load_images(data: str) -> List[ImageDescriptor]:
+def load_images(data: str, image_resolution: Optional[List[int]]) -> List[ImageDescriptor]:
     """
     Load an HTML file from a path or URL
 
@@ -56,6 +56,10 @@ def load_images(data: str) -> List[ImageDescriptor]:
                 logger.debug("Loading image: {}", img_url)
                 img_response = requests.get(img_url)
                 img_response.raise_for_status()
+                image = Image.open(BytesIO(img_response.content))
+
+                if image_resolution is not None:
+                    image = image.resize(image_resolution)
 
                 # Load image into Pillow
                 images.append(