smoke_cigarette-detection2-yolo11m

Fine-tuned from ultralytics/yolov11m (YOLOv11-Medium) using Ultralytics.

Classes

• cigarette

Dataset

• Source: smoking-detection-5e8hh/smoking-detection-ggnhq (version 1)
• Images: 1636 train / 461 valid / 240 test
• Check the dataset page above for its license -- not necessarily the same as this repo's license field, which reflects the base model's license.

Training

Parameter	Value
Base checkpoint	yolo11m.pt
Epochs	100
Image size	640
Batch size	18
Optimizer	AdamW
Initial LR (lr0)	0.001
Patience (early stop)	20

Results (held-out validation split)

Metric	Value
mAP50	0.9943
mAP50-95	0.7421
Precision	0.9811
Recall	0.9835

Usage

from ultralytics import YOLO

model = YOLO("best.pt")
results = model.predict("image.jpg", conf=0.25)
for r in results:
    for box in r.boxes:
        print(model.names[int(box.cls[0])], float(box.conf[0]))