CVAI Catalog

·

View Catalog

tune

24 models found

·

23 public code

·

16 public weights

CM-UNet (Contrastive Masked UNet)

EPFL / Lausanne University Hospital (CHUV) · 2025

graph_1

Code & model weights public

Self-supervised deep learning model for coronary artery segmentation from invasive X-ray coronary angiography (ICA), designed to reduce reliance on large annotated datasets. CM-UNet combines a Contrastive Masked Autoencoder (CMAE) with a UNet backbone: an online encoder-decoder branch reconstructs masked image patches while a momentum branch produces contrastive embeddings, jointly pretraining the network on unannotated angiography images before fine-tuning on a small labeled set. Fine-tuning with only 18 annotated images (instead of 500) led to just a 15.2% drop in Dice score, versus a 46.5% drop for baseline models trained without this self-supervised pretraining -- demonstrating strong label efficiency for coronary segmentation.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0137

code

Training code public

Cross-modality cardiac image segmentation model that addresses spatial-temporal confounding -- where the anatomy and imaging-modality elements of cardiac images are intertwined across space and time. DCL performs multi-dimensional causal intervention, modeling causal relationships between images and labels as well as causality along the time and space dimensions, integrating historical optimal interventions to transfer knowledge across temporal contexts. A diffusion mechanism further keeps extracted anatomical elements causally invariant across modalities. On cross-modality cardiac images (MR, CT, and ultrasound), DCL achieved a mean Dice of 0.951, outperforming other advanced segmentation methods.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac MRI

Filter by Modality:
Cardiac MRI

CT angiography

Filter by Modality:
Cardiac CT

Multimodal

Filter by Modality:
Multimodal

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0148

·

Subject Count: 60

DOSTA-Net

Northwestern University (Advanced AI in Medicine and Physics Lab) · 2026

graph_1

Code & model weights public

Domain-Shuffle Temporal Attention Network for coronary vessel extraction from X-ray coronary angiography (XCA), trained entirely on synthetic temporal XCA data without requiring manual vessel annotations. By leveraging synthetic data generation and a domain-shuffle temporal attention mechanism, DOSTA-Net avoids the need for costly expert-labeled real angiography sequences while still learning temporally consistent vessel segmentation across frames of an XCA sequence.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0160

RL4Seg3D

University of Sherbrooke / CREATIS-Lyon / iCardio.ai · 2026

graph_1

Code & model weights public

Reinforcement-learning-based unsupervised domain adaptation framework for spatio-temporal (2D+time) echocardiography segmentation, extending the authors' earlier RL4Seg work to full-length video sequences. RL4Seg3D uses a sliding-window approach supporting high-resolution, full-sized inputs, and fuses multiple reward mechanisms to improve segmentation reliability without requiring additional expert annotations in the target domain. Trained and evaluated on a large dataset of over 30,000 echocardiography videos, it outperforms baselines and foundation models on overall segmentation accuracy as well as echocardiography-specific metrics including anatomical/temporal validity and mitral-valve-commissure landmark precision, and supports test-time optimization via calibrated uncertainty estimates.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

Apache 2.0

Filter by License:
Permissive


Model ID: 0159

AngioPy

Lausanne University Hospital / EPFL (Ando, Thanou Labs) · 2025

graph_1

Code & model weights public

Open-source, user-guided deep learning tool for coronary artery segmentation from invasive coronary angiography (ICA), designed to improve on traditional quantitative coronary angiography (QCA) edge-detection algorithms that typically require manual correction. Rather than segmenting the whole coronary tree indiscriminately, AngioPy lets the user click a handful of ground-truth points along a specific target vessel (including side branches), and predicts a binary mask for that single artery at the chosen cardiac-cycle time-step. Evaluated against an established QCA system on angiograms from the FAME 2 trial, AngioPy achieved an average F1 score of 0.927 (internal) and 0.924 (external validation), with vessel-diameter and lesion minimal-lumen-diameter measurements showing excellent agreement with QCA (r=0.93-0.96).

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0133

CIS-UNet

University of Florida (Cooper, Shao Labs) · 2024

code

Training code public

Deep learning model for multi-class 3D segmentation of the aorta and its thirteen branches from CT angiography, intended to support planning of endovascular aortic interventions. CIS-UNet combines a CNN encoder with a symmetric decoder and a novel Context-aware Shifted Window Self-Attention (CSW-SA) bottleneck block that adapts the Swin transformer's patch-merging mechanism to more efficiently capture global spatial context. Trained and evaluated via 4-fold cross-validation on the first public multi-branch aorta CTA dataset (59 patients), CIS-UNet outperformed the state-of-the-art SwinUNETR baseline, achieving a mean Dice of 0.713 vs. 0.697 and mean surface distance of 2.78mm vs. 3.39mm, while being more computationally efficient.

