Skip to content

GitLab

"...git@developer.sourcefind.cn:dcuai/dlexamples.git" did not exist on "a32ffa95a400f8cc67c07d5b4788f82b670f351d"
Commit 7cce85cc authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix conflicts

parent e7ad27c3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/postprocess/db_postprocess.py
View file @ 7cce85cc
...@@ -140,7 +140,7 @@ class DBPostProcess(object): ...@@ -140,7 +140,7 @@ class DBPostProcess(object):
boxes_batch = [] boxes_batch = []
for batch_index in range(pred.shape[0]): for batch_index in range(pred.shape[0]):
height, width = shape_list[batch_index] src_h, src_w, ratio_h, ratio_w = shape_list[batch_index]
if self.dilation_kernel is not None: if self.dilation_kernel is not None:
mask = cv2.dilate( mask = cv2.dilate(
np.array(segmentation[batch_index]).astype(np.uint8), np.array(segmentation[batch_index]).astype(np.uint8),
...@@ -148,7 +148,7 @@ class DBPostProcess(object): ...@@ -148,7 +148,7 @@ class DBPostProcess(object):
else: else:
mask = segmentation[batch_index] mask = segmentation[batch_index]
boxes, scores = self.boxes_from_bitmap(pred[batch_index], mask, boxes, scores = self.boxes_from_bitmap(pred[batch_index], mask,
width, height) src_w, src_h)
boxes_batch.append({'points': boxes}) boxes_batch.append({'points': boxes})
return boxes_batch return boxes_batch
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment