提交 da6636dc 编写于 作者: W wangjingyeye

add polygons

上级 72f93cf9
...@@ -73,7 +73,6 @@ class DBPostProcess(object): ...@@ -73,7 +73,6 @@ class DBPostProcess(object):
epsilon = 0.002 * cv2.arcLength(contour, True) epsilon = 0.002 * cv2.arcLength(contour, True)
approx = cv2.approxPolyDP(contour, epsilon, True) approx = cv2.approxPolyDP(contour, epsilon, True)
points = approx.reshape((-1, 2)) points = approx.reshape((-1, 2))
# print(points)
if points.shape[0] < 4: if points.shape[0] < 4:
continue continue
...@@ -88,7 +87,6 @@ class DBPostProcess(object): ...@@ -88,7 +87,6 @@ class DBPostProcess(object):
else: else:
continue continue
box = box.reshape(-1, 2) box = box.reshape(-1, 2)
# print(box)
_, sside = self.get_mini_boxes(box.reshape((-1, 1, 2))) _, sside = self.get_mini_boxes(box.reshape((-1, 1, 2)))
if sside < self.min_size + 2: if sside < self.min_size + 2:
...@@ -101,7 +99,6 @@ class DBPostProcess(object): ...@@ -101,7 +99,6 @@ class DBPostProcess(object):
np.round(box[:, 1] / height * dest_height), 0, dest_height) np.round(box[:, 1] / height * dest_height), 0, dest_height)
boxes.append(box.tolist()) boxes.append(box.tolist())
scores.append(score) scores.append(score)
# print(boxes)
return boxes, scores return boxes, scores
def boxes_from_bitmap(self, pred, _bitmap, dest_width, dest_height): def boxes_from_bitmap(self, pred, _bitmap, dest_width, dest_height):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册