4 个月前

数据分割和指标用于手术动作三元组数据集的方法基准测试

数据分割和指标用于手术动作三元组数据集的方法基准测试

摘要

除了生成数据和注释外,制定合理的数据分割策略和评估指标对于创建基准数据集也至关重要。这一做法确保了数据使用的共识、评估的一致性和研究方法在该数据集上的统一比较。本研究重点关注CholecT50,这是一个包含50个手术视频的数据集,将手术活动形式化为<器械, 动词, 目标>三元组。本文介绍了CholecT50和CholecT45数据集的标准分割方法,并展示了它们与现有数据集使用方法的对比。CholecT45是CholecT50数据集中首次公开发布的45个视频。我们还开发了一个名为ivtmetrics的评估指标库,用于对手术三元组模型进行评估。此外,我们通过在最常用深度学习框架(PyTorch和TensorFlow)中重现基线方法,进行了基准研究,并使用所提出的数据分割方法和评估指标对这些方法进行了评估,同时公开发布了这些基线方法以支持未来的研究。所提出的数据分割方法和评估指标将有助于在全球范围内跟踪该数据集的研究进展,并促进最优模型的选择以进一步部署。

代码仓库

CAMMA-public/cholect50
pytorch
GitHub 中提及
CAMMA-public/cholect45
官方
pytorch
GitHub 中提及
camma-public/rendezvous-in-time
pytorch
GitHub 中提及
camma-public/tripnet
官方
pytorch
GitHub 中提及
camma-public/mcit-ig
pytorch
GitHub 中提及
camma-public/rendezvous
pytorch
GitHub 中提及
camma-public/attention-tripnet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
action-triplet-recognition-on-cholect45Rendezvous
mAP: 29.4±2.8
action-triplet-recognition-on-cholect45Attention Tripnet
mAP: 27.2±2.7
action-triplet-recognition-on-cholect45Tripnet
mAP: 24.4±4.7
action-triplet-recognition-on-cholect45-crossRendezvous
mAP: 29.4±2.8
action-triplet-recognition-on-cholect45-crossAttention Tripnet
mAP: 27.2±2.7
action-triplet-recognition-on-cholect45-crossTripnet
mAP: 24.4±4.7
action-triplet-recognition-on-cholect50Rendezvous (PyTorch)
Mean AP: 29.5
action-triplet-recognition-on-cholect50Attention Tripnet (PyTorch)
Mean AP: 23.3
action-triplet-recognition-on-cholect50Tripnet (PyTorch)
Mean AP: 21.6
action-triplet-recognition-on-cholect50-1Rendezvous (PyTorch)
mAP: 32.8
action-triplet-recognition-on-cholect50-1Attention Tripnet (PyTorch)
mAP: 27.7
action-triplet-recognition-on-cholect50-1Tripnet (PyTorch)
mAP: 27.4
action-triplet-recognition-on-cholect50-crossRendezvous
mAP: 29.4±2.5
action-triplet-recognition-on-cholect50-cross-1Tripnet
mAP: 25.3±2.4
action-triplet-recognition-on-cholect50-cross-1Attention Tripnet
mAP: 27.2±2.9
action-triplet-recognition-on-cholect50-cross-1Rendezvous
mAP: 29.4±2.5

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
数据分割和指标用于手术动作三元组数据集的方法基准测试 | 论文 | HyperAI超神经