MVTec-FS Dataset
收藏资源简介:
MVTec-FS数据集是MVTec AD数据集的改进版本,专门为少样本学习研究设计。当前异常检测任务主要关注缺陷定位,而针对缺陷类型分类的数据集较少。为了解决这一问题,我们对MVTec AD数据集进行了重新处理,创建了一个适合少样本缺陷分类任务的基准。此外,它还可用于少样本目标检测和统一多模态分类。
MVTec-FS dataset is an improved version of the MVTec AD dataset, specifically designed for few-shot learning research. Currently, most anomaly detection tasks prioritize defect localization, yet there is a scarcity of datasets focused on defect type classification. To address this research gap, we reprocessed the MVTec AD dataset to develop a benchmark suitable for few-shot defect classification tasks. Furthermore, this dataset can also be utilized for few-shot object detection and unified multimodal classification.
MVTec-FS 数据集概述
数据集简介
MVTec-FS 数据集是基于 MVTec AD 数据集改进的版本,专门用于少样本学习研究。该数据集旨在填补缺陷类型分类数据集的空白,适用于少样本缺陷分类任务、少样本目标检测和统一多模态分类。
主要特点
- 使用连通组件算法和手动细化从原始图像级注释创建实例级注释。
- 包含多类异常,并为每个缺陷实例提供校正后的类别标签。
- 适用于 1-shot、3-shot 和 5-shot 学习范式。
- 包含 14 个子数据集,代表具有不同异常特征的各种产品类别。
数据集结构
数据集包含每个子数据集的训练集和测试集。每个子数据集包括异常类别和实例数量,详细信息见 Table 1。
注释细节
- 原始粗糙的图像级注释被细化为实例级注释。
- 使用连通组件算法生成实例掩码。
- 通过手动审查和调整确保每个缺陷实例的类别标签准确。
使用方式
数据集格式
数据集按子数据集组织,每个子数据集包含训练和测试图像的独立目录。每张图像都有一个对应的 json 文件(labeme 格式)。
加载数据集
可以使用常见的深度学习库(如 PyTorch 或 TensorFlow)加载数据集。示例代码片段将在仓库中提供。
少样本学习实验
使用数据集进行少样本学习的步骤:
- 从训练集中采样创建支持集。
- 使用测试集作为查询集进行评估。
数据集支持:
- 1-shot 学习
- 3-shot 学习
- 5-shot 学习
引用
如果在研究中使用 MVTec-FS,请引用: MVREC: A General Few-shot Defect Classification Model Using Multi-View Region-Context (AAAI 2025)
许可证
遵循 MVTec AD,数据集在 CC BY-NC-SA 4.0 下分发。
联系方式
如有关于数据集的问题或疑问,请联系:
- 姓名:LYU Shuai
- 邮箱:shuai.lyu@connect.polyu.hk