CT angiography

Filter by Modality:
Cardiac CT

Aortic anatomy segmentation / measurement

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0130

·

Subject Count: 59

EchoNet-Peds

Stanford University / Cedars-Sinai Medical Center (Ouyang Lab) · 2023

graph_1

Code & model weights public

Pediatric-specific extension of EchoNet-Dynamic: a video-based deep learning model that segments the left ventricle and estimates ejection fraction (EF) from apical-4-chamber (A4C) and parasternal short-axis (PSAX) pediatric echocardiogram clips. Because adult-trained echo models generalize poorly to children (who vary widely in heart size, rate, and image quality), EchoNet-Peds was trained from scratch on a dedicated pediatric video dataset. It segments the LV with a Dice similarity coefficient of 0.89 in both views, estimates EF with a mean absolute error of 3.66%, and identifies pediatric systolic dysfunction with an AUC of 0.95, significantly outperforming an adult-trained model applied to the same pediatric data.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

LVEF estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

LV systolic dysfunction (LVSD)

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Segmentation

Filter by Task Type:
Segmentation & Detection

Regression

Filter by Task Type:
Regression

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0126

·

Subject Count: 1,923

Aortic Dissection Detection nnU-Net

German Cancer Research Center (DKFZ) / University Medical Centre Mannheim, Heidelberg University · 2025

lock

Code & model weights private

nnU-Net-based pipeline for automated detection and sub-classification of acute thoracic aortic dissection (AD) on heterogeneous CT imaging, formulated as a semantic segmentation task rather than direct image classification. The model segments the false lumen (ascending and descending) and the dissection membrane -- along with optional indirect signs such as hemopericardium, aortic wall hematoma, and supra-aortic branch dissection -- and a patient is classified as AD-positive if at least two of the three primary segmented regions exceed a volume threshold tuned via Youden's index; the same pipeline additionally flags Stanford type A dissections. Trained on 157 heterogeneous internal CT studies (not restricted to a single contrast protocol) from Mannheim University Medical Centre and evaluated on an internal held-out test set as well as public external datasets (ImageTBAD and AVT), the model reached an AUROC of 98.7% internally and 97.0% externally, and correctly flagged 93.3% of dissection cases that had not been clinically suspected before imaging. The authors state the trained network will be made publicly available as a non-medical device for further scientific research.

Aortic CT

Filter by Modality:
Cardiac CT

Aortic aneurysm / dissection

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0113

·

Subject Count: 263

CAMUS U-Net Baseline

CREATIS, University of Lyon (Leclerc et al.) / University of Sherbrooke (vitalab pretrained models) · 18,000,000 params · 2019

graph_1

Code & model weights public

The original U-Net baseline segmentation network introduced alongside the CAMUS (Cardiac Acquisitions for Multi-structure Ultrasound Segmentation) dataset, one of the largest fully open-access, expert-annotated 2D echocardiography benchmarks. The network segments the left ventricle endocardium (LVEndo), left ventricle epicardium/myocardium (LVEpi), and left atrium (LA) from apical 2-chamber and 4-chamber echo views at end-diastole (ED) and end-systole (ES). In the original ten-fold cross-validation benchmark comparing U-Net, U-Net++, Stacked Hourglass, Anatomically Constrained Neural Networks, and classical methods, the U-Net variant (18M parameters) achieved the best overall accuracy, reaching Dice scores of 0.939 (ED) / 0.916 (ES) for LVEndo and 0.954 (ED) / 0.945 (ES) for LVEpi, approaching inter-observer variability. A pretrained checkpoint of this baseline U-Net is distributed via the University of Sherbrooke's vitalab CASTOR project as part of a broader library for building anatomically-constrained cardiac segmentation pipelines.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

Apache 2.0

Filter by License:
Permissive


Model ID: 0115

·

Subject Count: 500

DeepSA (Deep Subtraction Angiography)

Chongqing Medical University (Zeng et al.) · 2024

graph_1

Code & model weights public

