Skip to content

Commit 971b760

Browse files
chyomin06fracape
authored andcommitted
[fix] bug - related to vcmrs support
1 parent e3044c6 commit 971b760

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

compressai_vision/codecs/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,7 @@ def decode(
141141
file_prefix: str = "",
142142
org_img_size: Dict = None,
143143
remote_inference=False,
144+
vcm_mode=False,
144145
):
145146
del org_img_size
146147
del file_prefix # used in other codecs that write log files

compressai_vision/codecs/ffmpeg.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -285,6 +285,7 @@ def decode(
285285
file_prefix: str = "",
286286
org_img_size: Dict = None,
287287
remote_inference=False,
288+
vcm_mode=False,
288289
) -> bool:
289290
"""
290291
Decodes a bitstream into video frames and extract features from the decoded frames.
@@ -297,6 +298,7 @@ def decode(
297298
Dict: dictionary of output features.
298299
"""
299300
assert not remote_inference # TODO (fracape) remote inference not supported yet
301+
assert not vcm_mode # TODO To be developed upon supporting remote inference.
300302
bitstream_path = Path(bitstream_path)
301303
assert bitstream_path.is_file()
302304

0 commit comments

Comments
 (0)