File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -137,13 +137,14 @@ def process(self):
137
137
# in relative coordinates for mask/cropping
138
138
polygons = [coordinates_of_segment (region , page_image , page_coords )
139
139
for region in regions ]
140
- masks = [pil2array (polygon_mask (page_image , polygon )).astype (np .uint8 )
141
- for polygon in polygons ]
142
- for i , mask in enumerate (masks [num_texts :], num_texts ):
140
+ for i , polygon in enumerate (polygons [num_texts :], num_texts ):
143
141
# for non-text regions, extend mask by 3 pixels in each direction
144
142
# to ensure they do not leak components accidentally
145
143
# (accounts for bad cropping of such regions in GT):
146
- masks [i ] = filters .maximum_filter (mask , 7 )
144
+ polygon = Polygon (polygon ).buffer (3 ).exterior .coords [:- 1 ] # keep open
145
+ polygons [i ] = polygon
146
+ masks = [pil2array (polygon_mask (page_image , polygon )).astype (np .uint8 )
147
+ for polygon in polygons ]
147
148
for i , region in enumerate (regions ):
148
149
if i >= num_texts :
149
150
break # keep non-text regions unchanged
You can’t perform that action at this time.
0 commit comments