HehealthVision/ViT-GenitalPathoDx

ViT_Genital_Patho_Dx

A Vision Transformer-based model for multi-class classification of penile-related pathological images, including both STD and non-STD conditions. This model is designed for diagnostic support and academic research in digital health and dermatology.

---

🧠 Model Overview

The ViT_Genital_Patho_Dx model classifies penile region images into six disease categories:

• Genital_warts
• HSV (Herpes Simplex Virus)
• Normal
• Penile_cancer
• Penile_candidiasis
• Syphillis

It is fine-tuned from google/vit-base-patch16-224-in21k and trained on a curated medical image dataset using TensorFlow and vit-keras.

---

📦 Model Metadata

Field	Value
License	CreativeML Open RAIL-M
Base model	google/vit-base-patch16-224-in21k`
Model type	Vision Transformer (ViT-B16)
Pipeline tag	image-classification`
Trained by	Yudara Kularathne,Janitha Prathapa, Thanveer Ahamad
Repository	GitHub Repo
Demo	Available on request

---

🧠 Model Architecture

This project uses:

• ViT-B16 (pretrained on ImageNet21k) as a base
• Custom classification head: Flatten -> Dense(6, softmax)
• Fine-tuned on a proprietary dataset of penile pathology images, including both STD and non-STD conditions

👨‍💻 Authors

• Yudara Kularathne
• Janitha Prathapa
• Thanveer Ahamad

📬 License

This project is licensed under the CC BY-NC 4.0 License.
Commercial use is prohibited without prior permission. For more details, see the LICENSE file.

📚 Citation

BibTeX:

@misc{genitalpathodx2024,
  title={ViT_Genital_Patho_Dx: Transformer-based Genital Pathology Classifier},
  author={Yudara Kularathn, Janitha Prathapa and Thanveer Ahamad},
  year={2024},
  howpublished={\url{https://huggingface.co/HehealthVision/ViT-GenitalPathoDx}},
}

APA: Yudara Kularathne, Janitha Prathapa, & Thanveer Ahamad. (2024). ViT_Genital_Patho_Dx: Transformer-based Genital Pathology Classifier. Hugging Face. https://huggingface.co/HehealthVision/ViT-GenitalPathoDx