Skip to content

The model is based on a YOLOv8 (Deep learning Neural network architecture) and is trained on the publicly available dataset, which consists of lung CT scans of patients with and without lung cancer.

License

Notifications You must be signed in to change notification settings

bedead/lung-cancer-classification-yoloV8---gradio

Repository files navigation

🫁 Lung Cancer Classification (YOLOv8)

project-banner


✅ Project Overview

Lung Cancer Detection is a deep-learning project that classifies lung CT scan images into three categories based on cancer risk.
The model leverages YOLOv8, a powerful neural network architecture designed for image classification and object detection.

The dataset contains 1,054 medical lung scan images, curated and hosted on Roboflow Universe, and a trained model is already available for experimentation and inference.


🩺 Classification Categories

Class Type Description
Normal Lungs showing no visible abnormalities or cancer-related nodules.
Benign Non-cancerous nodules or minor abnormalities detected.
Malignant Cancerous nodules or signs strongly indicating lung cancer.

📊 Dataset & Model Details

Parameter Value
Dataset Version v1
Total Images 1,054
Model Version YOLOv8 Classification Model (v1)
Dataset Host Roboflow Universe

The dataset and current model release are stable. Future dataset updates and incremental versions are planned.


🌐 Demo & Resources

🔹 Live Demo (Gradio App via Hugging Face Spaces):
👉 https://huggingface.co/spaces/bedead/Lung-cancer-classification

🔹 Dataset (Roboflow Universe):
👉 https://universe.roboflow.com/satyam-mishra-gfl0c/lung-cancer-detection-afeuf


🤝 Contribution & Labeling Guidelines

To maintain dataset quality and medical reliability:

  • Use only anonymized and ethically sourced medical images.
  • Label using the following categories only: Normal, Benign, Malignant.
  • Ensure image clarity (CT/X-ray format recommended).
  • Add metadata only if anonymized and ethically permissible.
  • Expert cross-verification is recommended due to medical sensitivity.

📧 For contribution or dataset collaboration, contact:
Satyam Mishra — [email protected]


🧾 Patent Information

Patent Application Number: 202341039647
This project and methodology are protected under Indian patent application laws.


🚀 Tech Stack

  • YOLOv8 (Ultralytics)
  • Python
  • Roboflow
  • Gradio (Demo UI)
  • Hugging Face Spaces

⭐ Support

If you found this useful, don’t forget to ⭐ star the repository!


About

The model is based on a YOLOv8 (Deep learning Neural network architecture) and is trained on the publicly available dataset, which consists of lung CT scans of patients with and without lung cancer.

Topics

Resources

License

Stars

Watchers

Forks