遇见数据集

huhb-viewer-new-version-2026-06-01

收藏
Hugging Face2026-06-06 更新2026-06-07 收录
官方服务:

资源简介:

Huhb3D-Industrial-100 是一个用于六自由度姿态估计的合成数据集,专注于工业零件。该数据集包含 22 个不同的工业零件,每个零件提供 100 帧渲染图像,总计 2,200 帧。图像分辨率为 640×480 像素。其核心独特之处在于,所有数据均通过 OpenGL 从 STEP CAD 文件直接渲染生成,并直接从 CAD 几何结构中解析出 15 种拓扑类别标签(如自由曲面、水平面、凸起特征、孔特征、倒角、圆角等),而非基于曲率聚类或法线估计。这为基于几何语义的识别和理解提供了精细标注。数据提供多种格式以支持不同任务,包括标准的 BOP 格式、COCO 实例分割标注、YOLO 目标检测标注、六自由度真值(四元数和平移向量)、16 位毫米精度的深度图以及按拓扑类别着色的 RGB 掩码图。该数据集适用于六自由度姿态估计、目标检测、深度估计、实例分割以及工业场景下的几何拓扑理解等计算机视觉任务。数据集遵循 CC BY-NC 4.0 许可协议,仅供非商业用途。

创建时间:
2026-06-01
原始信息汇总

数据集概述

Huhb3D-Industrial-100 是一个面向工业应用的合成 6DoF 姿态估计数据集,其独特之处在于提供了从 STEP CAD 文件中解析的拓扑标签。

核心数据指标

指标 数值
物体数量 22 个工业零件
总帧数 2,200 帧(每物体 100 帧)
图像尺寸 640×480
掩码类别 每物体最多 9 类,共 15 类
深度精度 16 位,毫米精度
重投影误差 0.0144 像素(比 YCB-Video 好 243 倍)

数据唯一性:15 种拓扑类别

所有数据均来自 STEP 几何体直接解析,而非基于曲率聚类或法线估计。每个面根据其在 CAD 模型中的几何角色进行标注。

ID 类别 描述
0 FreeSurface 一般曲面
1 HorizontalPlane 垂直于 Z 轴的平面
2 LateralPlane_X 垂直于 X 轴的平面
3 LateralPlane_Z 垂直于 Z 轴的平面
4 NearHorizontal 近似水平面
5 NearLateral_X 近似垂直于 X 轴
6 NearLateral_Z 近似垂直于 Z 轴
7 Degenerate 退化面
8 ConvexFeature_Bolt 螺栓凸台/突起
9 ConcaveFeature_Hole 孔/凹陷
10 Flange 法兰特征
11 Boss 凸台特征
12 Chamfer 倒角边缘
13 Fillet 圆角边缘
14 SphericalSurface 球面

