遇见数据集

WireSeg-36K

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

资源简介:

WireSeg-36K是一个用于电缆/电线实例分割的合成图像数据集,包含36,000张1024x1024分辨率的RGB图像。该数据集通过物理模拟的可变形线性对象(DLOs)渲染在多样化的真实背景上生成,提供了精确的每实例分割掩码和每图像的度量深度信息。数据集中仅包含一个类别:`cable`(电缆),总计有176,387个电缆实例掩码,采用COCO RLE格式编码。数据集涵盖三种场景类型:`data_center`(数据中心,6,000张图像,模拟服务器机架上的线缆)、`flying_wire`(悬垂线缆,15,000张图像,模拟悬挂或悬链线)和`wire_on_plane`(平面线缆,15,000张图像,模拟平铺在平面上的线缆),每种场景包含不同数量的电缆(如2/4/8/16根)。数据文件包括:包含RGB图像、实例掩码(RLE格式)和元数据的Parquet文件、完整的COCO格式标注文件(annotations.json)以及存储为float32 numpy数组的度量深度文件。数据集使用`cc-by-4.0`许可证,适用于图像分割、深度估计以及相关计算机视觉任务的研究与模型训练。需要注意的是,数据集前景为合成生成,与真实图像存在域差距,背景为真实照片合成。

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

WireSeg-36K 数据集概述

基本信息

  • 许可证: CC-BY-4.0
  • 数据集规模: 10K < n < 100K
  • 任务类型: 图像分割、深度估计
  • 标签: DLO、分割、线缆、电线、SAM、真实图像、合成图像

数据集内容

  • 图像数量: 36,000 张 (1024x1024 RGB)
  • 实例数量: 176,387 个线缆掩码 (COCO RLE格式)
  • 类别: 1 个 (cable)
  • 数据模态: RGB图像、实例掩码、逐图像度量深度

场景构成

场景系列 线缆数量 图像数量 描述
data_center 4 / 8 / 16 6,000 服务器机架上的布线
flying_wire 2 / 4 / 8 15,000 悬挂/悬链线
wire_on_plane 2 / 4 / 8 15,000 平铺在平面上的线缆

文件结构

