test_iou_similarity_op.py 2.5 KB
Newer Older
1
#   Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
W
wanghaox 已提交
2
#
3 4 5
# 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
W
wanghaox 已提交
6
#
7
#     http://www.apache.org/licenses/LICENSE-2.0
W
wanghaox 已提交
8
#
9 10 11 12 13 14
# 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.

W
wanghaox 已提交
15 16
import unittest
import numpy as np
B
baiyf 已提交
17
import numpy.random as random
W
wanghaox 已提交
18 19
import sys
import math
20
from .op_test import OpTest
W
wanghaox 已提交
21 22 23 24 25 26 27 28


class TestIOUSimilarityOp(OpTest):
    def test_check_output(self):
        self.check_output()

    def setUp(self):
        self.op_type = "iou_similarity"
B
baiyf 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
        self.boxes1 = random.rand(2, 4).astype('float32')
        self.boxes2 = random.rand(3, 4).astype('float32')
        self.output = random.rand(2, 3).astype('float32')
        for row in range(self.boxes1.shape[0]):
            for col in range(self.boxes2.shape[0]):
                xmin1, ymin1, xmax1, ymax1 = self.boxes1[row]
                xmin2, ymin2, xmax2, ymax2 = self.boxes2[col]
                area1 = (ymax1 - ymin1) * (xmax1 - xmin1)
                area2 = (ymax2 - ymin2) * (xmax2 - xmin2)
                inter_xmax = min(xmax1, xmax2)
                inter_ymax = min(ymax1, ymax2)
                inter_xmin = max(xmin1, xmin2)
                inter_ymin = max(ymin1, ymin2)
                inter_height = inter_ymax - inter_ymin
                inter_width = inter_xmax - inter_xmin
                inter_height = max(inter_height, 0)
                inter_width = max(inter_width, 0)
                inter_area = inter_width * inter_height
                union_area = area1 + area2 - inter_area
                sim_score = inter_area / union_area
                self.output[row, col] = sim_score
W
wanghaox 已提交
50 51 52
        self.inputs = {'X': self.boxes1, 'Y': self.boxes2}

        self.outputs = {'Out': self.output}
W
wanghaox 已提交
53 54


55 56 57 58 59 60
class TestIOUSimilarityOpWithLoD(TestIOUSimilarityOp):
    def test_check_output(self):
        self.check_output()

    def setUp(self):
        super(TestIOUSimilarityOpWithLoD, self).setUp()
61 62
        self.boxes1_lod = [[1, 1]]
        self.output_lod = [[1, 1]]
63 64 65 66 67

        self.inputs = {'X': (self.boxes1, self.boxes1_lod), 'Y': self.boxes2}
        self.outputs = {'Out': (self.output, self.output_lod)}


W
wanghaox 已提交
68 69
if __name__ == '__main__':
    unittest.main()