CVAI Catalog

·

View Catalog

tune

20 models found

·

17 public code

·

10 public weights

3D nnU-Net Aortic Dissection Pipeline

General Hospital of Northern Theater Command / Northeastern University, China · 2025

graph_1

Code & model weights public

Fully automatic four-module deep learning pipeline for identifying, segmenting, and Stanford-subtyping aortic dissection (AD) from CT angiography (CTA). A 3D full-resolution nnU-Net first segments the aorta; the segmented boundary is then multi-view projected for AD identification; for AD-positive cases, a second 3D nnU-Net segments the true lumen (TL) and false lumen (FL); finally, a classifier performs Stanford subtyping from multi-view maximum-density projections of the TL/FL. On 386 CTA scans, the pipeline achieved 0.979 accuracy for AD identification, Dice of 0.968 (TL) and 0.971 (FL) for lumen segmentation, and 0.990 accuracy for Stanford subtyping.

Aortic CT

Filter by Modality:
Cardiac CT

Aortic aneurysm / dissection

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0144

·

Subject Count: 386

Multilabel CNN for Atrial Fibrosis Assessment (CemrgApp)

King's College London (Niederer Lab / CEMRG) · 2020

code

Training code public

Fully automatic, open-source deep learning pipeline for estimating left atrial fibrosis from late gadolinium enhancement (LGE) cardiac MRI, built to remove the operator-dependent steps that limit reproducibility of conventional atrial LGE analysis. A multilabel convolutional neural network delineates the left atrial blood pool, pulmonary veins, and mitral valve; these structures are then used to automatically calculate fibrosis burden via established image-intensity-ratio thresholds, without manual tracing. Validated on a 3D LGE-CMR dataset of 207 scans, the pipeline's automatic segmentation achieved a 91% Dice score against manual tracing, and its fully automatic fibrosis quantification closely matched semi-automatic reference methods. The CNN and pipeline are distributed as part of the open-source CemrgApp platform.

Cardiac MRI

Filter by Modality:
Cardiac MRI

LGE scar burden

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0134

·

Subject Count: 207

ACDC Segmenter

ETH Zurich (Baumgartner, Koch, Pollefeys, Konukoglu) · 2017

code

Training code public

One of the foundational baseline segmentation networks submitted to the 2017 Automated Cardiac Diagnosis Challenge (ACDC), comparing 2D and 3D convolutional network designs for segmenting the left ventricle cavity, myocardium, and right ventricle cavity from short-axis cine cardiac MRI at end-diastole and end-systole. The accompanying study systematically explored the tradeoffs between 2D and 3D convolutions for this task, finding that, due to the highly anisotropic voxel spacing typical of clinical cine cardiac MRI, 2D networks that treat each slice independently can match or exceed 3D networks while being far cheaper to train. The public code and pretrained weights for the best-performing configuration have served as a widely used, simple baseline for later cardiac MRI segmentation research (including for automatically deriving ventricular volumes and ejection fraction).

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0116

·

Subject Count: 150

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

Calcium scoring in low-dose chest CT (dilated CNN)

University Medical Center Utrecht (Lessmann et al.) · 2018

code

Training code public

Two-stage convolutional neural network that automatically detects and anatomically labels coronary artery, thoracic aorta, and cardiac-valve calcifications in low-dose chest CT acquired for lung-cancer screening. A first CNN with a large receptive field (via dilated convolutions) identifies and anatomically labels candidate calcifications; a second CNN filters true positives from the candidates. Trained and evaluated on 1,744 CT scans from the National Lung Screening Trial (NLST), reaching an F1 of 0.89 (soft-filter reconstructions) / 0.84 (sharp-filter reconstructions) for coronary artery calcifications and a linearly-weighted kappa of 0.90-0.91 for per-subject cardiovascular risk categorization versus the manual reference standard.

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

Multi-class classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0102

·

Subject Count: 1,744

DeepAAA

Massachusetts General Hospital / Brigham and Women's Hospital (Center for Clinical Data Science) · 2019

lock

Code & model weights private

Deep learning pipeline for detection and quantification of abdominal aortic aneurysm (AAA) -- a typically asymptomatic condition often missed incidentally by radiologists -- from abdominal-pelvic CT. A modified 3D U-Net segments the aorta on both contrast and non-contrast CT volumes with a variable number of images, after which an ellipse-fitting post-processing step measures the aortic cross-sectional diameter along the vessel to detect aneurysmal dilation. Trained and validated on 321 abdominal-pelvic CT examinations from Massachusetts General Hospital, the model reached a sensitivity/specificity of 0.91/0.95 on the primary validation set, and 0.85/1.0 on a separate 57-exam generalization test set with different patient demographics and acquisition characteristics; the authors report that DeepAAA exceeded literature-reported radiologist performance for incidental AAA detection.

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)


Model ID: 0114

·

Subject Count: 321

DeepIVUS

Emory University (Molony & Samady) · 2019

graph_1

Code & model weights public

