Home

Heart Disease CNN Classifier

A convolutional neural network (CNN) project built with PyTorch and timm for detecting heart disease from medical images.
This repository provides a modular structure for training, evaluating, and deploying deep learning models.

---

About the Project

This project aims to classify heart conditions based on image data using modern deep learning techniques.
It provides reusable components for:

• Model training and evaluation (TrainingManager)
• Model inference and prediction (Predictor)
• Dataset loading and preprocessing (DatasetManager)

The goal is to offer a clear and extendable training pipeline for CNN-based image classification tasks.

---

🧬 The CNN Classifier project demonstrates a full deep learning workflow — from preprocessing to model training and evaluation — built with PyTorch and timm for efficient multiclass image recognition.