smoking-detection

🚬 Smoke Detection with YOLOv11-Medium

This repository contains a custom-trained YOLOv11-Medium object detection model designed to detect cigarette smoke in images and videos. It is ideal for use in surveillance systems, public safety, and smoking zone enforcement.

---

📊 Model Performance

Evaluated on a custom Roboflow dataset using YOLOv11 medium variant trained for optimal balance of accuracy and speed.

---

🧟‍♂️ Model Details

• Model: YOLOv11-Medium
• Task: Object Detection
• Classes: cigarette
• Framework: Ultralytics YOLOv11
• Training Source: Roboflow Universe

---

📁 Dataset

• Name: Cigarette Smoke Detection
• Source: Roboflow Dataset
• Format: YOLO
• Annotations: Bounding Boxes

---

💡 Usage

1. Load the Model

from ultralytics import YOLO

model = YOLO("path/to/best.pt")  # Replace with your model path

2. Run Inference on an Image

results = model("your_image.jpg", save=True)
results.show()

3. Run Inference on a Video

results = model("your_video.mp4", stream=True)
for r in results:
    r.show()

---

model.export(format="onnx")

---

📜 License

This model is available under the MIT License. Refer to the LICENSE file for more details.

---

🤖 Author

Uploaded and maintained by: [Uppada Enos] Contact: [[email protected]]

---

🌐 Model Hub

You can access and test this model directly via Hugging Face Spaces or API once published.