Skip to content

Removing segmentations with only one point #58

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
george-gca wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Removing segmentations with only one point #58

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
0559c9a
Fixed the pycocotools workaround
george-gca Feb 25, 2022
f5a52ec
Fixed infinite loop bug
george-gca Feb 27, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion imantics/annotation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -284,12 +284,14 @@ def coco(self, include=True):
if len(annotation['segmentation'][i]) == 2:
# discard segmentation that is only a point
annotation['segmentation'].pop(i)
i -= 1
elif len(annotation['segmentation'][i]) == 4:
# create another point in the middle of segmentation to
# avoid bug when using pycocotools, which thinks that a
# 4 value segmentation mask is a bounding box
polygon = annotation['segmentation'][i]
x1, y1, x2, y2 = polygon

if x2 == x1:
x = x1
y = (y2 + y1) // 2
Expand All @@ -304,7 +306,8 @@ def coco(self, include=True):

new_segmentation = polygon[:2] + [x, y] + polygon[2:]
annotation['segmentation'][i] = new_segmentation
i += 1

i += 1

if include:
image = category = {}
Expand Down