iou_similarity_op.cc 3.8 KB
Newer Older
1
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserved.
W
wanghaox 已提交
2 3 4 5 6 7 8 9 10 11 12 13 14

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. */

15
#include "paddle/fluid/operators/detection/iou_similarity_op.h"
W
wanghaox 已提交
16 17 18 19 20 21 22 23 24 25

namespace paddle {
namespace operators {

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

 protected:
  void InferShape(framework::InferShapeContext *ctx) const override {
W
wanghaox 已提交
26 27 28 29
    PADDLE_ENFORCE(ctx->HasInput("X"),
                   "Input(X) of IOUSimilarityOp should not be null.");
    PADDLE_ENFORCE(ctx->HasInput("Y"),
                   "Input(Y) of IOUSimilarityOp should not be null.");
W
wanghaox 已提交
30 31 32
    auto x_dims = ctx->GetInputDim("X");
    auto y_dims = ctx->GetInputDim("Y");

W
wanghaox 已提交
33
    PADDLE_ENFORCE_EQ(x_dims.size(), 2UL, "The rank of Input(X) must be 2.");
W
wanghaox 已提交
34
    PADDLE_ENFORCE_EQ(x_dims[1], 4UL, "The shape of X is [N, 4]");
W
wanghaox 已提交
35
    PADDLE_ENFORCE_EQ(y_dims.size(), 2UL, "The rank of Input(Y) must be 2.");
W
wanghaox 已提交
36 37
    PADDLE_ENFORCE_EQ(y_dims[1], 4UL, "The shape of Y is [M, 4]");

W
wanghaox 已提交
38
    ctx->ShareLoD("X", /*->*/ "Out");
W
wanghaox 已提交
39 40 41 42 43 44
    ctx->SetOutputDim("Out", framework::make_ddim({x_dims[0], y_dims[0]}));
  }
};

class IOUSimilarityOpMaker : public framework::OpProtoAndCheckerMaker {
 public:
Y
Yu Yang 已提交
45
  void Make() override {
W
wanghaox 已提交
46
    AddInput("X",
47 48 49
             "(LoDTensor, default LoDTensor<float>) "
             "Box list X is a 2-D LoDTensor with shape [N, 4] holds N boxes, "
             "each box is represented as [xmin, ymin, xmax, ymax], "
W
wanghaox 已提交
50 51 52 53 54 55 56
             "the shape of X is [N, 4]. [xmin, ymin] is the left top "
             "coordinate of the box if the input is image feature map, they "
             "are close to the origin of the coordinate system. "
             "[xmax, ymax] is the right bottom coordinate of the box. "
             "This tensor can contain LoD information to represent a batch "
             "of inputs. One instance of this batch can contain different "
             "numbers of entities.");
W
wanghaox 已提交
57 58
    AddInput("Y",
             "(Tensor, default Tensor<float>) "
W
wanghaox 已提交
59 60 61 62 63
             "Box list Y holds M boxes, each box is represented as "
             "[xmin, ymin, xmax, ymax], the shape of X is [N, 4]. "
             "[xmin, ymin] is the left top coordinate of the box if the "
             "input is image feature map, and [xmax, ymax] is the right "
             "bottom coordinate of the box.");
W
wanghaox 已提交
64

65
    AddOutput("Out",
W
wanghaox 已提交
66
              "(LoDTensor, the lod is same as input X) The output of "
67 68
              "iou_similarity op, a tensor with shape [N, M] "
              "representing pairwise iou scores.");
W
wanghaox 已提交
69 70 71

    AddComment(R"DOC(
IOU Similarity Operator.
W
wanghaox 已提交
72
Computes intersection-over-union (IOU) between two box lists.
73
 Box list 'X' should be a LoDTensor and 'Y' is a common Tensor,
W
wanghaox 已提交
74 75
 boxes in 'Y' are shared by all instance of the batched inputs of X.
 Given two boxes A and B, the calculation of IOU is as follows:
76 77 78 79 80 81

$$
IOU(A, B) = 
\frac{area(A\cap B)}{area(A)+area(B)-area(A\cap B)}
$$

W
wanghaox 已提交
82 83 84 85 86 87 88
)DOC");
  }
};
}  // namespace operators
}  // namespace paddle

namespace ops = paddle::operators;
89 90 91
REGISTER_OPERATOR(iou_similarity, ops::IOUSimilarityOp,
                  ops::IOUSimilarityOpMaker,
                  paddle::framework::EmptyGradOpMaker);
W
wanghaox 已提交
92 93 94 95 96

REGISTER_OP_CPU_KERNEL(
    iou_similarity,
    ops::IOUSimilarityKernel<paddle::platform::CPUDeviceContext, float>,
    ops::IOUSimilarityKernel<paddle::platform::CPUDeviceContext, double>);
反馈
建议
客服 返回
顶部