From 57009d919ed59bd416df3374d582102ddaf6da62 Mon Sep 17 00:00:00 2001 From: Jianwen Song Date: Mon, 10 Aug 2026 13:29:30 +1000 Subject: [PATCH] [fix] check decoded feature hash for image split inference --- .../pipelines/split_inference/image_split_inference.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/compressai_vision/pipelines/split_inference/image_split_inference.py b/compressai_vision/pipelines/split_inference/image_split_inference.py index 074638c..32fc25b 100644 --- a/compressai_vision/pipelines/split_inference/image_split_inference.py +++ b/compressai_vision/pipelines/split_inference/image_split_inference.py @@ -286,7 +286,10 @@ def __call__( 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] - if "decoded_feature_tensor_hash" in dec_features: + if ( + "decoded_feature_tensor_hash" in dec_features + and dec_features["decoded_feature_tensor_hash"] is not None + ): out_res["decoded_feature_tensor_hash"] = ( dec_features["decoded_feature_tensor_hash"][e] if e in dec_features["decoded_feature_tensor_hash"]