VessShape
收藏资源简介:
VessShape是一个为预训练形状感知血管分割模型而设计的合成数据集生成方法。该数据集包含由贝塞尔曲线生成的管状几何形状,并结合了广泛的前景和背景纹理。通过在保持几何先验的同时多样化纹理,数据集明确鼓励模型学习形状特征而不是表面的纹理。VessShape旨在通过使用合成数据显式地注入强烈的形状偏置,同时系统地多样化纹理,以训练神经网络分割任何血管组织,这些组织遵循从VessShape获得的形状先验。数据集适用于解决在医学图像分析中由于缺乏大型标注数据集和模型在不同成像模态上的泛化能力差而导致的血管分割问题。
VessShape is a synthetic dataset generation method designed for pre-trained shape-aware vascular segmentation models. This dataset comprises tubular geometric shapes generated via Bezier curves, combined with a wide range of foreground and background textures. By diversifying textures while preserving geometric priors, the dataset explicitly encourages models to learn shape features rather than superficial texture ones. VessShape aims to explicitly inject a strong shape bias using synthetic data while systematically diversifying textures, to train neural networks to segment any vascular tissues that follow the shape priors derived from VessShape. This dataset is applicable to addressing vascular segmentation issues in medical image analysis caused by the shortage of large-scale annotated datasets and the poor generalization ability of models across different imaging modalities.
Vess-Shape-Dataset 数据集概述
数据集简介
Vess-Shape-Dataset是一个用于自动生成模拟血管外观和结构的合成数据集的Python工具包。主要目标是为图像分割模型的训练和评估提供真实数据,特别是在血管分割领域。
核心特性
- 合成血管生成:使用随机贝塞尔曲线创建真实的血管状掩码
- 纹理转换:将数据集(如ImageNet)中的纹理应用于前景(血管)和背景
- 融合处理:使用高斯模糊平滑融合血管和背景纹理,避免硬过渡
- 灵活参数:可控制复杂度、曲率、半径和每张图像中的血管数量
- 元数据生成:为每个生成的图像生成元数据,包括纹理来源和类别标签
- 网格生成:可选生成图像网格以便于可视化和比较
技术背景
该项目受"IMAGENET-TRAINED CNNS ARE BIASED TOWARDS TEXTURE; INCREASING SHAPE BIAS IMPROVES ACCURACY AND ROBUSTNESS"研究启发,通过生成血管状曲线并应用不同纹理,促使模型关注形状而非纹理特征。
使用示例
python from vessel_shape import VesselShape
vs = VesselShape( image_size=256, n_control_points=5, max_vd=100, radius=2, num_curves=3, texture_dir="imagenet_val_sample", annotation_csv="imagenet_val_sample/ILSVRC2012_img_val_annotation.csv" )
results = vs.generate(grid_dir=True, n_samples=3)
系统要求
- Python 3.8+
- numpy
- pandas
- pillow
- matplotlib
- tqdm
- scipy
项目结构
vess_shape_dataset/:核心包代码notebooks/:实验性笔记本和使用示例example_imgs/:示例图像imagenet_val_sample/:纹理样本和标注
数据集准备
- 将纹理图像(如ImageNet样本)放置在目录中
- 提供包含图像ID和类别标签的标注CSV文件
参考文献
- Stylized-ImageNet
- Geirhos, R. et al. "ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustness." ICLR 2019.

- 1VessShape: Few-shot 2D blood vessel segmentation by leveraging shape priors from synthetic images圣卡洛斯联邦大学计算机科学系 · 2025年



