diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py index 6583af0a17420254bb2ac0282226796d68f91275..b0b48de5d6d6dc2c57b846a8d93dec44eb3de9c4 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:((self.LEN_Y+2)-1), 1:((self.LEN_X+2)-1), 3:] + input_2d_no_conv = input_2d[:, 1:-1, 1:-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)