diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py index da49a98bc9f8fe5996585116b274075a8e058e4e..6583af0a17420254bb2ac0282226796d68f91275 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py @@ -415,7 +415,7 @@ class SRCNN: input_2d = self.inputs[0] input_2d_conv = input_2d[:, :, :, 0:3] - input_2d_no_conv = input_2d[:, 1:(130-1), 1:(130-1), 3:] + input_2d_no_conv = input_2d[:, 1:((self.LEN_Y+2)-1), 1:((self.LEN_X+2)-1), 3:] print('input: ', input_2d.shape) print('input_2d_conv: ', input_2d_conv.shape) print('input_2d_no_conv: ', input_2d_no_conv.shape)