[fix] if-else for the final bits count

Author: chyomin06

compressai_vision/pipelines/remote_inference/image_remote_inference.py

@@ -172,10 +172,12 @@ def __call__(
             out_res["qp"] = (
                 "uncmp" if codec.qp_value is None else codec.qp_value
             )  # Assuming one qp will be used
-            if self.configs["codec"]["decode_only"]:
+
+            if not isinstance(res["bitstream"], dict):
                 out_res["bytes"] = os.stat(res["bitstream"]).st_size
             else:
                 out_res["bytes"] = res["bytes"][0]
+
             out_res["coded_order"] = e
             out_res["org_input_size"] = f'{d[0]["height"]}x{d[0]["width"]}'
             output_list.append(out_res)

compressai_vision/pipelines/remote_inference/video_remote_inference.py

@@ -197,10 +197,12 @@ def __call__(
                 "uncmp" if codec.qp_value is None else codec.qp_value
             )  # Assuming one qp will be used
 
-            if self.configs["codec"]["decode_only"]:
-                out_res["bytes"] = bitstream_bytes / len(dataloader)
+            if not isinstance(res["bitstream"], dict):
+                out_res["bytes"] = os.stat(res["bitstream"]).st_size / len(dataloader)
             else:
+                assert len(res["bytes"]) == len(dataloader)
                 out_res["bytes"] = res["bytes"][e]
+
             out_res["coded_order"] = e
             out_res["org_input_size"] = f'{d[0]["height"]}x{d[0]["width"]}'
 

compressai_vision/pipelines/split_inference/image_split_inference.py

@@ -229,7 +229,11 @@ def __call__(
                 "uncmp" if codec.qp_value is None else codec.qp_value
             )  # Assuming one qp will be used
 
-            out_res["bytes"] = os.stat(res["bitstream"]).st_size
+            if not isinstance(res["bitstream"], dict):
+                out_res["bytes"] = os.stat(res["bitstream"]).st_size
+            else:
+                out_res["bytes"] = res["bytes"][0]
+
             out_res["coded_order"] = e
             out_res["org_input_size"] = f'{d[0]["height"]}x{d[0]["width"]}'
             out_res["input_size"] = dec_features["input_size"][0]

compressai_vision/pipelines/split_inference/video_split_inference.py

@@ -319,11 +319,11 @@ def __call__(
             out_res = dec_features.copy()
             del (out_res["data"], out_res["org_input_size"])
 
-            if self.configs["codec"]["decode_only"]:
-                out_res["bytes"] = bitstream_bytes / len(dataloader)
+            if not isinstance(res["bitstream"], dict):
+                out_res["bytes"] = os.stat(res["bitstream"]).st_size / len(dataloader)
             else:
                 assert len(res["bytes"]) == len(dataloader)
-                out_res["bytes"] = os.stat(res["bitstream"]).st_size / len(dataloader)
+                out_res["bytes"] = res["bytes"][e]
 
             out_res["coded_order"] = e
             out_res["input_size"] = dec_features["input_size"][0]