forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2060 from LDOUBLEV/fix2013 

fix issue 2013
This commit is contained in:
Double_V 2021-02-22 09:57:04 +08:00 committed by GitHub
parent 816471aa42 8f9dda8608
commit 6a42745f96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
ppocr/data/imaug/make_shrink_map.py
View File

@ -44,21 +44,34 @@ class MakeShrinkMap(object):
ignore_tags[i] = True ignore_tags[i] = True
else: else:
polygon_shape = Polygon(polygon) polygon_shape = Polygon(polygon)
distance = polygon_shape.area * ( subject = [tuple(l) for l in polygon]
1 - np.power(self.shrink_ratio, 2)) / polygon_shape.length
subject = [tuple(l) for l in text_polys[i]]
padding = pyclipper.PyclipperOffset() padding = pyclipper.PyclipperOffset()
padding.AddPath(subject, pyclipper.JT_ROUND, padding.AddPath(subject, pyclipper.JT_ROUND,
pyclipper.ET_CLOSEDPOLYGON) pyclipper.ET_CLOSEDPOLYGON)
shrinked = []
# Increase the shrink ratio every time we get multiple polygon returned back
possible_ratios = np.arange(self.shrink_ratio, 1,
self.shrink_ratio)
np.append(possible_ratios, 1)
# print(possible_ratios)
for ratio in possible_ratios:
# print(f"Change shrink ratio to {ratio}")
distance = polygon_shape.area * (
1 - np.power(ratio, 2)) / polygon_shape.length
shrinked = padding.Execute(-distance) shrinked = padding.Execute(-distance)
if len(shrinked) == 1:
break
if shrinked == []: if shrinked == []:
cv2.fillPoly(mask, cv2.fillPoly(mask,
polygon.astype(np.int32)[np.newaxis, :, :], 0) polygon.astype(np.int32)[np.newaxis, :, :], 0)
ignore_tags[i] = True ignore_tags[i] = True
continue continue
shrinked = np.array(shrinked[0]).reshape(-1, 2)
cv2.fillPoly(gt, [shrinked.astype(np.int32)], 1) for each_shirnk in shrinked:
# cv2.fillPoly(gt[0], [shrinked.astype(np.int32)], 1) shirnk = np.array(each_shirnk).reshape(-1, 2)
cv2.fillPoly(gt, [shirnk.astype(np.int32)], 1)
data['shrink_map'] = gt data['shrink_map'] = gt
data['shrink_mask'] = mask data['shrink_mask'] = mask