diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
index 68f3342a45d7c79fa4050f6a10855ba034d91658..dcf44c69c5233b30424dd52ab70bcd76017a5901 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
@@ -840,7 +840,7 @@ class SRCNN:
         cld_opd_sres = self.do_inference(data)
         cld_opd_sres = denormalize(cld_opd_sres, label_param, mean_std_dct)
 
-        return cld_opd_sres, bt_us.shape[0,]
+        return cld_opd_sres, bt_us.shape[1], bt_us.shape[2]
 
 
 def run_restore_static(directory, ckpt_dir, out_file=None):