CVAI Catalog

·

View Catalog

tune

9 models found

·

8 public code

·

8 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

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

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

SVS-net

Shanghai Jiao Tong University (Qin Lab) / University of Texas Southwestern Medical Center · 2020

graph_1

Code & model weights public

Deep channel-attention network for segmenting the full coronary vessel tree from sequential X-ray coronary angiography (XCA) frames, rather than a single static image. An encoder-decoder architecture fuses temporal-spatial feature maps across the XCA sequence via skip connections, then uses channel-attention blocks in the decoder to refine features and separate thin vessel structures from complex, noisy backgrounds; a Dice loss addresses the severe foreground/background class imbalance typical of XCA. The authors report that SVS-net outperforms prior 2D and video-based baselines on both quantitative vessel-segmentation metrics and visual validation.

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

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0129

·

Subject Count: 120

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

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

CathAI

University of California, San Francisco (Avram, Tison et al.) · 2023

lock

Code & model weights private

Fully automated pipeline for interpreting coronary angiograms that chains four purpose-built neural networks: (1) angiographic projection-angle identification, (2) left/right coronary artery detection, (3) arterial segment localization, and (4) stenosis-severity estimation. Trained on 13,843 angiographic studies (195,195 videos) from 11,972 adult patients at UCSF (2008-2019), with projection-angle and LCA/RCA-detection tasks each reaching precision/sensitivity/F1 at or above 90%. For predicting obstructive coronary artery disease (>=70% stenosis), CathAI reaches an AUC of 0.862 internally, 0.869 on external angiograms from the University of Ottawa Heart Institute, and 0.775 after retraining on quantitative-coronary-angiography labels from the Montreal Heart Institute core lab. No public code or model weights have been released.

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

Multi-class classification

Filter by Task Type:
Classification

Detection / localization

Filter by Task Type:
Segmentation & Detection

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch


Model ID: 0091

·

Subject Count: 11,972

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