fix: image remote inference with resizing

Author: jianwensong

compressai_vision/pipelines/remote_inference/image_remote_inference.py

@@ -81,6 +81,11 @@ def __call__(
         """
         self._update_codec_configs_at_pipeline_level(len(dataloader))
         org_map_func = dataloader.dataset.get_org_mapper_func()
+        resize_mapper = (
+            dataloader.dataset.get_org_mapper_func(use_rgb=True)
+            if self._compress_after_resizing
+            else None
+        )
         output_list = []
         timing = {
             "encode": metric_tracking(),
@@ -104,12 +109,6 @@ def __call__(
                     "org_input_size": org_img_size,
                 }
 
-                resize_mapper = (
-                    dataloader.dataset.get_org_mapper_func(use_rgb=True)
-                    if self._compress_after_resizing
-                    else None
-                )
-
                 res, enc_time_details, _ = self._compress(
                     codec,
                     frame,
@@ -154,7 +153,7 @@ def __call__(
                 res["bitstream"],
                 self.codec_output_dir,
                 file_prefix,
-                org_img_size,
+                org_img_size=None if self._compress_after_resizing else org_img_size,
                 remote_inference=True,
                 vcm_mode=self.configs["codec"]["vcm_mode"],
                 output10b=output10b,