added medical waste detector

.gitignore

@@ -0,0 +1,11 @@
+flagged/
+*.png
+*.jpg
+*.mp4
+*.jpeg
+*.mkv
+.DS_Store
+gradio_cached_examples/
+venv/
+__pycache__
+flagged

README.md

@@ -1,14 +1,70 @@
----
-title: Medical Waste Detector
-emoji: 🚀
-colorFrom: gray
-colorTo: yellow
-sdk: gradio
-sdk_version: 5.25.2
-app_file: app.py
-pinned: false
-license: mit
-short_description: Detection of Medical Waste
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 🩺 Medical Waste Classifier
+
+This project uses a YOLOv8 model with a Gradio interface to detect and classify medical waste types based on an uploaded image.
+
+## 💡 Features
+
+- Upload medical waste images
+- Object detection using YOLOv8
+- Classification by:
+  - Waste name
+  - Waste type (Infectious, Pathological, etc.)
+  - Color code (Red, Blue, Green)
+
+## 🧠 Model
+
+Make sure your YOLOv8 model is trained on the relevant medical waste classes and saved as `best.pt`.
+
+## 📁 Folder Structure
+
+```
+medical-waste-detector/
+├── model/
+│   └── best.pt
+├── .env
+├── app.py
+├── labels.json
+├── requirements.txt
+└── README.md
+```
+
+## ⚙️ Setup Instructions
+
+1. **Clone the project**  
+```bash
+git clone https://github.com/sumn2u/medical-waste-detector
+cd medical-waste-detector
+```
+
+2. **Install dependencies**  
+```bash
+pip install -r requirements.txt
+```
+
+3. **Run the app**  
+```bash
+python app.py
+```
+
+## 🧾 Example Labels
+
+```json
+{
+  "Glove Pair Latex": ["Infectious Waste", "Red"],
+  "Glass Equipment Packaging": ["Non Biodegradable", "Blue"]
+}
+```
+
+## 🧼 Waste Color Codes
+
+| Type                             | Color Code |
+|----------------------------------|------------|
+| Pathological Waste               | Red        |
+| Infectious Waste                 | Red        |
+| General/Biodegradable Waste      | Green      |
+| General/Non-Biodegradable Waste  | Blue       |
+| Infectious Plastic Waste         | Red        |
+
+## 🔒 License
+
+MIT License

app.py