Self-supervised model that performs single-frame digital-subtraction-angiography-style vessel/background separation directly from a single live (non-subtracted) coronary angiogram frame, then supports fine-tuned coronary vessel segmentation. A U-Net-style network is pretrained via an image-to-image translation objective on 58,128 unannotated angiography DICOM series (3,756 patients), then fine-tuned for vessel segmentation on just 40 expert-annotated frames, reaching a Dice of 0.828 on the held-out fine-tuning set and a new state-of-the-art Dice of 0.755 on the public XCAD benchmark. Intended to help clinicians visualize potential stenosis sites without requiring true two-frame digital subtraction acquisition.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Generation

Filter by Task Type:
Generation

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0105

·

Subject Count: 3,796

StenUNet

Northwestern University (Bluhm Cardiovascular Institute) · 2023

graph_1

Code & model weights public

nnU-Net-based segmentation network that detects and delineates stenotic lesions directly from X-ray coronary angiography frames, developed for the ARCADE (MICCAI 2023) stenosis-detection challenge. A companion model (YOLO-Angio, same team) handles vessel-tree segmentation; StenUNet focuses specifically on pixel-wise localization of stenotic regions. Placed 3rd overall among ARCADE challenge entrants with an F1 score of 0.5348 on the hold-out test set, within 0.0005 of the 2nd-place team.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

Apache 2.0

Filter by License:
Permissive


Model ID: 0101

AI-CAC

Veterans Affairs Long Beach Healthcare System / UC Irvine / Mass General Brigham (Hagopian, Strebel, Bernatz, Aerts et al.) · 2025

graph_1

Code & model weights public

U-Net-variant segmentation model that identifies and quantifies coronary artery calcium (CAC) directly from routine non-gated, non-contrast chest CT scans -- the kind ordered for lung-cancer screening or unrelated indications rather than a dedicated cardiac scan -- so that the tens of millions of such scans performed annually can be opportunistically screened for cardiovascular risk without any extra imaging. Predicted calcium masks are combined with the CT's Hounsfield units to compute an Agatston-equivalent score. Trained on 446 expert-segmented scans from 98 medical centers across the U.S. Department of Veterans Affairs national health system (capturing substantial heterogeneity in scanners and protocols) and benchmarked against 795 patients with a paired same-year gated CAC study: nongated AI-CAC differentiates zero-vs-nonzero and <100-vs->=100 Agatston categories with 89.4% (F1 0.93) and 87.3% (F1 0.89) accuracy respectively, and its score stratifies 10-year all-cause mortality (CAC 0 vs. >400: 25.4% vs. 60.2%, hazard ratio 3.49) and composite stroke/MI/death risk (33.5% vs. 63.8%, hazard ratio 3.00). In a simulated opportunistic-screening run across 8,052 low-dose CT scans, cardiologists confirmed 99.2% of patients flagged with AI-CAC >400 would benefit from lipid-lowering therapy. Code and trained model weights are both public under an MIT license.

Non-contrast cardiac CT

Filter by Modality:
Cardiac CT

Coronary artery calcium (CAC) scoring

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Binary classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0099

DeepHeart (Pediatric Tricuspid Valve Segmentation)

Children's Hospital of Philadelphia / MIT / Queen's University (Herz, Jolley et al.) · 2021

code

Training code public

Deep learning framework, developed in collaboration with the MONAI community, for automatic segmentation of tricuspid valve leaflets from transthoracic 3D echocardiograms in children with hypoplastic left heart syndrome (HLHS) and other forms of single-ventricle congenital heart disease, integrated into 3D Slicer via MONAILabel for interactive clinical/research use. Addresses a modality (pediatric 3D echocardiography) and population (single-ventricle congenital heart disease) largely absent from adult-focused cardiac AI models.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0076

·

Subject Count: 129

EchoDFKD

Medical University of Innsbruck (Dlaska Lab) · 2025

code

Training code public

Framework for training an echocardiography left-ventricle segmentation model purely by data-free knowledge distillation: a ConvLSTM-based student network learns to reproduce the masks produced by an EchoNet-Dynamic (DeepLabV3-ResNet50) teacher on entirely synthetic echo videos, with no real labeled data or even real videos required. Achieves state-of-the-art results identifying end-diastolic/end-systolic frames, reaching segmentation quality close to real-data training with substantially fewer weights; also introduces a human-annotation-free evaluation method using a large auxiliary model.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

LVEF estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Segmentation

Filter by Task Type:
Segmentation & Detection

Regression

Filter by Task Type:
Regression

RNN / LSTM / GRU

Filter by Architecture:
Recurrent

PyTorch

Filter by Framework:
PyTorch


