Added keypoint detection to supported import projects

A-Artemis · A-Artemis · commit cb9f000bb5f6 · 2025-05-15T15:57:20.000+02:00
diff --git a/interactive_ai/services/dataset_ie/app/communication/helpers/import_utils.py b/interactive_ai/services/dataset_ie/app/communication/helpers/import_utils.py
@@ -28,6 +28,7 @@
     Domain.ANOMALY_DETECTION,
     Domain.ANOMALY_SEGMENTATION,
     Domain.ROTATED_DETECTION,
+    Domain.KEYPOINT_DETECTION,
 ]
 
 
@@ -142,6 +143,7 @@ def get_validated_task_type(cls, project: Project) -> TaskType:
             TaskType.ANOMALY_DETECTION,
             TaskType.ANOMALY_SEGMENTATION,
             TaskType.ROTATED_DETECTION,
+            TaskType.KEYPOINT_DETECTION,
         ]
 
         trainable_tasks = project.get_trainable_task_nodes()
diff --git a/interactive_ai/workflows/geti_domain/common/jobs_common_extras/datumaro_conversion/import_utils.py b/interactive_ai/workflows/geti_domain/common/jobs_common_extras/datumaro_conversion/import_utils.py
@@ -70,6 +70,7 @@ class ImportErrorDetail:
     Domain.ANOMALY_DETECTION,
     Domain.ANOMALY_SEGMENTATION,
     Domain.ROTATED_DETECTION,
+    Domain.KEYPOINT_DETECTION,
 ]
 
 
@@ -839,6 +840,7 @@ def get_validated_task_type(cls, project: Project) -> TaskType:
             TaskType.ANOMALY_DETECTION,
             TaskType.ANOMALY_SEGMENTATION,
             TaskType.ROTATED_DETECTION,
+            TaskType.KEYPOINT_DETECTION,
         ]
 
         trainable_tasks = project.get_trainable_task_nodes()

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@`
`28`	`28`	`Domain.ANOMALY_DETECTION,`
`29`	`29`	`Domain.ANOMALY_SEGMENTATION,`
`30`	`30`	`Domain.ROTATED_DETECTION,`
	`31`	`+ Domain.KEYPOINT_DETECTION,`
`31`	`32`	`]`
`32`	`33`
`33`	`34`
`@@ -142,6 +143,7 @@ def get_validated_task_type(cls, project: Project) -> TaskType:`
`142`	`143`	`TaskType.ANOMALY_DETECTION,`
`143`	`144`	`TaskType.ANOMALY_SEGMENTATION,`
`144`	`145`	`TaskType.ROTATED_DETECTION,`
	`146`	`+ TaskType.KEYPOINT_DETECTION,`
`145`	`147`	`]`
`146`	`148`
`147`	`149`	`trainable_tasks = project.get_trainable_task_nodes()`
Original file line number	Diff line number	Diff line change
`@@ -70,6 +70,7 @@ class ImportErrorDetail:`
`70`	`70`	`Domain.ANOMALY_DETECTION,`
`71`	`71`	`Domain.ANOMALY_SEGMENTATION,`
`72`	`72`	`Domain.ROTATED_DETECTION,`
	`73`	`+ Domain.KEYPOINT_DETECTION,`
`73`	`74`	`]`
`74`	`75`
`75`	`76`
`@@ -839,6 +840,7 @@ def get_validated_task_type(cls, project: Project) -> TaskType:`
`839`	`840`	`TaskType.ANOMALY_DETECTION,`
`840`	`841`	`TaskType.ANOMALY_SEGMENTATION,`
`841`	`842`	`TaskType.ROTATED_DETECTION,`
	`843`	`+ TaskType.KEYPOINT_DETECTION,`
`842`	`844`	`]`
`843`	`845`
`844`	`846`	`trainable_tasks = project.get_trainable_task_nodes()`