2025 · Computer Vision
Mitral Valve Segmentation
PythonPyTorchMedical ImagingComputer VisionDeep Learning
Role
ML Engineer
Year
2025
Category
Computer Vision
Context
Academic Project
01.
Problem
Drawing the mitral valve on an echo is messy. Speckle, low contrast, and a valve that moves fast. Two people often disagree on the contour. A plain CNN mixes the leaflet with nearby tissue.
I used an Attention U-Net with Squeeze-and-Excitation blocks and attention gates. The pipeline ran Non-local Means and CLAHE first, then elastic deformations so the training set had more tissue-like warps. Loss was Dice plus BCE. I trained on a mix of amateur and expert labels and tracked IoU.
02.
Approach
03. Outcome
The model followed the leaflet on the held-out echoes I had. I do not have a public leaderboard number for this one.
Gallery
01
02