Model ID: 0065

FedKD-SwinUNETR (Cardiac CT)

Heidelberg University Hospital (AICM) (Toelle, Engelhardt et al.) · 2025

code

Training code public

Largest federated cardiac CT analysis to date (n=8,104 scans) across a real-world federation of German university hospitals, addressing partially-labeled data across sites via a two-step semi-supervised knowledge-distillation strategy: task-specific CNNs first predict on unlabeled data per label type, then a SWIN-UNETR transformer learns from these predictions with label-specific heads. Learns a single federated model that simultaneously predicts TAVI-relevant landmarks (aortic hinge points, coronary ostia, membranous septum) and calcification from cardiac CT, improving generalizability over UNet-based baselines on downstream tasks.

CT angiography

Filter by Modality:
Cardiac CT

Procedural planning / outcome (PCI, TAVI)

Filter by Disease / Trait:
Prognosis & Aging

Coronary artery calcium (CAC) scoring

Filter by Disease / Trait:
Coronary & Ischemic Disease

Detection / localization

Filter by Task Type:
Segmentation & Detection

Segmentation

Filter by Task Type:
Segmentation & Detection

Transformer

Filter by Architecture:
Transformer

PyTorch

Filter by Framework:
PyTorch


Model ID: 0074

·

Subject Count: 8,104

MemSAM

Shenzhen University / Hong Kong Polytechnic University (Deng, Wu, Zeng, Qin) · 2024

code

Training code public

Adapts the Segment Anything Model (SAM) to echocardiography video segmentation by giving it a space-time memory that carries both spatial and temporal cues, so that only the first frame of a video needs an external point prompt and every subsequent frame is segmented from a propagated memory prompt instead. A memory reinforcement mechanism uses each frame's predicted mask to suppress speckle-noise features before they are written back into memory, addressing a key failure mode of naively adapting video object segmentation (e.g. XMem) to noisy ultrasound. Built on SAMUS (an ultrasound-adapted SAM) with a frozen SAM backbone and only the image-encoder adapter layers trained. On the semi-supervised CAMUS and EchoNet-Dynamic benchmarks (only end-diastole/end-systole frames labeled), MemSAM reaches 93.3% and 92.8% mean Dice respectively, outperforming UNet, SwinUNet, H2Former, and prior medical-SAM adaptations (MedSAM, MSA, SAMed, SonoSAM, SAMUS) with far fewer prompts, and derives LVEF (via Simpson's biplane method of disks) with a Pearson correlation of 78.9% against ground truth on CAMUS. Training/inference code is public (MIT license); only the starting SAM ViT-B checkpoint is linked for download, not a separately released fine-tuned MemSAM checkpoint.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

LVEF estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Segmentation

Filter by Task Type:
Segmentation & Detection

Regression

Filter by Task Type:
Regression

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0098

·

Subject Count: 10,530

SimLVSeg

Mohamed Bin Zayed University of Artificial Intelligence (BioMedIA) · 2024

graph_1

Code & model weights public

Self- and weakly-supervised pipeline for left-ventricle segmentation across the full cardiac cycle in apical-4-chamber echocardiography videos. A video segmentation network (2D super-image or 3D U-Net encoder) is first pretrained with a self-supervised temporal-masking objective on largely unannotated echo frames, then fine-tuned with weak supervision from the sparse end-diastole/end-systole frame labels that most echo datasets provide. Achieves 93.3% Dice on EchoNet-Dynamic, outperforming nnU-Net and non-SSL baselines, and generalizes to the external CAMUS dataset. Developed by the BioMedIA group at MBZUAI.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

CC BY-NC 4.0

Filter by License:
Non-commercial / Research-only


Model ID: 0081

·

Subject Count: 10,030

ViTa

Technical University of Munich (Zhang, Hager, Pan et al.) · 2025

graph_1

Code & model weights public

Multimodal cardiac MRI foundation model that fuses 3D+T cine CMR (short-axis and long-axis views) with tabular patient health records (demographics, metabolic, and lifestyle factors) from 42,000 UK Biobank participants. Two-stage self-supervised pretraining -- masked-image reconstruction, then imaging-tabular contrastive alignment -- produces representations that transfer to whole-heart segmentation, cardiac phenotype/physiological-feature regression, and cardiac/metabolic disease classification within one unified framework.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Structured EHR

Filter by Modality:
Text & EHR

Multimodal

Filter by Modality:
Multimodal

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Regression

Filter by Task Type:
Regression

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0062

·

