image_brightest.py

import cv2
import os, sys

class Window():
    def __init__(self, name='fire detection'):
        self.name = name
        self.win = cv2.namedWindow(self.name)

    def find_brightest(self, img, radius):
        #convert to greyscale
        grey = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

        grey = cv2.GaussianBlur(grey, (radius, radius), 0)
        #use Gaussian blur to average out everything and supress random bright spots(eg. lights)
        (min_val, max_val, min_loc, max_loc) = cv2.minMaxLoc(grey)
        #circle the spot

        return max_loc

    def show_fire(self, img, radius=5):
        bspot = self.find_brightest(img, radius)
        print(bspot)
        #show brightest spot, bspot is pair of coords
        cv2.circle(img, bspot, radius, (237, 223, 68), 2)

        #update window with image whose fire is circled
        cv2.imshow(self.name, img)

if __name__ == "__main__":
    win = Window()

    print('enter path of image as command line arg')
    im_path = sys.argv[1]
    img = cv2.imread(im_path)

    radius = int(input('radius of detection circle in pixels: '))

    key_hit = None
    win.show_fire(img, radius)
    while key_hit != ord('q'):
        key_hit = cv2.waitKey(1)

    cv2.destroyAllWindows()