@@ -0,0 +1,59 @@
+import gradio as gr
+from ultralytics import YOLO
+from PIL import Image
+import json
+
+# Load classification model and label metadata
+model = YOLO("model/best.pt")
+with open("labels.json") as f:
+    labels_info = json.load(f)
+
+# Helper: Try to best match the class name
+def get_label_info(name):
+    name = name.replace("_", " ")
+    if name in labels_info:
+        return labels_info[name]
+    title_name = name.title()
+    if title_name in labels_info:
+        return labels_info[title_name]
+    capitalized_name = " ".join(word.capitalize() for word in name.split())
+    if capitalized_name in labels_info:
+        return labels_info[capitalized_name]
+    print(f"Label '{name}' not found. Available labels are: {list(labels_info.keys())}")
+    return ["Unknown", "Unknown"]
+
+# Classification function
+def classify_image(img, conf_threshold= 0.25, iou_threshold=0.45):
+    results = model.predict(source=img, conf=conf_threshold, iou=iou_threshold)
+    if not results:
+        return None, [["No prediction", "-", "-"]]
+    result = results[0]
+    annotated_img = result.plot()
+    if result.probs is not None:
+        probs = result.probs.data.tolist()
+        max_index = probs.index(max(probs))
+        raw_name = model.names[max_index]
+        waste_type, color_code = get_label_info(raw_name)
+        waste_details = [[raw_name, waste_type, color_code]]
+    else:
+        waste_details = [["No classification", "-", "-"]]
+    return Image.fromarray(annotated_img), waste_details
+
+# Gradio interface
+iface = gr.Interface(
+    fn=classify_image,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        # gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence Threshold"),
+        # gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU Threshold"),
+    ],
+    outputs=[
+        gr.Image(label="Result"),
+        gr.Dataframe(headers=["Name", "Type", "Color Code"], label="Details")
+    ],
+    title="Medical Waste Detection",
+    description="DWaste uses advanced AI to classify and sort medical waste into various categories to streamline waste management processes for healthcare facilities. Upload an image to classify the medical waste type.<p><strong>Disclaimer:</strong> This tool is for informational purposes only. Predictions made by the AI model may not always be accurate. Please use the results cautiously and verify if necessary.</p>",
+    article="<h3>Waste Classification Table</h3><p>Here is the mapping of waste items with their associated types and color codes:</p><table border='1'><tr><th>Item</th><th>Type</th><th>Color Code</th></tr><tr><td>Mask</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Gauze</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Glass Equipment Packaging</td><td>Non Biodegradable</td><td>Blue</td></tr><tr><td>Glove Pair Latex</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Glove Pair Nitrile</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Glove Pair Surgery</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Glove Single Latex</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Glove Single Nitrile</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Glove Single Surgery</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Gloves</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Mask</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Medical Cap</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Medical Glasses</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Metal Equipment Packaging</td><td>General/ Non-Biodegradable Waste</td><td>Blue</td></tr><tr><td>Organic Waste</td><td>General/ Biodegradable Waste</td><td>Green</td></tr><tr><td>Paper Equipment Packaging</td><td>General/ Non-Biodegradable Waste</td><td>Blue</td></tr><tr><td>Plastic Equipment Packaging</td><td>General/ Non-Biodegradable Waste</td><td>Blue</td></tr><tr><td>Shoe Cover Pair</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Shoe Cover Single</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Syringe</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Syringe Needle</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Test Tube</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Tweezers</td><td>Infectious Waste</td><td>Red</td></tr><tr><td>Urine Bag</td><td>Infectious Plastic Waste</td><td>Red</td></tr></table><br><h3>General Waste Categories</h3><p>Here is the mapping of waste groups, types, color codes, and categories:</p><table border='1'><tr><th>Group</th><th>Type</th><th>Color Code</th><th>Category</th></tr><tr><td>General</td><td>Bio-Degradable</td><td>Green</td><td>Organic wastes; peels of vegetables and fruits; rotten or stale foods</td></tr><tr><td>General</td><td>Non-Biodegradable</td><td>Blue</td><td>Paper(A4, Dublex- Medicine Covers, Normal Papers), Plastic(Polythene Bags, Wrappers, Packaging Plastic Materials, Cling Wraps), Bottles (Water, Soft Drinks), vials, saline Bottles</td></tr><tr><td>Infectious</td><td>Infectious</td><td>Red</td><td>Cotton, Gauze, Used Gloves, IV sets, blood bags, urine bag, used PPE, diapers, pad, sputum containers, collection tubes, Dialyzer</td></tr><tr><td>Infectious</td><td>Pathological</td><td>Red</td><td>Body organs or tissues, placenta</td></tr><tr><td>Infectious</td><td>Cytotoxic</td><td>Red</td><td>Unused or Expired cytotoxic drugs, chemotherapy agents</td></tr><tr><td>Infectious</td><td>Pharmaceutical</td><td>Red</td><td>Expired medicines</td></tr><tr><td>Infectious</td><td>Sharps</td><td>Red</td><td>Needles, Blades, Scalpels, Broken glass, Lancets, Suture needles</td></tr><tr><td>Infectious</td><td>Chemical</td><td>Yellow</td><td>Discarded chemicals, reagents, solvents, disinfection</td></tr><tr><td>Infectious</td><td>Radioactive</td><td>Black</td><td>X-Ray films, radio isotopes</td></tr></table><br><p>For more information about Dwaste, visit our website: <a href='http://dwaste.live' target='_blank'>dwaste.live</a></p>"
+)
+
+iface.launch()

labels.json

@@ -0,0 +1,27 @@
+{
+    "Body Tissue or Organ": ["Pathological Waste", "Red"],
+    "Gauze": ["Infectious Waste", "Red"],
+    "Glass Equipment Packaging": ["Non Biodegradable", "Blue"],
+    "Glove Pair Latex": ["Infectious Waste", "Red"],
+    "Glove Pair Nitrile": ["Infectious Waste", "Red"],
+    "Glove Pair Surgery": ["Infectious Waste", "Red"],
+    "Glove Single Latex": ["Infectious Waste", "Red"],
+    "Glove Single Nitrile": ["Infectious Waste", "Red"],
+    "Glove Single Surgery": ["Infectious Waste", "Red"],
+    "Gloves": ["Infectious Waste", "Red"],
+    "Mask": ["Infectious Waste", "Red"],
+    "Medical Cap": ["Infectious Waste", "Red"],
+    "Medical Glasses": ["Infectious Waste", "Red"],
+    "Metal Equipment Packaging": ["General/ Non-Biodegradable Waste", "Blue"],
+    "Organic Waste": ["General/ Biodegradable Waste", "Green"],
+    "Paper Equipment Packaging": ["General/ Non-Biodegradable Waste", "Blue"],
+    "Plastic Equipment Packaging": ["General/ Non-Biodegradable Waste", "Blue"],
+    "Shoe Cover Pair": ["Infectious Waste", "Red"],
+    "Shoe Cover Single": ["Infectious Waste", "Red"],
+    "Syringe": ["Infectious Waste", "Red"],
+    "Syringe Needle": ["Infectious Waste", "Red"],
+    "Test Tube": ["Infectious Waste", "Red"],
+    "Tweezers": ["Infectious Waste", "Red"],
+    "Urine Bag": ["Infectious Plastic Waste", "Red"]
+  }
+  

model/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e33ef8197d3df71798b42e5990687117b84311f600d0ad46acd93a0cff7e929
+size 3020801

requirements.txt

@@ -0,0 +1,6 @@
+gradio==4.10.0
+ultralytics
+Pillow
+pydantic==2.8.2
+pydantic-core==2.20.1
+fastapi==0.112.4