Spaces:
Runtime error
Runtime error
File size: 5,537 Bytes
2366e36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
# Copyright (c) OpenMMLab. All rights reserved.
import numpy as np
import pytest
from mmocr.utils import (bezier_to_polygon, is_on_same_line, sort_points,
stitch_boxes_into_lines)
def test_box_on_line():
# regular boxes
box1 = [0, 0, 1, 0, 1, 1, 0, 1]
box2 = [2, 0.5, 3, 0.5, 3, 1.5, 2, 1.5]
box3 = [4, 0.8, 5, 0.8, 5, 1.8, 4, 1.8]
assert is_on_same_line(box1, box2, 0.5)
assert not is_on_same_line(box1, box3, 0.5)
# irregular box4
box4 = [0, 0, 1, 1, 1, 2, 0, 1]
box5 = [2, 1.5, 3, 1.5, 3, 2.5, 2, 2.5]
box6 = [2, 1.6, 3, 1.6, 3, 2.6, 2, 2.6]
assert is_on_same_line(box4, box5, 0.5)
assert not is_on_same_line(box4, box6, 0.5)
def test_stitch_boxes_into_lines():
boxes = [ # regular boxes
[0, 0, 1, 0, 1, 1, 0, 1],
[2, 0.5, 3, 0.5, 3, 1.5, 2, 1.5],
[3, 1.2, 4, 1.2, 4, 2.2, 3, 2.2],
[5, 0.5, 6, 0.5, 6, 1.5, 5, 1.5],
# irregular box
[6, 1.5, 7, 1.25, 7, 1.75, 6, 1.75]
]
raw_input = [{'box': boxes[i], 'text': str(i)} for i in range(len(boxes))]
result = stitch_boxes_into_lines(raw_input, 1, 0.5)
# Final lines: [0, 1], [2], [3, 4]
# box 0, 1, 3, 4 are on the same line but box 3 is 2 pixels away from box 1
# box 3 and 4 are on the same line since the length of overlapping part >=
# 0.5 * the y-axis length of box 5
expected_result = [{
'box': [0, 0, 3, 0, 3, 1.5, 0, 1.5],
'text': '0 1'
}, {
'box': [3, 1.2, 4, 1.2, 4, 2.2, 3, 2.2],
'text': '2'
}, {
'box': [5, 0.5, 7, 0.5, 7, 1.75, 5, 1.75],
'text': '3 4'
}]
result.sort(key=lambda x: x['box'][0])
expected_result.sort(key=lambda x: x['box'][0])
assert result == expected_result
def test_bezier_to_polygon():
bezier_points = [
37.0, 249.0, 72.5, 229.55, 95.34, 220.65, 134.0, 216.0, 132.0, 233.0,
82.11, 240.2, 72.46, 247.16, 38.0, 263.0
]
pts = bezier_to_polygon(bezier_points)
target = np.array([[37.0, 249.0], [42.50420761043885, 246.01570199737577],
[47.82291296107305, 243.2012392477038],
[52.98102930456334, 240.5511007435486],
[58.00346989357049, 238.05977547747486],
[62.91514798075522, 235.721752442047],
[67.74097681877824, 233.53152062982943],
[72.50586966030032, 231.48356903338674],
[77.23473975798221, 229.57238664528356],
[81.95250036448464, 227.79246245808432],
[86.68406473246829, 226.13828546435346],
[91.45434611459396, 224.60434465665548],
[96.28825776352238, 223.18512902755504],
[101.21071293191426, 221.87512756961655],
[106.24662487243039, 220.6688292754046],
[111.42090683773145, 219.5607231374836],
[116.75847208047819, 218.5452981484181],
[122.28423385333137, 217.6170433007727],
[128.02310540895172, 216.77044758711182],
[134.0, 216.0], [132.0, 233.0],
[124.4475521213005, 234.13617728531858],
[117.50700976818779, 235.2763434903047],
[111.12146960198277, 236.42847645429362],
[105.2340282840064, 237.6005540166205],
[99.78778247557953, 238.80055401662054],
[94.72582883802303, 240.0364542936288],
[89.99126403265781, 241.31623268698053],
[85.52718472080478, 242.64786703601104],
[81.27668756378483, 244.03933518005545],
[77.1828692229188, 245.49861495844874],
[73.18882635952762, 247.0336842105263],
[69.23765563493221, 248.65252077562326],
[65.27245371045342, 250.3631024930748],
[61.23631724741216, 252.17340720221605],
[57.07234290712931, 254.09141274238226],
[52.723627350925796, 256.12509695290856],
[48.13326724012247, 258.2824376731302],
[43.24435923604024, 260.5714127423822], [38.0, 263.0]])
assert np.allclose(pts, target)
bezier_points = [0, 0, 0, 1, 0, 2, 0, 3, 1, 0, 1, 1, 1, 2, 1, 3]
pts = bezier_to_polygon(bezier_points, num_sample=3)
target = np.array([[0, 0], [0, 1.5], [0, 3], [1, 0], [1, 1.5], [1, 3]])
assert np.allclose(pts, target)
with pytest.raises(AssertionError):
bezier_to_polygon(bezier_points, num_sample=-1)
bezier_points = [0, 1]
with pytest.raises(AssertionError):
bezier_to_polygon(bezier_points)
def test_sort_points():
points = np.array([[1, 1], [0, 0], [1, -1], [2, -2], [0, 2], [1, 1],
[0, 1], [-1, 1], [-1, -1]])
target = np.array([[-1, -1], [0, 0], [-1, 1], [0, 1], [0, 2], [1, 1],
[1, 1], [2, -2], [1, -1]])
assert np.allclose(target, sort_points(points))
points = np.array([[1, 1], [1, -1], [-1, 1], [-1, -1]])
target = np.array([[-1, -1], [-1, 1], [1, 1], [1, -1]])
assert np.allclose(target, sort_points(points))
points = [[1, 1], [1, -1], [-1, 1], [-1, -1]]
assert np.allclose(target, sort_points(points))
with pytest.raises(AssertionError):
sort_points([1, 2])
|