def from_json(cls, data, project_meta, key_id_map: KeyIdMap=None):
'''
The function from_json convert videoannotation from json format to VideoAnnotation class object.
:param data: input videoannotation in json format
:param project_meta: ProjectMeta class object
:param key_id_map: KeyIdMap class object
:return: VideoAnnotation class object
'''
#video_name = data[VIDEO_NAME]
video_key = uuid.UUID(data[KEY]) if KEY in data else uuid.uuid4()
if key_id_map is not None:
key_id_map.add_video(video_key, data.get(VIDEO_ID, None))
img_size_dict = data[IMG_SIZE]
img_height = img_size_dict[IMG_SIZE_HEIGHT]
img_width = img_size_dict[IMG_SIZE_WIDTH]
img_size = (img_height, img_width)
description = data.get(DESCRIPTION, "")
frames_count = data[FRAMES_COUNT]
tags = VideoTagCollection.from_json(data[TAGS], project_meta.tag_metas, key_id_map)
objects = VideoObjectCollection.from_json(data[OBJECTS], project_meta, key_id_map)
frames = FrameCollection.from_json(data[FRAMES], objects, frames_count, key_id_map)
return cls(img_size=img_size,
frames_count=frames_count,
objects=objects,
frames=frames,
tags=tags,
description=description,
key=video_key)
def from_json(cls, data, objects: VideoObjectCollection, frame_index, key_id_map: KeyIdMap = None):
object_id = data.get(OBJECT_ID, None)
object_key = None
if OBJECT_KEY in data:
object_key = uuid.UUID(data[OBJECT_KEY])
if object_id is None and object_key is None:
raise RuntimeError("Figure can not be deserialized from json: object_id or object_key are not found")
if object_key is None:
if key_id_map is None:
raise RuntimeError("Figure can not be deserialized: key_id_map is None")
object_key = key_id_map.get_object_key(object_id)
if object_key is None:
raise RuntimeError("Object with id={!r} not found in key_id_map".format(object_id))
object = objects.get(object_key)
if object is None:
raise RuntimeError("Figure can not be deserialized: corresponding object {!r} not found in ObjectsCollection".format(object_key.hex))
shape_str = data[ApiField.GEOMETRY_TYPE]
geometry_json = data[ApiField.GEOMETRY]
shape = GET_GEOMETRY_FROM_STR(shape_str)
geometry = shape.from_json(geometry_json)
key = uuid.UUID(data[KEY]) if KEY in data else uuid.uuid4()
if key_id_map is not None:
key_id_map.add_figure(key, data.get(ID, None))
return cls(object, geometry, frame_index, key)
def __init__(self,
objects=None,
figures=None,
tags=None,
description="",
key=None):
self._description = description
self._tags = take_with_default(tags, VideoTagCollection())
self._objects = take_with_default(objects, VideoObjectCollection())
self._figures = take_with_default(figures, [])
self._key = take_with_default(key, uuid.uuid4())
def from_json(cls, data, objects: VideoObjectCollection, frame_index, key_id_map: KeyIdMap = None):
'''
The function from_json convert VideoFigure from json format to VideoFigure class object.
:param data: input VideoFigure in json format
:param objects: VideoObjectCollection
:param frame_index: int
:param key_id_map: KeyIdMap class object
:return: VideoFigure class object
'''
object_id = data.get(OBJECT_ID, None)
object_key = None
if OBJECT_KEY in data:
object_key = uuid.UUID(data[OBJECT_KEY])
if object_id is None and object_key is None:
raise RuntimeError("Figure can not be deserialized from json: object_id or object_key are not found")
if object_key is None:
if key_id_map is None:
raise RuntimeError("Figure can not be deserialized: key_id_map is None")
object_key = key_id_map.get_object_key(object_id)
if object_key is None:
raise RuntimeError("Object with id={!r} not found in key_id_map".format(object_id))
object = objects.get(object_key)
if object is None:
raise RuntimeError("Figure can not be deserialized: corresponding object {!r} not found in ObjectsCollection".format(object_key.hex))
shape_str = data[ApiField.GEOMETRY_TYPE]
geometry_json = data[ApiField.GEOMETRY]
shape = GET_GEOMETRY_FROM_STR(shape_str)
geometry = shape.from_json(geometry_json)
key = uuid.UUID(data[KEY]) if KEY in data else uuid.uuid4()
if key_id_map is not None:
key_id_map.add_figure(key, data.get(ID, None))
class_id = data.get(CLASS_ID, None)
labeler_login = data.get(LABELER_LOGIN, None)
updated_at = data.get(UPDATED_AT, None)
created_at = data.get(CREATED_AT, None)
return cls(object, geometry, frame_index, key,
class_id=class_id, labeler_login=labeler_login, updated_at=updated_at, created_at=created_at)
def __init__(self,
objects=None,
figures=None,
tags=None,
description="",
key=None):
'''
:param objects: VideoObjectCollection
:param figures: list of figures(Point, Cuboid, etc)
:param tags: VideoTagCollection
:param description: str
:param key: uuid class object
'''
self._description = description
self._tags = take_with_default(tags, VideoTagCollection())
self._objects = take_with_default(objects, VideoObjectCollection())
self._figures = take_with_default(figures, [])
self._key = take_with_default(key, uuid.uuid4())
def __init__(self,
img_size,
frames_count,
objects=None,
frames=None,
tags=None,
description="",
key=None):
if not isinstance(img_size, (tuple, list)):
raise TypeError(
'{!r} has to be a tuple or a list. Given type "{}".'.format(
'img_size', type(img_size)))
self._img_size = tuple(img_size)
self._frames_count = frames_count
self._description = description
self._tags = take_with_default(tags, VideoTagCollection())
self._objects = take_with_default(objects, VideoObjectCollection())
self._frames = take_with_default(frames, FrameCollection())
self._key = take_with_default(key, uuid.uuid4())
self.validate_figures_bounds()
def __init__(self, img_size, frames_count, objects=None, frames=None, tags=None, description="", key=None):
'''
The constructor for VideoAnnotation class.
:param img_size: size of the image(tuple or list of integers)
:param frames_count: int
:param objects: VideoObjectCollection
:param frames: FrameCollection
:param tags: VideoTagCollection
:param description: str
:param key: uuid class object
'''
if not isinstance(img_size, (tuple, list)):
raise TypeError('{!r} has to be a tuple or a list. Given type "{}".'.format('img_size', type(img_size)))
self._img_size = tuple(img_size)
self._frames_count = frames_count
self._description = description
self._tags = take_with_default(tags, VideoTagCollection())
self._objects = take_with_default(objects, VideoObjectCollection())
self._frames = take_with_default(frames, FrameCollection())
self._key = take_with_default(key, uuid.uuid4())
self.validate_figures_bounds()