路径 说明
data/*.parquet (配置 default) RGB图像 + 逐实例COCO-RLE掩码 + 元数据,每行一张图像
annotations.json 完整COCO标注 (36,000张图像 / 176,387个标注,单一类别 cable)
depth/*.tar.gz 逐图像度量深度,float32 [1024,1024] .npy,打包在100个gzip压缩的tar分片中

数据字段 (Parquet default)

  • image (RGB)
  • file_name
  • scene_family
  • n_cable
  • masks_rle
  • bboxes
  • areas
  • num_instances
  • wire_area_frac
  • det_f1_iou_50
  • det_f1_iou_75
  • det_map_50_95
  • jaccard_j
  • category
  • 掩码/边界框/面积为JSON编码的逐实例列表 (RLE = COCO压缩格式)

数据划分

提供单个 train 分割,用户可根据需要自行划分(建议按 scene_family / n_cable 分层抽样)。

数据生成

通过 DeformX 模拟和渲染线缆,并与背景图像合成;地面真实实例掩码和度量深度直接来自渲染器(无人工标注),导出为COCO RLE格式。

局限与注意事项

  • 合成前景,与真实图像存在域差距
  • 背景为合成使用的真实照片,可能包含偶然场景内容

引用

bibtex @inproceedings{yang2026deformx, title = {DeformX: A Versatile Co-Simulation Framework for Deformable Linear Objects}, author = {Yang, Yi and Fei, Xiang and Wang, Lehong and Li, Chenhao and Dai, Zilin and Kou, Henry and Li, Lu and Choset, Howie}, booktitle = {2026 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)}, year = {2026}, organization = {IEEE} }

搜集汇总
数据集介绍
WireSeg-36K 数据集图片
构建方式
WireSeg-36K数据集由DeformX物理仿真框架生成,专注于可变形线性物体(DLO)的实例分割。通过模拟线缆在数据中心机架、悬浮状态和平面铺设三种典型场景中的物理形态,渲染出36,000张1024x1024分辨率的RGB图像,并叠加真实背景照片以增强视觉多样性。每张图像均配备精确的逐实例掩码(以COCO压缩RLE格式存储)和度量深度图,所有标注均由渲染器自动导出,无需人工介入,确保了标注的一致性与规模化生产能力。
特点
该数据集包含176,387个线缆实例掩码,单一类别“cable”,覆盖了从2根到16根不等的线缆密度变化。其独特之处在于提供了RGB图像、实例掩码和每张图像的度量深度图三种模态数据,深度图为float32的1024x1024矩阵,可直接用于三维感知任务。数据集基于物理仿真生成,背景为真实照片合成,虽存在合成域与真实域的差异,但为模型提供了大量可控、多样化的训练样本。
使用方法
用户可通过HuggingFace的datasets库加载Parquet格式的默认配置,直接获取RGB图像、RLE编码掩码及元数据。深度数据需额外从gzipped tar分片中提取,并依据图像文件路径匹配对应的npy文件。数据集仅提供单一训练划分,建议用户按场景类型和线缆密度进行分层抽样以构建验证和测试集。使用PyCOCO工具可解码RLE掩码为HxW的uint8矩阵,便于训练分割模型。
背景与挑战
背景概述
在机器人操作与工业自动化领域,可变形线性物体(DLOs)如电缆、线缆的感知与操控是长期存在的技术难题。由于DLOs具有高自由度、自遮挡以及复杂拓扑结构等特性,传统视觉方法难以实现精确的实例分割与深度估计。为此,由Yi Yang、Xiang Fei等研究人员于2026年在IEEE/RSJ国际智能机器人与系统会议(IROS)上提出WireSeg-36K数据集,旨在填补大规模、高精度标注的DLO分割数据空白。该数据集依托于DeformX仿真框架生成,包含36,000张1024×1024分辨率的RGB图像,涵盖数据中心、悬浮线缆及平面放置线缆三类场景,提供超过176,000个精确的电缆实例掩码及度量深度图,为线缆实例分割与深度估计研究提供了高质量的基准资源。
当前挑战
WireSeg-36K数据集所解决的领域问题主要在于:电缆等可变形线性物体的实例分割与深度估计长期受限于数据匮乏与标注成本高昂,传统算法难以处理线缆之间的交叉、缠绕及复杂背景干扰。构建过程中面临的核心挑战包括:1)物理仿真的真实性——需模拟线缆在重力、接触等条件下的真实形变,以确保生成图像与真实场景的域差距可控;2)多场景多样性——需设计数据中心布线、悬浮线缆与平面放置三种典型场景,并控制线缆数量变化(2至16根),以覆盖不同复杂度的操作环境;3)大规模数据生成与标注自动化——利用DeformX框架实现全自动渲染与标注,避免人工标注的高昂成本与误差,但合成图像与真实图像之间存在域偏移,影响模型泛化能力。
常用场景
经典使用场景
在机器人操作与自动化领域,线缆等可变形线性对象(DLO)的视觉感知长期面临数据匮乏的困境。WireSeg-36K数据集凭借其36,000张高分辨率合成图像、176,387个精确的实例分割掩码以及逐图像度量深度信息,为线缆实例分割任务提供了系统性的研究基准。该数据集覆盖数据中心机架布线、悬空垂链线缆、平面放置线缆三类典型场景,并精细调控每张图像中的线缆数量(2至16根),有效支撑了从稀疏到密集场景下的分割模型训练与评估。研究者可借助该数据集开发鲁棒的单类别实例分割算法,探索合成数据向真实场景的迁移能力,从而推动DLO操作中视觉感知模块的进步。
实际应用
在实际生产环境中,线缆的自主操作是工业机器人与服务机器人面临的核心挑战之一。WireSeg-36K数据集可直接用于训练部署于数据中心、装配流水线及家庭服务场景中的线缆分割模型。例如,在数据中心中,维护机器人借助该数据集训练的模型能够精准识别并区分密集交错的服务器背部线缆,从而完成自主插拔、重新布线与故障排查任务。在制造业中,装配机器人可通过实例分割结果规划抓取姿态,实现杂乱工作台面上线缆的自动拾取与整理。此外,结合深度信息,该数据集还可赋能机器人在立体空间中对悬空线缆进行三维重建与无碰撞路径规划,从而在实际应用中实现从视觉感知到操作执行的完整闭环。
衍生相关工作
围绕WireSeg-36K数据集,学术界已衍生出多项具有影响力的工作。核心基础是其生成框架DeformX,发表于IROS 2026,提出了面向可变形线性对象的协同仿真方法论,可扩展生成各类DLO的视觉数据。在该数据集之上,研究者开发了面向合成到真实域迁移的域适应分割方法,例如采用对抗性训练或风格迁移技术提升模型在真实工业场景中的泛化能力。此外,基于该数据集训练的分割模型常被用作线缆操纵系统的前端视觉模块,催生了面向特定任务(如线缆穿孔、插接)的精细操作策略研究。同时,该数据集结合深度信息,推动了面向DLO的6D姿态估计与几何重建工作,进一步丰富了机器人灵巧操作在非刚性物体上的研究体系。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务