@@ -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