Fix automatic tag annotation support (#7839)

cvat-ai · May 30, 2024 · 483502a · 483502a
1 parent da71018
commit 483502a
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/changelog.d/20240528_180958_omer.sarioglu_semi_auto_tag_annotation.md b/changelog.d/20240528_180958_omer.sarioglu_semi_auto_tag_annotation.md
@@ -0,0 +1,4 @@
+### Fixed
+
+- A classifier model can not be used on annotation view (unknown object shape error)
+  (<https://github.com/cvat-ai/cvat/pull/7839>)
diff --git a/cvat-core/src/lambda-manager.ts b/cvat-core/src/lambda-manager.ts
@@ -1,5 +1,5 @@
 // Copyright (C) 2019-2022 Intel Corporation
-// Copyright (C) 2022-2023 CVAT.ai Corporation
+// Copyright (C) 2022-2024 CVAT.ai Corporation
 //
 // SPDX-License-Identifier: MIT
 
@@ -21,7 +21,7 @@ export interface InteractorResults {
 }
 
 export interface DetectedShape {
-    type: ShapeType;
+    type: ShapeType | 'tag';
     rotation?: number;
     attributes: { name: string; value: string }[];
     label: string;

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
@@ -1242,6 +1242,16 @@ export class ToolsControlComponent extends React.PureComponent<Props, State> {
 
                                 if (!jobLabel) return null;
 
+                                if (data.type === 'tag') {
+                                    return new core.classes.ObjectState({
+                                        frame,
+                                        label: jobLabel,
+                                        attributes: loadAttributes(data.attributes, jobLabel),
+                                        objectType: ObjectType.TAG,
+                                        source: core.enums.Source.AUTO,
+                                    });
+                                }
+
                                 const objectData = {
                                     label: jobLabel,
                                     objectType: ObjectType.SHAPE,