Support rotation in detector results (cvat-ai#6961)

retailnext · Oct 25, 2023 · 22dfe07 · 22dfe07
1 parent bf22f8d
commit 22dfe07
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/...-ui/src/components/annotation-page/standard-workspace/controls-side-bar/tools-control.tsx b/...-ui/src/components/annotation-page/standard-workspace/controls-side-bar/tools-control.tsx
@@ -1239,6 +1239,9 @@ export class ToolsControlComponent extends React.PureComponent<Props, State> {
                                     objectType: ObjectType.SHAPE,
                                     frame,
                                     occluded: false,
+                                    rotation: [
+                                        ShapeType.RECTANGLE, ShapeType.ELLIPSE,
+                                    ].includes(data.type) ? (data.rotation || 0) : 0,
                                     source: core.enums.Source.AUTO,
                                     attributes: (data.attributes as { name: string, value: string }[])
                                         .reduce((acc, attr) => {

diff --git a/cvat/apps/lambda_manager/views.py b/cvat/apps/lambda_manager/views.py
@@ -627,6 +627,9 @@ def reset(self):
                         "source": "auto"
                     }
 
+                    if shape["type"] in ("rectangle", "ellipse"):
+                        shape["rotation"] = anno.get("rotation", 0)
+
                     if anno["type"] == "mask" and "points" in anno and conv_mask_to_poly:
                         shape["type"] = "polygon"
                         shape["points"] = anno["points"]