# OpenMed Medical Imaging Models

This repository contains trained ResNet50 models for medical imaging tasks from the OpenMed platform.

## Models Included

### Brain Tumor Classification
- **Model**: `resnet50_brain_tumor_full`
- **Classes**: brain_glioma, brain_menin, brain_tumor
- **Dataset**: Brain Cancer MRI Dataset
- **Description**: ResNet50 model fine-tuned for brain tumor classification in MRI scans

### Tuberculosis Detection
- **Model**: `resnet50_tb_full`
- **Classes**: Normal, Tuberculosis
- **Dataset**: TB Chest Radiography Database
- **Description**: ResNet50 model fine-tuned for tuberculosis detection in chest X-rays

### Pneumonia Detection
- **Model**: `resnet50_pneumonia_full`
- **Classes**: Normal, Pneumonia
- **Dataset**: Chest X-ray Pneumonia Dataset
- **Description**: ResNet50 model fine-tuned for pneumonia detection in chest X-rays

## Quick Start

```python
from huggingface_hub import hf_hub_download
import torch

# Download a specific model (example: brain tumor classification)
model_path = hf_hub_download(
    repo_id="hitmanonholiday/openmed-medical-imaging-models",
    filename="resnet50_brain_tumor_full/pytorch_model.bin"
)

# Load the model
checkpoint = torch.load(model_path, map_location="cpu")
```

## Model Architecture

All models are based on ResNet50 with ImageNet pre-trained weights:
- **Input Size**: 224x224x3 RGB images
- **Normalization**: ImageNet statistics
- **Fine-tuning**: Full network training (all layers trainable)

## Usage Guidelines

**Important**: These models are for research and educational purposes only. They should **NOT** be used for actual medical diagnosis without proper validation and oversight by qualified healthcare professionals.

## Performance Metrics

Performance metrics for each model can be found in their respective model cards.

## Citation

```bibtex
@misc{openmed_models_2024,
  title={OpenMed: AI-Powered Medical Imaging Analysis Platform},
  author={OpenMed Team},
  year={2024},
  howpublished={\url{https://huggingface.co/hitmanonholiday/openmed-medical-imaging-models}}
}
```

## License

MIT License - See individual model cards for details.

---

**Disclaimer**: These models are for research and educational purposes only. They should not be used for medical diagnosis or treatment decisions without proper validation and oversight by qualified healthcare professionals.