From fc975a3f7d229f10cc214b013829631cfce77444 Mon Sep 17 00:00:00 2001 From: Ben Hoff Date: Fri, 20 Sep 2019 16:20:44 -0400 Subject: [PATCH] added method docs to Auto Annotation inference.py --- cvat/apps/auto_annotation/inference.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/cvat/apps/auto_annotation/inference.py b/cvat/apps/auto_annotation/inference.py index 8d3442e6d838..ef0c02fe7b5c 100644 --- a/cvat/apps/auto_annotation/inference.py +++ b/cvat/apps/auto_annotation/inference.py @@ -38,7 +38,14 @@ def __init__(self): "tracks": [] } - def add_box(self, xtl, ytl, xbr, ybr, label, frame_number, attributes=None): + # https://stackoverflow.com/a/50928627/2701402 + def add_box(self, xtl: float, ytl: float, xbr: float, ybr: float, label: int, frame_number: int, attributes: dict=None): + """ + xtl - x coordinate, top left + ytl - y coordinate, top left + xbr - x coordinate, bottom right + ybr - y coordinate, bottom right + """ self.get_shapes().append({ "label": label, "frame": frame_number, @@ -47,17 +54,17 @@ def add_box(self, xtl, ytl, xbr, ybr, label, frame_number, attributes=None): "attributes": attributes or {}, }) - def add_points(self, points, label, frame_number, attributes=None): + def add_points(self, points: list, label: int, frame_number: int, attributes: dict=None): points = self._create_polyshape(points, label, frame_number, attributes) points["type"] = "points" self.get_shapes().append(points) - def add_polygon(self, points, label, frame_number, attributes=None): + def add_polygon(self, points: list, label: int, frame_number: int, attributes: dict=None): polygon = self._create_polyshape(points, label, frame_number, attributes) polygon["type"] = "polygon" self.get_shapes().append(polygon) - def add_polyline(self, points, label, frame_number, attributes=None): + def add_polyline(self, points: list, label: int, frame_number: int, attributes: dict=None): polyline = self._create_polyshape(points, label, frame_number, attributes) polyline["type"] = "polyline" self.get_shapes().append(polyline) @@ -69,7 +76,7 @@ def get_tracks(self): return self._results["tracks"] @staticmethod - def _create_polyshape(points, label, frame_number, attributes=None): + def _create_polyshape(points: list, label: int, frame_number: int, attributes: dict=None): return { "label": label, "frame": frame_number,