Nhận diện đám cháy và thông báo qua telegram

Sơ qua dự án

Dự án sử dụng mạng YOLO train tận dụng lại model pre-trained YOLO11s.pt và bộ dataset về các đám lửa cháy đã được gán nhãn tạo nên model có thể phát hiện lửa và các đám cháy

Thông tin bộ dataset có thể được tải xuống ở đây.

Demo kết quả

Khi phát hiện ra lửa

Khi Không có lửa

Cách chạy

Cài thư viện cần thiết:

• Mở terminal ở thư mục vừa clone và chạy đoạn lệnh sau:

  pip install -r 'requirements.txt' 

Chạy thử code

• Chạy nhanh demo theo data video có sẳn:

  Chỉ cần chạy file main.py

• Chạy theo camera local realtime:

  • B1: thay dòng lệnh sau

    cam = cv2.VideoCapture('test.mp4)
    ở dòng 8 file main.py thành

    cam = cv2.VideoCapture(0)
  • B2: nhấn run để chạy.

Tự train lại cho để model tốt hơn

Model model.pt trên được train dựa trên pre_train model YOLO11n.py với epochs là 200. Nếu cấu hình GPU khỏe có thể tự train với số epochs lớn hơn 200.

Tích hợp báo cháy thông qua telegram

• Thay api_key và id của telegram vào 2 dòng này trong code ở file main.py để có thể báo về telegram của bạn

  api_key = '7278180996:AAF3zjRmDm2tpTYzl5W1rRXMfTBkt47xWBA'
  id = '5510302349'

  

  Thông báo qua telegram