File size: 4,380 Bytes
01590eb b3bb92a f100315 b3bb92a f100315 b3bb92a f100315 b3bb92a f100315 b3bb92a f100315 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | ---
license: mit
tags:
- image-classification
- pneumonia-detection
- healthcare
- medical-imaging
- pytorch
- resnet18
library_name: pytorch
model_name: resnet18_pneumonia_classifier
---
# ResNet18 Pneumonia Detection Model
This model is a fine-tuned version of the ResNet18 architecture for pneumonia detection. It was trained on the [Kaggle Chest X-ray Pneumonia dataset](https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia), which includes images of normal lungs and lungs with pneumonia. The model is capable of distinguishing between **Pneumonia** and **Normal** chest X-rays.
## Model Details
- **Model Architecture**: ResNet18
- **Input Size**: 224 x 224
- **Number of Classes**: 2 (Pneumonia, Normal)
- **Framework**: PyTorch
- **Training Dataset**: [Kaggle Chest X-ray Pneumonia Dataset](https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia)
- **Library**: PyTorch
## Model Performance
- **Accuracy**: 83.3%
- **Loss**: 0.2459
## Intended Use
This model is designed to assist healthcare professionals in identifying pneumonia from chest X-ray images. It should not be used as a sole diagnostic tool but as a supplement to medical expertise.
## Training Details
The model was trained using the following setup:
- **Architecture**: ResNet18 (Pre-trained on ImageNet)
- **Optimizer**: SGD (Stochastic Gradient Descent)
- **Learning Rate**: 0.001
- **Momentum**: 0.9
- **Loss Function**: CrossEntropyLoss
- **Batch Size**: 32
- **Data Augmentation**:
- Random Rotation (±30 degrees)
- Random Zoom (20%)
- Random Horizontal Shift (±10% width)
- Random Vertical Shift (±10% height)
- Random Horizontal Flip
- **Training Epochs**: 1
- **Evaluation Metric**: Cross Entropy Loss
### Augmentation Details
The dataset was augmented during training with the following transformations:
- Randomly rotated some training images by 30 degrees
- Randomly zoomed some training images by 20%
- Randomly shifted images horizontally by 10% of the width
- Randomly shifted images vertically by 10% of the height
- Randomly flipped images horizontally
## How to Use the Model
You can use this model with the `transformers` and `torch` libraries.
```python
import torch
from huggingface_hub import hf_hub_download
from torchvision import transforms
from PIL import Image
import requests
# Download the model weights from Hugging Face Hub
model_path = hf_hub_download(repo_id="izeeek/resnet18_pneumonia_classifier", filename="resnet18_pneumonia_classifier.pth")
# Load the model architecture (ResNet18)
model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=False)
# Adjust the final layer for binary classification (if necessary)
model.fc = torch.nn.Linear(model.fc.in_features, 2)
# Load the downloaded weights
model.load_state_dict(torch.load(model_path))
# Set the model to evaluation mode
model.eval()
# Image preprocessing
transform = transforms.Compose([
transforms.Grayscale(num_output_channels=3),
transforms.Resize((224, 224)),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])
# Sample Image (replace with your own image URL)
url = 'https://storage.googleapis.com/kagglesdsdata/datasets/17810/23812/chest_xray/test/NORMAL/IM-0005-0001.jpeg?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=databundle-worker-v2%40kaggle-161607.iam.gserviceaccount.com%2F20240913%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20240913T014624Z&X-Goog-Expires=345600&X-Goog-SignedHeaders=host&X-Goog-Signature=1f6b37d181f12d083ffc951657e85fea087bb4e81ab955ec955dafcdae49c0d53ce20bc0be93605e2672b9bdd59e752eba9d5a3a0da2e3b3a03c888580b88d63d87611b4e4cec8b8802d53abd53fda165dd04765b8d9f30ddd4e908cd7a2a389ce8244fca7bfa36b3c9cff79d7c5e3f9ee7d59d5b9ef97a2e5c083997892ee3023302313fafff48ded58232db57d6affcfaee704eebba55f2b0abac40b14a38137275ad19cdb1b787930d134f7c30710e29c409bd765ca02e46851470a871cc697f614d464086373f43f5462f241eaf023cfd31e217d7b11e24e1ff34857deb200f5dc1a8c28c8115048ee840be8481f1bd79a2d8e2de1b30cb71420c007d32c'
img = Image.open(requests.get(url, stream=True).raw)
# Preprocess the image
input_img = transform(img).unsqueeze(0)
# Inference
with torch.no_grad():
output = model(input_img)
_, predicted = torch.max(output, 1)
# Labels for classification
labels = {0: 'Pneumonia', 1: 'Normal'}
print(f'Predicted label: {labels[predicted.item()]}') |