目标检测任务
目标检测任务用于识别图像中目标的类别和位置,并以边界框标记目标所在区域。训练前需要准备原始图像,以及与图像对应的已准备标注信息数据集;模型学习目标的特征和位置规律后,可用于识别和定位新的图像中的目标。
图像数据上传
目标检测训练需要准备两类数据:包含原始图像的文件夹,以及包含图像路径和标注信息的数据集。先在流程界面点击【导入数据】,选择【文件夹】。
选择导入图像文件夹
在【新增文件夹】窗口中选择待训练图像的 ZIP 压缩包,然后点击【确认】完成导入。本示例导入 train.zip。
导入训练图像压缩包
导入完成后,可在数据模块查看图像文件夹及其中的图片文件。
查看已导入的图像文件夹
接着再次点击【导入数据】,选择【上传】,导入包含图片路径列和已准备标注信息列的训练数据集。数据集支持 CSV、Excel 等格式。
选择上传数据集
选择训练数据文件(如 train.CSV)后,点击【创建】完成导入。
导入训练标注数据集
配置图像视图
打开训练数据集 train,在右上角点击【设置】,配置图像视图。
打开数据集设置
在「图像视图」区域选择已导入的图像文件夹,设置路径列(如 record_id),勾选【包含标注】并选择已导入数据集中的标注列(如 label)。将标注类型设置为【目标检测】,点击【自动检测类别】后确认类别列表,最后点击【保存】。
配置目标检测图像视图
保存后切换到「图像视图」,可查看带有检测框和类别标签的训练样本,确认图像和标注匹配正确。
查看目标检测图像视图
目标检测模型训练
回到流程画布,选中 train 数据集,在右侧「机器学习」中点击【目标检测】。
选择目标检测算子
在初始配置弹窗中,选择标签列和 train 数据集对应的图像文件夹,然后点击【确认】。
配置目标检测初始参数
进入项目「实验」模块的目标检测训练界面,可进行具体参数的配置,核心基础配置包含数据源、配置参数、评价指标等。
在数据源配置中,指定目标列、图片路径列,并关联原始图像的存储目录,完成训练数据的关联。界面右侧会同步展示训练数据预览和类别信息,用于确认训练数据配置是否正确。
配置目标检测训练数据源
训练集/测试集分为测试集拆分与模型训练两类参数,测试集拆分可自定义分割方法、训练集占比与随机种子,固定拆分结果以保证实验可复现。
配置训练集和测试集
在【配置参数】中,可选择预训练模型,并设置微调层数、优化器、初始学习率、学习率调度器、批量大小和训练轮数等参数。本示例将【微调层数】设置为 2,关闭【早停】。
配置目标检测训练参数
评价指标配置,可按需勾选 mAP、精确率、召回率、F1 Score、交并比(IOU)等多维度评估指标,用于训练后全面衡量模型目标检测效果。
配置目标检测评价指标
数据增强中可以选择是否启用随机翻转、随机旋转、随机裁剪、颜色扭曲和高斯模糊。
在高级配置中选择相应的运行环境。完成全部配置后,点击【训练】;在弹窗中可选择是否启用 GPU 加速训练,填写任务名称(必填)与任务描述,确认训练集、测试集数据量无误后,再次点击【训练】,即可启动目标检测模型训练任务。
训练过程中,系统将实时展示训练进度与性能曲线,包括验证集 mAP、训练集 / 验证集损失等核心指标,直观呈现模型收敛情况。训练完成后,界面将生成完整的训练结果报告,展示任务耗时、各评估指标(mAP、mAP (IoU=0.5)、mAP (IoU=0.75) 等)的最终数值,便于后续模型评估与优化。
查看目标检测训练结果
查看目标检测训练指标
查看目标检测训练任务
训练完成后,在「训练结果」页面,点击测试集右侧的【查看详情】按钮,即可进入目标检测模型详情页面。
在模型详情页面,可查看模型 ID、模型类型、目标变量、检测类别、训练开始时间、训练集行数、训练 / 验证集划分比例、预训练模型、微调层数、优化器、学习率调度器、初始学习率、最终学习率等训练参数,以及 mAP 等核心评估指标。
确认模型效果符合预期后,可在该页面完成模型部署,用于后续目标检测预测任务。
查看目标检测模型详情
部署目标检测模型
目标检测模型预测
部署完毕后,模型会在流程界面中出现。随后上传用于预测的图像文件夹和预测数据集。预测数据集只需包含图片路径列,例如 record_id;无需包含标注列或类别列。路径列中的值应能与预测图像文件夹中的图片对应。
查看目标检测预测数据集
点击训练好的模型,展开右侧算子列表,点击【预测】。在弹出的「新建预测」窗口中,完成输入配置与输出配置:输入配置需指定预测数据集、训练好的目标检测模型与预测图像文件夹;输出配置可自定义预测结果名称、存储路径与输出格式,确认后点击【确认】。
进入预测配置页面,可进一步配置预测参数,包括批量大小、是否使用 GPU、分数阈值、最大检测数等,按需调整后点击【运行】,即可启动目标检测预测任务。
预测完成后,可在数据模块中查看预测结果,支持以图片视图直观展示检测框、目标类别与置信度。
创建目标检测预测任务
配置目标检测预测任务
运行目标检测预测任务
查看目标检测预测结果
