diff --git a/runtime/datamate-python/app/module/annotation/service/builtin_templates.py b/runtime/datamate-python/app/module/annotation/service/builtin_templates.py index 3c3b9e0..7c9ec8b 100644 --- a/runtime/datamate-python/app/module/annotation/service/builtin_templates.py +++ b/runtime/datamate-python/app/module/annotation/service/builtin_templates.py @@ -17,6 +17,15 @@ CATEGORY_COMPUTER_VISION = "computer-vision" STYLE_HORIZONTAL = "horizontal" VERSION_DEFAULT = "1.0.0" +IMAGE_CLASSIFICATION_LABEL_CONFIG = """ + + + + + + +""" + OBJECT_DETECTION_LABEL_CONFIG = """ @@ -57,6 +66,20 @@ class BuiltInTemplateDefinition: BUILT_IN_TEMPLATES: List[BuiltInTemplateDefinition] = [ + BuiltInTemplateDefinition( + id="tpl-image-classification-001", + name="图像分类", + description=( + "对图像进行分类,适用于内容审核、安全检测、社交媒体审核等场景。" + "关联模型:ResNet、EfficientNet、Vision Transformer" + ), + data_type=DATA_TYPE_IMAGE, + labeling_type="image-classification", + label_config=IMAGE_CLASSIFICATION_LABEL_CONFIG, + style=STYLE_HORIZONTAL, + category=CATEGORY_COMPUTER_VISION, + version=VERSION_DEFAULT, + ), BuiltInTemplateDefinition( id="tpl-object-detection-001", name="目标检测(边界框)",