提交 69b12225 编写于 作者: W wanghaoshuang

fix crop layer python wrapper bug

上级 0b788ef7
7 合并请求!11636[IMPORTANT] MKLDNN layout: Support for sum operator,!8482Release/0.11.0,!8190Release/0.11.0,!8189Release/0.11.0,!6633给线性回归的get-started代码加上了预测的示例~~,!4615Feature/tensor array add python binding,!2490Add crop layers for supporting FCN model
......@@ -1988,16 +1988,11 @@ class PadLayer(LayerBase):
@config_layer('crop')
class CropLayer(LayerBase):
def __init__(self, inputs, axis, offset, shape, name, **xargs):
def __init__(self, name, inputs, axis, offset, shape, **xargs):
super(CropLayer, self).__init__(name, 'crop', 0, inputs=inputs, **xargs)
self.conf.axis = axis
self.conf.axis = offset
self.conf.axis = shape
crop = self.inputs[0].crop
self.config.inputs[0].crop_conf.axis = crop.axis
self.config.inputs[0].crop_conf.offset.extend(crop.offset)
self.config.inputs[0].crop_conf.shape.extend(crop.shape)
self.config.axis = axis
self.config.offset.extend(offset)
self.config.shape.extend(shape)
# get channel, width and height from input_0 layer
input_layer = self.get_input_layer(0)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册
反馈
建议
客服 返回
顶部