YOLO Fire & Smoke Detection Model#

1
from ultralytics import YOLO
2
import cv2
3

4
# โหลดโมเดล
5
model = YOLO("fire_smoke_model.pt")
6

7
# ทำการตรวจจับจากรูปภาพ
8
results = model.predict(source="image.jpg", conf=0.5, save=True)
9

10
# แสดงผลลัพธ์
11
for result in results:
12
    boxes = result.boxes
13
    for box in boxes:
14
        cls = int(box.cls[0])
15
        conf = float(box.conf[0])
16
        class_name = model.names[cls]
17
        print(f"พบ: {class_name} ความมั่นใจ: {conf:.2f}")

1
from ultralytics import YOLO
2
import cv2
3

4
model = YOLO("fire_smoke_model.pt")
5
cap = cv2.VideoCapture(0)
6

7
while True:
8
    ret, frame = cap.read()
9
    if not ret:
10
        break
11

12
    results = model.predict(frame, conf=0.5, verbose=False)
13
    annotated_frame = results[0].plot()
14

15
    cv2.imshow("Fire & Smoke Detection", annotated_frame)
16

17
    if cv2.waitKey(1) & 0xFF == ord('q'):
18
        break
19

20
cap.release()
21
cv2.destroyAllWindows()

1
from ultralytics import YOLO
2

3
model = YOLO("fire_smoke_model.pt")
4

5
results = model.predict(
6
    source="video.mp4",
7
    conf=0.5,
8
    save=True,
9
    show=True
10
)

1
from ultralytics import YOLO
2

3
model = YOLO("fire_smoke_model.pt")
4
model.export(format="onnx")