Skip to content

Commit e3044c6

Browse files
chyomin06fracape
authored andcommitted
[minor] fctm yaml updated
1 parent 0548311 commit e3044c6

File tree

3 files changed

+14
-11
lines changed

3 files changed

+14
-11
lines changed

cfgs/codec/fctm.yaml

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -10,16 +10,18 @@ bitstream_name: "${dataset.config.dataset_name}_qp${codec.tools.inner_codec.enc_
1010
verbosity: 0
1111

1212
coding_behaviour: "all"
13-
device: "cpu"
13+
device:
14+
all: "cpu"
15+
enc: ${codec.device.all}
16+
dec: ${codec.device.all}
17+
1418
mac_computation: False # multipy-accumulate computation
1519

1620
enc_configs:
1721
qp: 27
1822
frame_rate: 30
1923
intra_period: 1
2024
n_bit: 10
21-
cfg_file: "use/local/path/of/vtm-23.3/cfg/encoder_intra_vtm.cfg"
22-
default_cfg_file: "optional/use/local/path/for/JM-19.1/cfgs/encoder.cfg"
2325
parallel_encoding: False # std inner codec parallel option
2426
hash_check: 0 # md5 hash check
2527

@@ -38,20 +40,22 @@ tools:
3840
on_restored_ftensor:
3941
enabled: True
4042
period: ${codec.enc_configs.intra_period}
41-
on_fused_ftensor:
43+
on_reduced_ftensor:
4244
enabled: True
4345
period: ${codec.enc_configs.intra_period}
4446

4547
conversion:
4648
type: 'tensor_packing' # or 'bypass'
4749
n_bit: ${codec.enc_configs.n_bit}
48-
packing:
49-
symmetric_feature_channel_flipping: False
5050

5151
inner_codec:
5252
type: 'vtm'
5353
stash_outputs: True
5454
enc_configs: ${codec.enc_configs}
55-
enc_exe: "use/local/path/of/vtm-23.3/bin/EncoderAppStatic"
56-
dec_exe: "use/local/path/of/vtm-23.3/bin/DecoderAppStatic"
57-
merge_exe: "use/local/path/of/vtm-23.3/bin/parcatStatic"
55+
codec_paths:
56+
_root: "/path/to/vtm" # of HM, JM...
57+
enc_exe: "${._root}/bin/EncoderAppStatic" # or TappEncoderStatic, lencod...
58+
dec_exe: "${._root}/bin/DecoderAppStatic" # or TappDecoderStatic, ldecod...
59+
merge_exe: "${._root}/bin/parcatStatic"
60+
cfg_file: "${._root}/cfg/encoder_intra_vtm.cfg"
61+
default_cfg_file: "${._root}/cfg/encoder.cfg"

compressai_vision/codecs/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def reshape_feature_pyramid_to_frame(self, x: Dict, packing_all_in_one=False):
196196

197197
# find the largest tensor
198198
x_sorted = sorted(
199-
x.values(), key=lambda item: math.prod(item[1].size()), reverse=True
199+
x.values(), key=lambda item: math.prod(item[0].size()), reverse=True
200200
)
201201

202202
nbframes, C, H, W = x_sorted[0].size()

scripts/evaluation/sfu_hw_obj/eval_on_sfu_hw_obj_vtm.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,6 @@ echo "==========================================================================
121121

122122
compressai-${PIPELINE}-inference --config-name=${CONF_NAME} \
123123
++pipeline.type=video \
124-
++pipeline.codec.vcm_mode=True \
125124
++paths._run_root=${OUTPUT_DIR} \
126125
++vision_model.arch=faster_rcnn_X_101_32x8d_FPN_3x \
127126
++dataset.type=Detectron2Dataset \

0 commit comments

Comments
 (0)