Skip to content

Commit cd2464f

Browse files
committed
ellipse and rbbox handling
1 parent c10c055 commit cd2464f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/superannotate/lib/app/analytics/aggregators.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ class DataAggregator:
126126
"cuboid": lambda annotation: annotation["points"],
127127
"comment": lambda annotation: annotation["points"],
128128
"point": lambda annotation: {"x": annotation["x"], "y": annotation["y"]},
129-
"annotation_type": lambda annotation: dict(
129+
"ellipse": lambda annotation: dict(
130130
cx=annotation["cx"],
131131
cy=annotation["cy"],
132132
rx=annotation["rx"],
@@ -135,7 +135,8 @@ class DataAggregator:
135135
),
136136
"tag": lambda annotation: None,
137137
"mask": lambda annotation: {"parts": annotation["parts"]},
138-
"template": lambda annotation : None
138+
"template": lambda annotation : None,
139+
"rbbox": lambda annotation: annotation["points"]
139140
}
140141

141142
def __init__(
@@ -169,7 +170,7 @@ def get_annotation_paths(self):
169170
if self.folder_names is None:
170171
self._set_annotation_suffix(self.project_root)
171172
for path in self.project_root.glob("*.json"):
172-
if path.is_file() and path.suffix == self._annotation_suffix:
173+
if path.is_file() and self._annotation_suffix in path.name:
173174
annotations_paths.append(path)
174175
elif path.is_dir() and path.name != "classes":
175176
annotations_paths.extend(

0 commit comments

Comments
 (0)