crop_op.cc 4.6 KB
Newer Older
W
wanghaoshuang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

   http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License. */

#include "paddle/operators/crop_op.h"
W
wanghaoshuang 已提交
16
#include <boost/lexical_cast.hpp>
W
wanghaoshuang 已提交
17 18 19 20 21 22 23 24 25 26 27

namespace paddle {
namespace operators {

using framework::Tensor;

class CropOp : public framework::OperatorWithKernel {
 public:
  using framework::OperatorWithKernel::OperatorWithKernel;

 protected:
Q
Qiao Longfei 已提交
28 29 30 31 32 33 34 35
  void InferShape(framework::InferShapeContextBase* ctx) const override {
    PADDLE_ENFORCE(ctx->HasInput("X"),
                   "Input(X) of CropOp should not be null.");
    PADDLE_ENFORCE(ctx->HasOutput("Out"),
                   "Output(Out) of CropOp should not be null.");
    auto x_dim = ctx->GetInputDim("X");
    if (!ctx->HasInput("Y")) {
      auto shape = ctx->Attrs().Get<std::vector<int>>("shape");
W
wanghaoshuang 已提交
36
      PADDLE_ENFORCE_EQ(
37
          int64_t(shape.size()), x_dim.size(),
W
wanghaoshuang 已提交
38
          "Shape size should be equal to dimention size of input tensor.");
W
wanghaoshuang 已提交
39
      std::vector<int64_t> tensor_shape(shape.size());
40
      for (size_t i = 0; i < shape.size(); ++i) {
41
        tensor_shape[i] = static_cast<int64_t>(shape[i]);
W
wanghaoshuang 已提交
42
      }
Q
Qiao Longfei 已提交
43
      ctx->SetOutputDim("Out", framework::make_ddim(tensor_shape));
W
wanghaoshuang 已提交
44
    } else {
Q
Qiao Longfei 已提交
45 46
      auto y_dim = ctx->GetInputDim("Y");
      PADDLE_ENFORCE_EQ(framework::arity(x_dim), framework::arity(y_dim),
W
wanghaoshuang 已提交
47 48
                        "Tensor rank of both CropOp's "
                        "inputs must be same.");
Q
Qiao Longfei 已提交
49
      ctx->SetOutputDim("Out", y_dim);
W
wanghaoshuang 已提交
50 51 52 53 54 55
    }
  }
};

class CropOpMaker : public framework::OpProtoAndCheckerMaker {
 public:
Q
Qiao Longfei 已提交
56
  CropOpMaker(framework::OpProto* proto, framework::OpAttrChecker* op_checker)
W
wanghaoshuang 已提交
57
      : OpProtoAndCheckerMaker(proto, op_checker) {
58 59 60 61 62 63 64 65 66
    AddInput("X",
             "The input of pad op. "
             "The input should be a k-D tensor(k > 0 and k < 7)");
    AddInput("Y",
             "The input used as reference for cropping"
             " with the same dimension as X. ");
    AddOutput("Out",
              "The output of crop op "
              "with the same dimension as X.");
67 68 69 70 71 72 73 74 75
    AddAttr<std::vector<int>>("offsets",
                              "A list<int> describing offsets to be cropped."
                              "The size of offsets list should be as same as "
                              "dimension size of  input X.");
    AddAttr<std::vector<int>>("shape",
                              "A list<int> describing the shape of output."
                              "The size of shape list should be as same as "
                              "dimension size of  input X.")
        .SetDefault(std::vector<int>());
W
wanghaoshuang 已提交
76 77
    AddComment(R"DOC(
Crop Operator.
78 79
Crop input into output, as specified by offsets and shape.

Q
Qiao Longfei 已提交
80
There are two ways to set shape:
81
1. referenc input: crop input X as shape as reference input.
Q
Qiao Longfei 已提交
82
                    The dimension of reference input should
83 84
                    be as same as input X.
2. shape list: crop input X by shape described by a list<int>.
Q
Qiao Longfei 已提交
85
               The size of shape list should be as same as
86 87 88 89 90 91
               dimension size of  input X.

The input should be a k-D tensor(k > 0 and k < 7). As an example:

Given:

92 93 94
    X = [[0, 1, 2, 0, 0]
         [0, 3, 4, 0, 0]
         [0, 0, 0, 0, 0]]
95

Q
Qiao Longfei 已提交
96
and
97

98
    offsets = [0, 1]
99 100

and
Q
Qiao Longfei 已提交
101

102
    shape = [2, 2]
103

Q
Qiao Longfei 已提交
104
then we get
105

106 107
    Out = [[1, 2],
           [3, 4]]
108

W
wanghaoshuang 已提交
109 110 111 112 113 114 115 116 117
)DOC");
  }
};

class CropOpGrad : public framework::OperatorWithKernel {
 public:
  using framework::OperatorWithKernel::OperatorWithKernel;

 protected:
Q
Qiao Longfei 已提交
118 119 120 121 122 123 124 125
  void InferShape(framework::InferShapeContextBase* ctx) const override {
    PADDLE_ENFORCE(ctx->HasInput("X"), "Input(X) should not be null");
    PADDLE_ENFORCE(ctx->HasInput(framework::GradVarName("Out")),
                   "Input(Out@GRAD) should not be null");
    auto x_dims = ctx->GetInputDim("X");
    auto x_grad_name = framework::GradVarName("X");
    if (ctx->HasOutput(x_grad_name)) {
      ctx->SetOutputDim(x_grad_name, x_dims);
126
    }
W
wanghaoshuang 已提交
127 128 129 130 131 132 133 134
  }
};

}  // namespace operators
}  // namespace paddle

namespace ops = paddle::operators;
REGISTER_OP(crop, ops::CropOp, ops::CropOpMaker, crop_grad, ops::CropOpGrad);
135
REGISTER_OP_CPU_KERNEL(crop, ops::CropKernel<float>);
W
wanghaoshuang 已提交
136 137
REGISTER_OP_CPU_KERNEL(crop_grad,
                       ops::CropGradKernel<paddle::platform::CPUPlace, float>);