🦷 Oral Diseases Image Classification

ResNet50 fine-tuned to classify 6 intraoral conditions from a single photo — benchmarked against 3 other architectures, evaluated on a held-out test set.

🏆 94.77% Accuracy · 0.9411 F1-Score (macro)

Model Description

This repository hosts the winning checkpoint from a 4-way benchmark of image classifiers trained to recognize intraoral conditions:

Calculus · Caries · Gingivitis · Ulcers · Tooth Discoloration · Hypodontia

Four architectures were trained under identical conditions (same data split, same augmentation, same evaluation protocol) and compared on a test set none of them saw during training:

ResNet50 (ImageNet-pretrained, fine-tuned in two stages — freeze then unfreeze) came out on top and is the model served by Gradio.py and packaged as checkpoints/best_model.pth.

Intended Use & Limitations

This model is a research and educational tool for preliminary visual screening. It is not a certified diagnostic device and must not be used to replace examination by a licensed dentist or physician. Performance depends on image quality and lighting similar to the training data, and may not generalize to conditions or populations outside the training distribution.

Files in this Repository

How to Use

Load directly with huggingface_hub

from huggingface_hub import hf_hub_download
import torch

weights_path = hf_hub_download(
    repo_id="nsr51324/Oral_Diseases_Image_Classification",
    filename="checkpoints/best_model.pth"
)

checkpoint = torch.load(weights_path, map_location="cpu")
class_names = checkpoint["class_names"]

Rebuild the model and run inference

import torch.nn as nn
from torchvision.models import resnet50
from torchvision import transforms
from PIL import Image

model = resnet50(weights=None)
model.fc = nn.Sequential(
    nn.Dropout(0.3),
    nn.Linear(model.fc.in_features, len(class_names))
)
model.load_state_dict(checkpoint["state_dict"])
model.eval()

transform = transforms.Compose([
    transforms.Resize((224, 224)),
    transforms.ToTensor(),
    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
])

image = Image.open("sample.jpg").convert("RGB")
tensor = transform(image).unsqueeze(0)

with torch.no_grad():
    probs = torch.softmax(model(tensor), dim=1)[0]

pred = class_names[probs.argmax().item()]
print(f"{pred}: {probs.max().item()*100:.2f}%")

Run the interactive demo

pip install torch torchvision gradio pillow huggingface_hub
python Gradio.py

Point MODEL_PATH at the top of Gradio.py to your local copy of checkpoints/best_model.pth.

Training Data

Trained on nsr51324/Oral_Diseases, sourced from the Oral Diseases dataset on Kaggle (salmansajid05), split 80/10/10 (train/val/test) with stratification to preserve class balance across all three sets.

Training Procedure

• Image size: 224×224 · Batch size: 32 · Epochs: up to 30 (early stopping)
• Two-stage fine-tuning: backbone frozen for 5 epochs (head-only training, lr=1e-3), then fully unfrozen for fine-tuning at lr=1e-5
• Regularization: dropout (0.4), weight decay (1e-4), label smoothing (0.1), early stopping on validation loss
• Augmentation (train split only): random resized crop, horizontal flip, rotation, color jitter, random erasing

Evaluation

Full classification report, confusion matrix, and training curves for all 4 models are in outputs/. Summary metrics are in outputs/models_comparison.csv.

Disclaimer

This model is provided for research and educational purposes only. It is not intended for clinical decision-making. Always consult a qualified healthcare professional for medical diagnosis.

License

MIT. The underlying dataset has its own terms — see the dataset card before commercial use.

Author

Nasr Mohamed — AI Engineer 🤗 huggingface.co/nsr51324