Skip to content

Commit f1deb46

Browse files
committed
fix: do not return empty segments
1 parent fe96aa9 commit f1deb46

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

decimer_segmentation/decimer_segmentation.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,10 @@ def segment_chemical_structures(
114114
if len(segments) > 0:
115115
segments, bboxes = sort_segments_bboxes(segments, bboxes)
116116

117+
segments = [segment for segment in segments
118+
if segment.shape[0] > 0
119+
if segment.shape[1] > 0]
120+
117121
return segments
118122

119123

@@ -235,6 +239,7 @@ def get_expanded_masks(image: np.array) -> np.array:
235239
image_array=image,
236240
mask_array=masks,
237241
max_depiction_size=size,
242+
debug=False
238243
)
239244
return expanded_masks
240245

0 commit comments

Comments
 (0)