ec2f475ddf9a2b301e3ccd5d9255596fbcbeb524,mmdnn/conversion/caffe/caffe_emitter.py,CaffeEmitter,compute_output_shape,#CaffeEmitter#Any#Any#Any#,214

Before Change

            shape = shape_to_list(shape)
            h_i = shape[1]
            w_i = shape[2]
            pad_h = IR_node.get_attr("pads")[1]
            pad_w = IR_node.get_attr("pads")[2]
            stride_h = IR_node.get_attr("strides")[1]
            stride_w = IR_node.get_attr("strides")[2]

            if IR_node.type == "Pool":

After Change

            shape = shape_to_list(shape)
            h_i = shape[1]
            w_i = shape[2]
            pad_h, pad_w = self._get_symmetric_padding(IR_node)
            stride_h = IR_node.get_attr("strides")[1]
            stride_w = IR_node.get_attr("strides")[2]
            
            if IR_node.type == "Pool":

Instances