Veriserum
收藏Veriserum (Ongoing)
概述
Veriserum 是一个专门为推进医学影像中的深度学习而设计的开源数据集,特别是用于荧光透视图像分析。该数据集提供带有植入假体的双平面荧光透视图像,使研究人员能够在生物力学、计算机视觉和姿态估计等领域进行探索。Veriserum 免费提供,以支持研究社区在图像配准、校准、分割和 3D 重建等领域开发和测试算法。
主要特点
- 双平面荧光透视:从两个视角配对的图像有助于 3D 重建和姿态估计。
- 植入假体:合成植入结构允许对算法进行详细测试。
- 预校准图像:图像在尺寸和强度上标准化,可直接应用于常见的深度学习模型。
- 开源:免费用于学术和研究用途,促进透明度和协作。
目录结构
项目目录结构如下:
plaintext project_directory/ ├── model.py # 定义 PoseEstimationModel ├── train.py # 包含数据加载和训练逻辑 ├── veriserum_dataset.py # 定义自定义 Veriserum_calibrated 数据集 ├── sampler.py # 定义自定义 NonObsoleteSampler ├── csv_files/ │ ├── veriserum_pose_reloaded_refined_all.csv │ ├── distortion_calibration_reloaded.csv │ └── source_int_calibration_reloaded.csv
目录和文件描述
- model.py:包含
PoseEstimationModel,该模型利用预训练的 ResNeXt 主干网络在荧光透视图像上执行姿态估计。 - train.py:包括数据加载、增强和使用 PyTorch Lightning 进行模型训练。配置训练/验证分割、日志记录和模型检查点。
- veriserum_dataset.py:定义
Veriserum_calibrated,一个自定义的 PyTorch 数据集类,用于加载双平面荧光透视图像,必要时进行校准,并检索相应的姿态数据。 - sampler.py:实现
NonObsoleteSampler,该采样器根据特定条件(例如,移除过时条目)过滤数据,以确保高质量样本。 - csv_files/:包含校准和姿态数据文件:
veriserum_pose_reloaded_refined_all.csv:每张图像的姿态数据。distortion_calibration_reloaded.csv:用于校正图像的畸变校准数据。source_int_calibration_reloaded.csv:源增强器的校准数据。
设置和安装
-
克隆仓库:
bash git clone https://github.com/yourusername/veriserum.git cd veriserum
-
安装依赖项:该项目需要 Python 3.8+ 和 PyTorch。使用以下命令安装所需包:
bash pip install -r requirements.txt
-
数据准备:确保数据集和校准文件正确放置在
csv_files/目录中。如果使用自定义目录结构,请修改veriserum_dataset.py中的路径。
使用
训练模型
要在 Veriserum 上训练模型:
-
在
train.py中配置训练参数(例如,批量大小、学习率)。 -
运行训练脚本:
bash python train.py
这将训练 PoseEstimationModel 并根据验证损失保存最佳模型检查点。
自定义数据加载和变换
veriserum_dataset.py 中的 Veriserum_calibrated 类可以自定义以应用特定变换或动态调整校准。train.py 还提供了一个模板,用于使用 PyTorch 的 torchvision.transforms 模块对数据集进行额外变换。




