[dev] update fcm api interface and output bitstream size calculation corrected

Author: chyomin06

compressai_vision/codecs/fctm.py

@@ -45,13 +45,12 @@ class FCTM(feature_coding_model):
         """Does no encoding/decoding whatsoever. Use for debugging."""
 
         def __init__(self, **kwargs):
-            super().__init__(
-                kwargs["coding_behaviour"],
-                kwargs["tools"],
-                kwargs["device"],
-                kwargs["verbosity"],
-                kwargs["mac_computation"],
-            )
+
+            enc_configs = kwargs.pop("enc_configs")
+            self.qp = enc_configs["qp"]
+            self.eval_encode = kwargs.pop("eval_encode")
+
+            super().__init__(**kwargs)
 
             tools = kwargs["tools"]
             if (
@@ -68,9 +67,6 @@ def __init__(self, **kwargs):
 
                 raise SystemExit()
 
-            self.qp = kwargs["enc_configs"]["qp"]
-            self.eval_encode = kwargs["eval_encode"]
-
         @property
         def qp_value(self):
             return self.qp

compressai_vision/pipelines/split_inference/image_split_inference.py

@@ -222,10 +222,8 @@ 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"]:
-                out_res["bytes"] = os.stat(res["bitstream"]).st_size
-            else:
-                out_res["bytes"] = res["bytes"][0]
+
+            out_res["bytes"] = os.stat(res["bitstream"]).st_size
             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

@@ -28,6 +28,7 @@
 # ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 
+import os
 from typing import Dict, List, Tuple, TypeVar
 
 import torch
@@ -304,10 +305,13 @@ 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)
             else:
-                out_res["bytes"] = res["bytes"][e]
+                assert len(res["bytes"]) == len(dataloader)
+                out_res["bytes"] = os.stat(res["bitstream"]).st_size / len(dataloader)
+
             out_res["coded_order"] = e
             out_res["input_size"] = dec_features["input_size"][0]
             out_res["org_input_size"] = (