|
|
---
|
|
|
tags:
|
|
|
- image-classification
|
|
|
- pytorch
|
|
|
- mnist
|
|
|
license: apache-2.0
|
|
|
library_name: pytorch
|
|
|
pipeline_tag: image-classification
|
|
|
---
|
|
|
|
|
|
# MNIST Digit Classifier
|
|
|
|
|
|
A convolutional neural network trained on MNIST to classify digits 0-9.
|
|
|
|
|
|
## Usage
|
|
|
|
|
|
```python
|
|
|
from src.model import DigitClassifier
|
|
|
import torch
|
|
|
|
|
|
model = DigitClassifier()
|
|
|
model.load_state_dict(torch.load("model_weights.pth"))
|
|
|
model.eval()
|
|
|
|
|
|
# Preprocessing (same as training):
|
|
|
transform = transforms.Compose([
|
|
|
transforms.Resize((28, 28)),
|
|
|
transforms.Grayscale(),
|
|
|
transforms.ToTensor(),
|
|
|
transforms.Normalize((0.1307,), (0.3081,))
|
|
|
])
|
|
|
``` |
