Skip to content

Add Pneumonia Detection AI model under medical imaging (#252)#281

Open
2004ARYAN wants to merge 1 commit intonidhaloff:masterfrom
2004ARYAN:medical-ai-pneumonia
Open

Add Pneumonia Detection AI model under medical imaging (#252)#281
2004ARYAN wants to merge 1 commit intonidhaloff:masterfrom
2004ARYAN:medical-ai-pneumonia

Conversation

@2004ARYAN
Copy link

📌 Summary

This PR adds support for Pneumonia Detection AI using chest X-ray images as part of issue #252 — “Medical Imaging AI”.


✅ What's Included

  • pneumonia_model.py — Torch-based classifier with ResNet18
  • pneumonia_model.pth — Trained weights file
  • config.yaml — Model configuration file for IGEL integration
  • infer.py — Example script to run inference on chest X-rays
  • pneumonia_notebook.ipynb — Kaggle notebook used for training & reference

📂 Folder Structure

igel/
├── medical/
│ └── models/
│ ├── pneumonia_model.py
│ ├── pneumonia_model.pth
│ └── config.yaml
examples/
└── pneumonia_detection/
├── infer.py
igel/
└── examples/
└── pneumonia_detection/
└── pneumonia_notebook.ipynb


🧪 Test Instructions

Run the model with:

python examples/pneumonia_detection/infer.py path_to_chest_xray.jpg

@2004ARYAN
Copy link
Author

Hi @nidhaloff 👋

This PR adds pneumonia detection support using a pretrained ResNet model trained on chest X-rays (see #252). It includes:

  • Model weights (.pth)
  • Inference script
  • Config file
  • Example usage
  • Kaggle notebook for full training reference

Let me know if you’d like ONNX export or evaluation script as a follow-up. Excited to contribute to medical AI with IGEL!

Thanks 🙌
– Aryan

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant