Skip to content

Commit 2339312

Browse files
committed
[fix] bug - adding default vcm_mode argument for remote inference pipeline
1 parent 59e5e45 commit 2339312

File tree

6 files changed

+18
-10
lines changed

6 files changed

+18
-10
lines changed

cfgs/codec/fctm.yaml

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,17 @@ enc_configs:
2727

2828
tools:
2929
feature_reduction: # reserved for restoration part as well
30-
type: 'pre-trained' # or 'bypass'
31-
qidx: 6 # Learned based feature tensor reduction (m66341)
32-
split_ctx: "obj" # or "seg" # or "alt1" or "dn53"
33-
temporal_resampling_enabled: False
34-
# To properly enable Temporal resampling (m67615),
30+
type: 'pre-trained' # by default # or 'bypass'
31+
learned_model:
32+
name: 'light_fedrnet' # fedrnet (m66341) light_fedrnet (m71247)
33+
selective_learning_strategy: False # to (m71203) be True by default for FCTM v6.1.0
34+
split_ctx: "obj" # or "seg" # or "alt1" or "dn53"
35+
36+
range_based_channel_truncation: False # (m71203) to be True by default for FCTM v6.1.0
37+
temporal_resampling_enabled: False # (m67615)
38+
# To properly enable Temporal resampling ,
3539
# temporal_resampling_enabled = True and
3640
# "cfgs/codec/vtm-23.3/encoder_lowdelay_vtm_gop4.cfg" must be used
37-
intra_period: ${codec.enc_configs.intra_period} # as reference
3841

3942
refinements:
4043
on_restored_ftensor:
@@ -58,4 +61,4 @@ tools:
5861
dec_exe: "${._root}/bin/DecoderAppStatic" # or TappDecoderStatic, ldecod...
5962
merge_exe: "${._root}/bin/parcatStatic"
6063
cfg_file: "${._root}/cfg/encoder_intra_vtm.cfg"
61-
default_cfg_file: "${._root}/cfg/encoder.cfg"
64+
default_cfg_file: "${._root}/cfg/encoder.cfg"

cfgs/pipeline/remote_inference.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ codec:
2525
skip_n_frames: 0 # This is encoder only option
2626
n_frames_to_be_encoded: -1 #(-1 = encode all input), This is encoder only option
2727
measure_complexity: "${codec.mac_computation}"
28+
vcm_mode: False
2829
nn_task:
2930
dump_results: False
3031
output_results_dir: "${codec.output_dir}/output_results"

compressai_vision/pipelines/remote_inference/image_remote_inference.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,8 @@ def __call__(
141141
self.codec_output_dir,
142142
file_prefix,
143143
org_img_size,
144-
True,
144+
remote_inference=True,
145+
vcm_mode=self.configs["codec"]["vcm_mode"],
145146
)
146147
end = time_measure()
147148
timing["decode"].append((end - start))

scripts/evaluation/hieve/eval_on_hieve_vcmrs.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,9 @@ echo "==========================================================================
9393

9494
compressai-${PIPELINE}-inference --config-name=${CONF_NAME} \
9595
++pipeline.type=video \
96+
++pipeline.codec.vcm_mode=True \
9697
++paths._run_root=${OUTPUT_DIR} \
97-
++vision_model.arch=jde_1088x608 \
98+
++vision_model.arch=jde_1088x608 \
9899
++vision_model.jde_1088x608.splits="[105, 90, 75]" \
99100
++dataset.type=TrackingDataset \
100101
++dataset.datacatalog=MPEGHIEVE \

scripts/evaluation/mpeg_oiv6/eval_on_mpeg_oiv6_vcmrs.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ echo "==========================================================================
8585

8686
compressai-${PIPELINE}-inference --config-name=${CONF_NAME} \
8787
++pipeline.type=image \
88+
++pipeline.codec.vcm_mode=True \
8889
++paths._run_root=${OUTPUT_DIR} \
8990
++vision_model.arch=${NETWORK_MODEL} \
9091
++dataset.type=Detectron2Dataset \

scripts/evaluation/tvd/eval_on_tvd_vcmrs.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,9 @@ echo "==========================================================================
8888

8989
compressai-${PIPELINE}-inference --config-name=${CONF_NAME} \
9090
++pipeline.type=video \
91+
++pipeline.codec.vcm_mode=True \
9192
++paths._run_root=${OUTPUT_DIR} \
92-
++vision_model.arch=jde_1088x608 \
93+
++vision_model.arch=jde_1088x608 \
9394
++vision_model.jde_1088x608.splits="[36, 61, 74]" \
9495
++dataset.type=TrackingDataset \
9596
++dataset.datacatalog=MPEGTVDTRACKING \

0 commit comments

Comments
 (0)