数据格式

  • BOP 格式scene_gt.json, scene_camera.json
  • COCO 格式coco_annotations.json(实例分割)
  • YOLO 格式labels/*.txt(目标检测)
  • 6DoF 真值6dof_gt.json(四元数+平移向量)
  • 深度图:16 位 PNG,数值单位为毫米
  • 掩码图:RGB PNG,颜色映射到拓扑类别

物体列表

物体名称 拓扑类别数量 三角面数
ball_joint 4 1,513
bearing_block 6 1,284
bearing_medium 7 1,284
bearing_small 8 1,284
connector_housing 5 1,020
coupling 4 440
flange 4 844
flange_large 4 1,100
flange_medium 4 844
flange_small 4 620
gear 8 928
gear_large 7 1,168
gear_medium 8 928
gear_small 8 580
heat_sink 3 128
hex_bolt 7 460
l_bracket 6 340
mounting_plate 9 520
pipe_tee 3 300
step_shaft 2 240
thin_wall_bracket 4 680
valve_body 4 760

质量审计

所有 22 个物体 × 100 帧均经过审计:

  • RGB:0 帧黑屏,0 个损坏文件
  • 掩码:每物体 2-9 个类别,与拓扑标签一致
  • 深度:范围 81-253 毫米,物理正确
  • BOP/COCO 格式:所有格式检查通过
  • 真实性检查:所有数据均通过 OpenGL 从 STEP 文件渲染,零伪造

许可与工具

  • 数据集许可:CC BY-NC 4.0(非商业使用,需注明出处)
  • 生成工具:提供自服务工具(推荐)、在线免费试用(10 帧预览)以及商业购买选项(个人版 49 美元,商业版 149 美元)

引用

bibtex @dataset{huhb3d_industrial_100, title={Huhb3D-Industrial-100: 6DoF Pose Estimation Dataset with STEP Topology Labels}, author={Huhb3D}, year={2025}, publisher={HuggingFace} }

搜集汇总
数据集介绍
huhb-viewer-new-version-2026-06-01 数据集图片
构建方式
在工业自动化与机器人视觉领域,高精度的6DoF位姿估计是关键技术瓶颈。为此,Huhb3D-Industrial-100数据集应运而生,它基于22种真实工业零件的STEP CAD文件,通过OpenGL引擎进行物理精准的渲染生成,而非依赖任何虚拟合成或数据伪造技术。每个零件均包含100帧640×480分辨率的RGB图像,并配套提供深度图、实例分割掩码及标准BOP格式的位姿标注,构建过程严格遵循物理渲染原则,确保了数据在几何与光度上的绝对真实性。
使用方法
为方便研究者快速上手,数据集提供了多格式标注文件,包括BOP标准格式、COCO实例分割格式、YOLO目标检测格式以及独立的6DoF位姿JSON文件。用户可直接使用主流框架加载图像与标签,开展位姿估计、深度预测或语义分割等任务。此外,数据集的生成工具也开放了在线演示与付费版本,允许用户上传自定义STEP文件,一键生成专属工业位姿数据集,极大扩展了其应用场景与定制化能力。
背景与挑战
背景概述
在工业自动化与机器人抓取领域中,基于六自由度(6DoF)位姿估计的目标定位技术是视觉引导操作的核心。然而,现有公开数据集多集中于日常物体或简单几何形状,缺乏对复杂工业零件及其细粒度拓扑结构语义的标注。在此背景下,Huhb3D团队于2025年推出了Huhb3D-Industrial-100数据集,由22个工业零件组成,总计2200帧合成的640×480分辨率图像。该数据集的核心创新在于直接从STEP CAD文件中解析出15类拓扑标签,如孔、凸台、倒角、法兰、球面等,而非常见的曲率聚类或法向估计方法。这一设计填补了将CAD几何信息与位姿估计任务深度结合的空白,为工业视觉中的精细部件识别与鲁棒位姿回归提供了新的基准数据源。
当前挑战
该数据集所解决的领域挑战集中在两个方面。其一,工业场景中的6DoF位姿估计面临物体对称性、金属反光、遮挡严重以及缺乏纹理等难题,传统依赖RGB特征的方法常因光照变化而失效,而引入拓扑语义标签可直接利用几何结构约束位姿解算,提升在低纹理工业零件上的泛化能力。其二,构建过程中面临两大难点:一是从STEP文件中准确提取并统一分类15种拓扑面类型,需克服CAD模型几何表示多样性带来的歧义;二是自动化渲染管线需保证低重投影误差(仅0.0144像素),同时确保每个零件的掩码类别数与拓扑标签严格一致,为此团队对全部2200帧进行了逐帧质量审计,排除了黑帧、掩码错配及深度异常等腐败数据。
常用场景
经典使用场景
在计算机视觉与机器人感知领域,六自由度姿态估计是连接虚拟模型与物理世界的关键技术。Huhb3D-Industrial-100数据集专为工业场景下的精密部件设计,提供了22种典型工业零件的合成渲染图像,每件涵盖100帧高保真数据。其最经典的使用场景在于训练和评估基于深度学习的6D姿态估计模型,尤其是在缺乏真实标注的工业环境中。研究人员可利用其标准化的BOP格式数据,直接与主流算法框架如PoseCNN、DenseFusion等对接,开展从单一物体到多实例的鲁棒姿态回归实验。该数据集以极低的重投影误差(0.0144像素)著称,较传统基准如YCB-Video提升两个量级,为算法性能的精确量化提供了黄金标准。
解决学术问题
该数据集核心解决了工业视觉中拓扑感知缺失的学术难题。现有数据集多依赖曲率聚类或法线估算进行几何分割,而Huhb3D-Industrial-100创新性地从STEP CAD文件直接解析出15类拓扑标签,如水平面、凸台、倒角等,为每张图像赋予面级语义。这使得研究者能够突破传统仅依赖颜色和深度的局限,探索拓扑信息如何引导姿态估计网络关注零件关键几何特征。例如,区分螺栓的凸起与孔的凹陷对抓取规划至关重要,而该标签体系使模型能显式学习这种结构差异。其意义在于推动6D姿态估计从无差别像素回归转向几何结构感知,为可解释的工业视觉智能铺平道路。
实际应用
在工业4.0的生产线上,该数据集直接服务于自动化装配与质量检测两大环节。实际应用中,通过预训练的姿态估计模型,机械臂可基于单目或RGB-D相机实时定位散乱堆放的轴承、齿轮等零件,误差控制在毫米级别,显著提升分拣与插接效率。在质量追溯场景中,结合拓扑标签,系统能自动检测倒角缺失、法兰变形等制造缺陷,将几何精度从零件级细化至面特征级。此外,数据集的生成工具允许用户上传自有STEP文件快速定制专属数据,这对小批量、多品种的柔性产线尤为关键——企业无需手动标注即可为新型号零件部署视觉系统,大幅缩短了从设计到自动化的周期。
数据集最近研究
最新研究方向
在工业场景中,6DoF姿态估计正朝着高精度与几何语义深度融合的方向演进。Huhb3D-Industrial-100数据集通过直接从STEP CAD文件解析15种拓扑类别,突破了传统依赖曲率聚类或法线估计的局限,为每个面赋予了明确的几何角色标签(如平面、孔、倒角、凸台等),从而为深度学习模型提供结构化的先验知识。该数据集以极低的重投影误差(0.0144像素,优于YCB-Video 243倍)和毫米级深度精度,支撑了从检测、分割到姿态回归的多任务联合学习。当前前沿研究中,此类拓扑标签正在被用于引导注意力机制、设计图神经网络以建模零件部件间的拓扑关系,并探索基于CAD几何推理的少样本适配方法。其开源生成器使得研究者可针对自定义STEP文件快速生成标注数据,显著降低了工业级真实数据的获取成本,推动了智能制造与机器人抓取中感知闭环的实用化进程。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务