You've already forked DataMate
feat(annotation): 自动标注任务支持非图像类型数据集(TEXT/AUDIO/VIDEO)
移除自动标注任务创建流程中的 IMAGE-only 限制,使 TEXT、AUDIO、VIDEO 类型数据集均可用于自动标注任务。 - 新增数据库迁移:t_dm_auto_annotation_tasks 表添加 dataset_type 列 - 后端 schema/API/service 全链路传递 dataset_type - Worker 动态构建 sample key(image/text/audio/video)和输出目录 - 前端移除数据集类型校验,下拉框显示数据集类型标识 - 输出数据集继承源数据集类型,不再硬编码为 IMAGE - 保持向后兼容:默认值为 IMAGE,worker 有元数据回退和目录 fallback Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -6,7 +6,7 @@ import { ArrowLeft } from "lucide-react";
|
||||
import { Link, useNavigate } from "react-router";
|
||||
import { queryDatasetsUsingGet } from "@/pages/DataManagement/dataset.api";
|
||||
import { mapDataset } from "@/pages/DataManagement/dataset.const";
|
||||
import { Dataset, DatasetType } from "@/pages/DataManagement/dataset.model";
|
||||
import { Dataset } from "@/pages/DataManagement/dataset.model";
|
||||
import { createAnnotationOperatorTaskUsingPost } from "../annotation.api";
|
||||
import { useCreateStepTwo } from "./hooks/useCreateStepTwo";
|
||||
import PipelinePreview from "./components/PipelinePreview";
|
||||
@@ -85,11 +85,6 @@ export default function AnnotationOperatorTaskCreate() {
|
||||
try {
|
||||
if (currentStep === 1) {
|
||||
await form.validateFields();
|
||||
|
||||
if (selectedDataset?.datasetType !== DatasetType.IMAGE) {
|
||||
message.error("自动标注算子编排当前仅支持图片数据集");
|
||||
return;
|
||||
}
|
||||
}
|
||||
setCurrentStep((prev) => Math.min(prev + 1, 2));
|
||||
} catch {
|
||||
@@ -109,11 +104,6 @@ export default function AnnotationOperatorTaskCreate() {
|
||||
return;
|
||||
}
|
||||
|
||||
if (selectedDataset?.datasetType !== DatasetType.IMAGE) {
|
||||
message.error("自动标注算子编排当前仅支持图片数据集");
|
||||
return;
|
||||
}
|
||||
|
||||
const outputDatasetName = values.outputDatasetName?.trim();
|
||||
const pipeline = selectedOperators.map((operator, index) => {
|
||||
const overrides = {
|
||||
@@ -200,10 +190,10 @@ export default function AnnotationOperatorTaskCreate() {
|
||||
label="选择数据集"
|
||||
name="datasetId"
|
||||
rules={[{ required: true, message: "请选择数据集" }]}
|
||||
extra="自动标注算子编排当前仅支持图片数据集"
|
||||
extra="请选择用于自动标注的数据集"
|
||||
>
|
||||
<Select
|
||||
placeholder="请选择图片数据集"
|
||||
placeholder="请选择数据集"
|
||||
optionFilterProp="label"
|
||||
options={datasets.map((dataset) => ({
|
||||
label: (
|
||||
@@ -215,12 +205,11 @@ export default function AnnotationOperatorTaskCreate() {
|
||||
{dataset.name}
|
||||
</div>
|
||||
<div className="text-xs text-gray-500">
|
||||
{dataset?.fileCount} 文件 • {dataset.size}
|
||||
{dataset.datasetType} • {dataset?.fileCount} 文件 • {dataset.size}
|
||||
</div>
|
||||
</div>
|
||||
),
|
||||
value: dataset.id,
|
||||
disabled: dataset.datasetType !== DatasetType.IMAGE,
|
||||
}))}
|
||||
/>
|
||||
</Form.Item>
|
||||
|
||||
Reference in New Issue
Block a user