Skip to content

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