Subject Count: 74,916

EchoNet-AS

Cedars-Sinai Medical Center (Smidt Heart Institute) / Ouyang Lab · 2025

graph_1

Code & model weights public

Open-source pipeline that classifies aortic stenosis (AS) severity from transthoracic echocardiography by combining structural and functional information. Video-based R(2+1)D convolutional networks read six B-mode and color Doppler views while a segmentation model measures peak aortic-jet velocity, and an ensemble integrates these into a final severity prediction. Trained on 210,193 images from Kaiser Permanente Northern California and validated across held-out, temporally distinct, and external Stanford and Cedars-Sinai cohorts, reaching AUCs up to 0.96–0.99 for severe AS. Developed by the Ouyang lab.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Multi-class classification

Filter by Task Type:
Classification

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0054

CineMA

UCL / Mycardium (Fu et al.) · 2025

graph_1

Code & model weights public

Foundation model for cine cardiac MRI, self-supervised via masked autoencoding on nearly 75,000 UK Biobank scans. Uses a Vision Transformer with a convolutional stem, unified across long-axis and short-axis views. Fine-tuned checkpoints are released for ventricle and myocardium segmentation, ejection-fraction regression, cardiovascular disease classification, and landmark localization across several public benchmark datasets (ACDC, M&Ms, M&Ms2, EMIDEC, and others).

Cardiac MRI

Filter by Modality:
Cardiac MRI

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

LVEF estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Binary classification

Filter by Task Type:
Classification

Segmentation

Filter by Task Type:
Segmentation & Detection

Regression

Filter by Task Type:
Regression

Detection / localization

Filter by Task Type:
Segmentation & Detection

Vision Transformer

Filter by Architecture:
Transformer

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0003

·

Subject Count: 74,916

EchoNet-Dynamic

Stanford University / Ouyang Lab · 2020

graph_1

Code & model weights public

End-to-end pipeline for apical-4-chamber echocardiogram videos that segments the left ventricle, estimates ejection fraction on a beat-to-beat basis, and classifies cardiomyopathy with reduced ejection fraction. Combines a DeepLabV3-ResNet50 segmentation model with a 3D CNN (R2+1D/R3D/MC3) initialized on the Kinetics-400 video dataset. Trained on the public EchoNet-Dynamic dataset released alongside it, and one of the most widely reused open echocardiography models since its 2020 Nature publication. Developed by Stanford University.

Echocardiography video

Filter by Modality:
Echocardiography

LVEF estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

LV systolic dysfunction (LVSD)

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Regression

Filter by Task Type:
Regression

Binary classification

Filter by Task Type:
Classification

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0036

·

Subject Count: 10,030

M&Ms-winning nnU-Net ensemble

DKFZ (Full, Isensee, Jager, Maier-Hein) · Task114_heart_mnms · 2021

graph_1

Code & model weights public

Ensemble of ten self-configuring nnU-Net models (five 2D, five 3D) that segments the left ventricle, right ventricle, and myocardium from short-axis cardiac cine MRI. Won the 2020 M&Ms challenge, a multi-centre, multi-vendor, multi-disease benchmark spanning scanners from four vendors and three countries, demonstrating strong generalization across acquisition protocols. Developed by DKFZ, the group behind the widely used nnU-Net framework.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

CC BY 4.0

Filter by License:
Open — Attribution


Model ID: 0005

·

Subject Count: 350

MSWS Network + Boundary2Patches (LAScarQS2022)

Queen Mary University of London (Khan et al.) · 2022

code

Training code public

Two-stage pipeline that segments the left atrium and quantifies atrial scar tissue from 3D late-gadolinium-enhancement cardiac MRI, supporting atrial-fibrillation ablation planning. A Multi-Scale Weight Sharing network first delineates the atrial cavity, then a boundary-patch method segments scar tissue around the detected wall. Developed at Queen Mary University of London for the LAScarQS 2022 MICCAI/STACOM segmentation challenge.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

LGE scar burden

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0004

SAM-VMNet

Ocean University of China / Shandong University · 2025

graph_1

Code & model weights public

Segments coronary vessels from invasive X-ray angiography images and automatically quantifies the degree of stenosis along the extracted centerlines. Combines MedSAM, a Segment-Anything-style vision model, with a Mamba-based VM-UNet segmentation branch for efficient long-range feature modeling. Trained and evaluated on the ARCADE, DCA1, and GH angiography datasets by researchers at Ocean University of China and Shandong University.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Detection / localization

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0001