Update project with minor fixes and improvements

application.py

@@ -13,72 +13,79 @@ check_settings()
 VIDEO = VideoStreaming()
 
 
-@application.route('/')
+@application.route("/")
 def home():
-    TITLE = 'Object detection'
+    TITLE = "Object detection"
-    return render_template('index.html', TITLE=TITLE)
+    return render_template("index.html", TITLE=TITLE)
 
 
-@application.route('/video_feed')
+@application.route("/video_feed")
 def video_feed():
-    '''
+    """
     Video streaming route.
-    '''
+    """
     return Response(
         VIDEO.show(),
-        mimetype='multipart/x-mixed-replace; boundary=frame'
+        mimetype="multipart/x-mixed-replace; boundary=frame"
     )
 
 
-# Button requests called from ajax
+# * Button requests
-@application.route('/request_preview_switch')
+@application.route("/request_preview_switch")
 def request_preview_switch():
     VIDEO.preview = not VIDEO.preview
-    print('*'*10, VIDEO.preview)
+    print("*"*10, VIDEO.preview)
     return "nothing"
 
-@application.route('/request_flipH_switch')
+
+@application.route("/request_flipH_switch")
 def request_flipH_switch():
     VIDEO.flipH = not VIDEO.flipH
-    print('*'*10, VIDEO.flipH)
+    print("*"*10, VIDEO.flipH)
     return "nothing"
 
-@application.route('/request_model_switch')
+
+@application.route("/request_model_switch")
 def request_model_switch():
     VIDEO.detect = not VIDEO.detect
-    print('*'*10, VIDEO.detect)
+    print("*"*10, VIDEO.detect)
     return "nothing"
 
-@application.route('/request_exposure_down')
+
+@application.route("/request_exposure_down")
 def request_exposure_down():
     VIDEO.exposure -= 1
-    print('*'*10, VIDEO.exposure)
+    print("*"*10, VIDEO.exposure)
     return "nothing"
 
-@application.route('/request_exposure_up')
+
+@application.route("/request_exposure_up")
 def request_exposure_up():
     VIDEO.exposure += 1
-    print('*'*10, VIDEO.exposure)
+    print("*"*10, VIDEO.exposure)
     return "nothing"
 
-@application.route('/request_contrast_down')
+
+@application.route("/request_contrast_down")
 def request_contrast_down():
     VIDEO.contrast -= 4
-    print('*'*10, VIDEO.contrast)
+    print("*"*10, VIDEO.contrast)
     return "nothing"
 
-@application.route('/request_contrast_up')
+
+@application.route("/request_contrast_up")
 def request_contrast_up():
     VIDEO.contrast += 4
-    print('*'*10, VIDEO.contrast)
+    print("*"*10, VIDEO.contrast)
     return "nothing"
 
-@application.route('/reset_camera')
+
+@application.route("/reset_camera")
 def reset_camera():
     STATUS = reset_settings()
-    print('*'*10, STATUS)
+    print("*"*10, STATUS)
     return "nothing"
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     application.run(debug=True)

camera_settings.py

@@ -1,48 +1,51 @@
 import os
 import cv2
 
+
 attrib_list = {
-    'exposure': cv2.CAP_PROP_EXPOSURE,
+    "exposure": cv2.CAP_PROP_EXPOSURE,
-    'contrast': cv2.CAP_PROP_CONTRAST
+    "contrast": cv2.CAP_PROP_CONTRAST
 }
 
+
 def check_settings():
     VIDEO_CHECK = cv2.VideoCapture(0)
 
-    if not os.path.exists('camera_settings.log'):
+    if not os.path.exists("camera_settings.log"):
-        f = open('camera_settings.log', 'w')
+        f = open("camera_settings.log", "w")
         for attrib, index in attrib_list.items():
-            f.writelines(f'{attrib} = {VIDEO_CHECK.get(index)}\n')
+            f.writelines(f"{attrib} = {VIDEO_CHECK.get(index)}\n")
         f.close()
 
     else:
-        f = open('camera_settings.log', 'r')
+        f = open("camera_settings.log", "r")
-        lines = f.read().split('\n')
+        lines = f.read().split("\n")
         for line in lines:
-            attrib = line.split(' = ')
+            attrib = line.split(" = ")
             if attrib[0] in attrib_list.keys():
                 VIDEO_CHECK.set(attrib_list[attrib[0]], eval(attrib[1]))
         f.close()
 
-    print('*'*28)
+    print("*"*28)
-    print('* Checking camera settings *')
+    print("* Checking camera settings *")
-    print('*'*28)
+    print("*"*28)
     for attrib, index in attrib_list.items():
-        print(f'{attrib} = {VIDEO_CHECK.get(index)}')
+        print(f"{attrib} = {VIDEO_CHECK.get(index)}")
 
     VIDEO_CHECK.release()
 
+
 def reset_settings():
-    if not os.path.exists('camera_settings.log'):
+    if not os.path.exists("camera_settings.log"):
-        print('"camera_settings.log" does not exist!')
+        print("'camera_settings.log' does not exist!")
-        print('Verify your camera settings!')
+        print("Verify your camera settings!")
         return False
     else:
         VIDEO_CHECK = cv2.VideoCapture(0)
-        f = open('camera_settings.log', 'r')
+        f = open("camera_settings.log", "r")
-        lines = f.read().split('\n')
+        lines = f.read().split("\n")
         for line in lines:
-            attrib = line.split(' = ')
+            attrib = line.split(" = ")
             if attrib[0] in attrib_list.keys():
                 VIDEO_CHECK.set(attrib_list[attrib[0]], eval(attrib[1]))
         f.close()

object_detection.py

@@ -1,3 +1,4 @@
+import os
 import time
 import cv2
 import numpy as np
@@ -5,27 +6,34 @@ import numpy as np
 
 class ObjectDetection:
     def __init__(self):
+        PROJECT_PATH = os.path.abspath(os.getcwd())
+        MODELS_PATH = os.path.join(PROJECT_PATH, "models")
+
         self.MODEL = cv2.dnn.readNet(
-            'models/yolov3.weights',
+            os.path.join(MODELS_PATH, "yolov3.weights"),
-            'models/yolov3.cfg'
+            os.path.join(MODELS_PATH, "yolov3.cfg")
         )
 
         self.CLASSES = []
-        with open("models/coco.names", "r") as f:
+        with open(os.path.join(MODELS_PATH, "coco.names"), "r") as f:
             self.CLASSES = [line.strip() for line in f.readlines()]
 
-        self.OUTPUT_LAYERS = [self.MODEL.getLayerNames()[i[0] - 1] for i in self.MODEL.getUnconnectedOutLayers()]
+        self.OUTPUT_LAYERS = [
+            self.MODEL.getLayerNames()[i - 1] for i in self.MODEL.getUnconnectedOutLayers()
+        ]
         self.COLORS = np.random.uniform(0, 255, size=(len(self.CLASSES), 3))
         self.COLORS /= (np.sum(self.COLORS**2, axis=1)**0.5/255)[np.newaxis].T
 
     def detectObj(self, snap):
         height, width, channels = snap.shape
-        blob = cv2.dnn.blobFromImage(snap, 1/255, (416, 416), swapRB=True, crop=False)
+        blob = cv2.dnn.blobFromImage(
+            snap, 1/255, (416, 416), swapRB=True, crop=False
+        )
 
         self.MODEL.setInput(blob)
         outs = self.MODEL.forward(self.OUTPUT_LAYERS)
 
-        # Showing informations on the screen
+        # ! Showing informations on the screen
         class_ids = []
         confidences = []
         boxes = []
@@ -35,13 +43,13 @@ class ObjectDetection:
                 class_id = np.argmax(scores)
                 confidence = scores[class_id]
                 if confidence > 0.5:
-                    # Object detected
+                    # * Object detected
                     center_x = int(detection[0]*width)
                     center_y = int(detection[1]*height)
                     w = int(detection[2]*width)
                     h = int(detection[3]*height)
 
-                    # Rectangle coordinates
+                    # * Rectangle coordinates
                     x = int(center_x - w/2)
                     y = int(center_y - h/2)
 
@@ -133,16 +141,17 @@ class VideoStreaming(object):
                         int(self.VIDEO.get(cv2.CAP_PROP_FRAME_HEIGHT)),
                         int(self.VIDEO.get(cv2.CAP_PROP_FRAME_WIDTH))
                     ), np.uint8)
-                    label = 'camera disabled'
+                    label = "camera disabled"
                     H, W = snap.shape
                     font = cv2.FONT_HERSHEY_PLAIN
-                    color = (255,255,255)
+                    color = (255, 255, 255)
-                    cv2.putText(snap, label, (W//2 - 100, H//2), font, 2, color, 2)
+                    cv2.putText(snap, label, (W//2 - 100, H//2),
+                                font, 2, color, 2)
 
-                frame = cv2.imencode('.jpg', snap)[1].tobytes()
+                frame = cv2.imencode(".jpg", snap)[1].tobytes()
                 yield (b'--frame\r\n'b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
                 time.sleep(0.01)
 
             else:
                 break
-        print('off')
+        print("off")

static/script.js

@@ -1,6 +1,6 @@
-// Functions to deal with button events
+// ! Functions that deal with button events
 $(function () {
-  // Preview switch
+  // * Preview switch
   $("a#cam-preview").bind("click", function () {
     $.getJSON("/request_preview_switch", function (data) {
       // do nothing
@@ -10,7 +10,7 @@ $(function () {
 });
 
 $(function () {
-  // Flip horizontal switch
+  // * Flip horizontal switch
   $("a#flip-horizontal").bind("click", function () {
     $.getJSON("/request_flipH_switch", function (data) {
       // do nothing
@@ -20,7 +20,7 @@ $(function () {
 });
 
 $(function () {
-  // Model switch
+  // * Model switch
   $("a#use-model").bind("click", function () {
     $.getJSON("/request_model_switch", function (data) {
       // do nothing
@@ -30,7 +30,7 @@ $(function () {
 });
 
 $(function () {
-  // exposure down
+  // * exposure down
   $("a#exposure-down").bind("click", function () {
     $.getJSON("/request_exposure_down", function (data) {
       // do nothing
@@ -40,7 +40,7 @@ $(function () {
 });
 
 $(function () {
-  // exposure up
+  // * exposure up
   $("a#exposure-up").bind("click", function () {
     $.getJSON("/request_exposure_up", function (data) {
       // do nothing
@@ -50,7 +50,7 @@ $(function () {
 });
 
 $(function () {
-  // contrast down
+  // * contrast down
   $("a#contrast-down").bind("click", function () {
     $.getJSON("/request_contrast_down", function (data) {
       // do nothing
@@ -60,7 +60,7 @@ $(function () {
 });
 
 $(function () {
-  // contrast up
+  // * contrast up
   $("a#contrast-up").bind("click", function () {
     $.getJSON("/request_contrast_up", function (data) {
       // do nothing
@@ -70,7 +70,7 @@ $(function () {
 });
 
 $(function () {
-  // reset camera
+  // * reset camera
   $("a#reset-cam").bind("click", function () {
     $.getJSON("/reset_camera", function (data) {
       // do nothing