使用 Python 和 OpenCV 实现火焰检测(附源码)

https://github.com/mushfiq1998/fire-detection-python-opencv?source=post_page-----e55c8fc6fa54--------------------------------

import cv2         # Library for openCVimport threading   # Library for threading -- which allows code to run in backendimport playsound   # Library for alarm soundimport smtplib     # Library for email sending# To access xml file which includes positive and negative images of fire. # (Trained images) File is also provided with the code.fire_cascade = cv2.CascadeClassifier('fire_detection_cascade_model.xml') vid = cv2.VideoCapture("videos\\fire2.mp4")runOnce = False # created boolean# defined function to play alarm post fire detection using threadingdef play_alarm_sound_function():     # to play alarm # mp3 audio file is also provided with the code.    playsound.playsound('fire_alarm.mp3',True)     print("Fire alarm end") # to print in consol# Defined function to send mail post fire detection using threadingdef send_mail_function():     recipientmail = "add recipients mail" # recipients mail    recipientmail = recipientmail.lower() # To lower case mail    try:        server = smtplib.SMTP('smtp.gmail.com', 587)        server.ehlo()        server.starttls()        # Senders mail ID and password        server.login("add senders mail", 'add senders password')         # recipients mail with mail message        server.sendmail('add recipients mail', recipientmail, "Warning fire accident has been reported")          # to print in consol to whome mail is sent        print("Alert mail sent sucesfully to {}".format(recipientmail))        server.close() ## To close server    except Exception as e:        print(e) # To print error if anywhile(True):    Alarm_Status = False    # Value in ret is True # To read video frame    ret, frame = vid.read()     # To convert frame into gray color    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)     # to provide frame resolution    fire = fire_cascade.detectMultiScale(frame, 1.2, 5)     ## to highlight fire with square     for (x,y,w,h) in fire:        cv2.rectangle(frame,(x-20,y-20),(x+w+20,y+h+20),(255,0,0),2)        roi_gray = gray[y:y+h, x:x+w]        roi_color = frame[y:y+h, x:x+w]        print("Fire alarm initiated")        # To call alarm thread        threading.Thread(target=play_alarm_sound_function).start()          if runOnce == False:            print("Mail send initiated")            # To call alarm thread            threading.Thread(target=send_mail_function).start()             runOnce = True        if runOnce == True:            print("Mail is already sent once")            runOnce = True    cv2.imshow('frame', frame)    if cv2.waitKey(1) & 0xFF == ord('q'):        break

python -m venv myenv

myenv\Scripts\activate

pip install playsound

pip install opencv-python