ECG Analysis Model
This is a lightweight CNN model trained for ECG image classification.
Model Details
- Model Type: Convolutional Neural Network
- Classes: NORM, AFIB, SBRAD, STACH, BIGEMINY, VT
- Input Size: 224x224x3 (RGB images)
- Framework: PyTorch
Usage
import torch
from model import SimpleCNN
model = SimpleCNN(num_classes=6)
model.load_state_dict(torch.load('model.pt'))
model.eval()
Training Data
- Synthetic ECG patterns
- 6 classes: NORM, AFIB, SBRAD, STACH, BIGEMINY, VT
- Training samples: 800
- Test samples: 200
Created: 2025-09-19 15:23:09