Deep learning platform for fully automatic segmentation and phenotyping of coronary intravascular ultrasound (IVUS) pullbacks, packaged with a desktop GUI and CLI. A convolutional encoder-decoder network delineates the internal (lumen) and external elastic lamina borders on each cross-sectional IVUS frame; downstream rule-based analysis derives lumen area, plaque area, plaque burden, automatically flags lesions with plaque burden exceeding 40%, and reports minimum lumen area and maximum plaque burden along the pullback. Also supports end-diastolic gating and manual contour editing. Trained on 305 clinical IVUS pullbacks (270 train / 35 validation) from Philips and Boston Scientific catheters at Emory University; downstream evaluations have applied DeepIVUS to tasks such as automated detection of stent underexpansion.

Intravascular ultrasound (IVUS)

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

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

Apache 2.0

Filter by License:
Permissive


Model ID: 0104

·

Subject Count: 305

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

Fetal QRS Octave-ResNet

University of California, Irvine · 2020

lock

Code & model weights private

End-to-end deep learning model for detecting fetal QRS complexes directly from non-invasive abdominal ECG (aECG) signals, without requiring a separate reference maternal ECG or heavy hand-crafted feature engineering. The model adopts a ResNet architecture built from 1-D octave convolutions (OctConv), which factorize feature maps into high- and low-frequency components to capture multiple temporal frequency scales while reducing memory and compute cost relative to a standard 1-D ResNet; the resulting feature-importance weighting also highlights the signal regions most relevant to each detection. Evaluated on the PhysioNet/CinC Challenge 2013 fetal ECG database (with added Gaussian and motion-artifact noise to mimic real-world recording conditions), the model reached an F1 score of 91.1% while cutting computation by more than 50% for less than a 2% drop in performance versus a non-octave ResNet baseline.

Fetal ECG

Filter by Modality:
ECG

Fetal / maternal cardiac monitoring

Filter by Disease / Trait:
Other Conditions

Detection / localization

Filter by Task Type:
Segmentation & Detection

CNN (1D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0112

·

Subject Count: 75

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

caroSegDeep

Univ Lyon / INSA-Lyon / CREATIS (Laine et al.) · 2022

code

Training code public

Dilated U-Net model for fully automatic segmentation of the intima-media complex (IMC) of the common carotid artery on longitudinal B-mode ultrasound images, used to measure carotid intima-media thickness (cIMT) -- a standard imaging biomarker of subclinical atherosclerosis. A far-wall detection step first localizes the region of interest, and the dilated U-Net then segments the near- and far-wall IMC boundaries within it. Trained and evaluated with 5-fold cross-validation on a multicenter database of 2,176 images annotated by two experts, the method reached a mean absolute thickness difference of under 120 micrometres versus the reference annotations -- smaller than the approximately 180-micrometre inter-observer variability -- with a 98.7% fully-automatic success rate (only 1.3% of cases required manual correction).

Carotid ultrasound

Filter by Modality:
Vascular Ultrasound

Carotid atherosclerosis / stenosis

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0110

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

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

4Dsegment (bi-ventricular segmentation + motion tracking)

Imperial College London (Duan et al., UK Digital Heart Project) · 2019

graph_1

Code & model weights public

Automated pipeline that segments both heart ventricles and tracks their motion throughout the cardiac cycle from short-axis cine cardiac MRI, producing 3D bi-ventricular models with per-vertex wall-thickness and curvature measurements over time. Built on a shape-refined multi-task fully convolutional network, followed by non-rigid registration and mesh-based motion tracking. Trained on roughly 400 manually annotated pulmonary hypertension patients as part of Imperial College London's UK Digital Heart Project, and underlies the related 4Dsurvival cardiac-motion survival-prediction study.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

GPL 3.0

Filter by License:
Copyleft


Model ID: 0002

·

Subject Count: 400

DeepCAC

Harvard AIM Lab (Zeleznik et al.) · 2021

graph_1

Code & model weights public

Fully automatic pipeline that localizes the heart, segments coronary calcium, and produces an Agatston-style coronary artery calcium score from gated and non-gated chest/cardiac CT. A three-stage 3D CNN performs each step in sequence. Validated across the Framingham, NLST, PROMISE, and ROMICAT-II cohorts, where the resulting calcium score predicted cardiovascular events with hazard ratios up to 4.3. Developed by the Harvard AIM Lab.

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

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

GPL 3.0

Filter by License:
Copyleft


Model ID: 0008

·

Subject Count: 3,380

code

Training code public

3D CNN that segments seven cardiac substructures - both ventricles, both atria, the LV myocardium, ascending aorta, and pulmonary artery trunk - from cardiac CT angiography. Trained with a hybrid loss function combining multiple segmentation objectives. Developed at CUHK for the MICCAI 2017 Multi-Modality Whole Heart Segmentation (MM-WHS) challenge.

CT angiography

Filter by Modality:
Cardiac CT

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0009

·

Subject Count: 60

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

ukbb_cardiac segmentation network

Imperial College London (Bai et al.) · 2018

graph_1

Code & model weights public

Long-standing toolbox for automated segmentation of the ventricles and atria and derivation of cardiac imaging phenotypes from short- and long-axis cine cardiac MRI. Built on a fully convolutional network trained per slice, and widely reused across UK Biobank cardiac imaging studies since its 2018 publication. Developed at Imperial College London.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

Apache 2.0

Filter by License:
Permissive


Model ID: 0006

·

Subject Count: 74,916