提交 55eb2fcf 编写于 作者: H Haonan

format correction

上级 f5995300
......@@ -40,19 +40,17 @@ void RotateLayer::forward(PassType passType) {
MatrixPtr outV = getOutputValue();
for (int b = 0; b < batchSize_; b ++) {
MatrixPtr inputSample
= Matrix::create(input->getData() + b * sampleSize_,
sampleHeight_,
sampleWidth_,
false,
useGpu_);
MatrixPtr outputSample
= Matrix::create(outV->getData() + b * sampleSize_,
sampleWidth_,
sampleHeight_,
false,
useGpu_);
for (int b = 0; b < batchSize_; b++) {
MatrixPtr inputSample = Matrix::create(input->getData() + b * sampleSize_,
sampleHeight_,
sampleWidth_,
false,
useGpu_);
MatrixPtr outputSample = Matrix::create(outV->getData() + b * sampleSize_,
sampleWidth_,
sampleHeight_,
false,
useGpu_);
inputSample->rotate(outputSample, false, true);
}
......@@ -71,21 +69,21 @@ void RotateLayer::backward(const UpdateCallback& callback) {
// the grad should be rotated in the reverse direction
MatrixPtr preGrad = getInputGrad(0);
for (int b = 0; b < batchSize_; b ++) {
MatrixPtr inputSampleGrad
= Matrix::create(preGrad->getData() + b * sampleSize_,
sampleHeight_,
sampleWidth_,
false,
useGpu_);
MatrixPtr outputSampleGrad
= Matrix::create(outputGrad->getData() + b * sampleSize_,
sampleWidth_,
sampleHeight_,
false,
useGpu_);
MatrixPtr tmpGrad
= Matrix::create(sampleHeight_, sampleWidth_, false, useGpu_);
for (int b = 0; b < batchSize_; b++) {
MatrixPtr inputSampleGrad =
Matrix::create(preGrad->getData() + b * sampleSize_,
sampleHeight_,
sampleWidth_,
false,
useGpu_);
MatrixPtr outputSampleGrad =
Matrix::create(outputGrad->getData() + b * sampleSize_,
sampleWidth_,
sampleHeight_,
false,
useGpu_);
MatrixPtr tmpGrad =
Matrix::create(sampleHeight_, sampleWidth_, false, useGpu_);
outputSampleGrad->rotate(tmpGrad, false, false);
inputSampleGrad->add(*tmpGrad);
}
......
......@@ -1833,7 +1833,6 @@ class PoolLayer(LayerBase):
pool_conf.channels)
@config_layer('spp')
class SpatialPyramidPoolLayer(LayerBase):
def __init__(self, name, inputs, **xargs):
......
......@@ -1707,15 +1707,17 @@ def rotate_layer(input, height, name=None, layer_attr=None):
:rtype: LayerOutput
"""
assert isinstance(input, LayerOutput)
l = Layer(name=name,
height=height,
type=LayerType.ROTATE_LAYER,
inputs=[input.name],
**ExtraLayerAttribute.to_kwargs(layer_attr))
return LayerOutput(name=name,
layer_type=LayerType.ROTATE_LAYER,
parents=[input],
size=l.config.size)
l = Layer(
name=name,
height=height,
type=LayerType.ROTATE_LAYER,
inputs=[input.name],
**ExtraLayerAttribute.to_kwargs(layer_attr))
return LayerOutput(
name=name,
layer_type=LayerType.ROTATE_LAYER,
parents=[input],
size=l.config.size)
@wrap_name_default()
......@@ -1750,11 +1752,12 @@ def flip_layer(input, height, name=None, layer_attr=None):
:rtype: LayerOutput
"""
assert isinstance(input, LayerOutput)
return rotate_layer(input=rotate_layer(input=input,
height=height),
height=height,
name=name,
layer_attr=layer_attr)
return rotate_layer(
input=rotate_layer(
input=input, height=height),
height=height,
name=name,
layer_attr=layer_attr)
@wrap_name_default()
......
......@@ -39,10 +39,8 @@ z1 = mixed_layer(
assert z1.size > 0
y2 = fc_layer(input=y, size=15)
z2 = rotate_layer(input=y2,
height=5)
z3 = flip_layer(input=y2,
height=3)
z2 = rotate_layer(input=y2, height=5)
z3 = flip_layer(input=y2, height=3)
cos1 = cos_sim(a=x1, b=y1)
cos3 = cos_sim(a=x1, b=y2, size=3)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册