DEV Community

SEMIU AMAO
SEMIU AMAO

Posted on

Security of environment with Image Processing on a Live camera using Python

Scene Manager
The concept of securing the environment based on the frames captured from a live can be simplified as making the camera to take the pictures of changes that happen in an environment that are different from the initial state that the camera knows.

Before delving into the codes, the following are the basic steps (pseudocode) involved in this application :

  1. Connect to a live camera
  2. Capture the initial state of the environment and store it
  3. Initiate the application to start monitoring the scene inside each frame that the camera captures and compare the content with the initial state.
  4. If there is significant differences, then the application labels that frame and save that for user's reference in the future.

Application Template

File structure for the application :

Download the source code

Scene_manager folder
|
|-app.py
|-/templates
|-index.html
|-views.html
|-/static

app.py contains the codes that implement all the steps mentioned above.

The Codes in detail

from flask import Flask, redirect, request, render_template, Response
import cv2
from threading import Thread
import datetime
import numpy as np
import os
import time
Enter fullscreen mode Exit fullscreen mode

Importing the required python modules. It should be noted that not all the modules are pre-installed with python interpreter during installation. The flask module, cv2 (opencv-python) module and numpy module are not pre-installed with python but they can be installed via the Package Installer for Python (pip) as follows:

pip install flask
pip install numpy
pip install opencv-python
Enter fullscreen mode Exit fullscreen mode

Open the application in browser

app = Flask(__name__)

frm = []
status = False
@app.route('/')
def home():
    return render_template("index.html")
Enter fullscreen mode Exit fullscreen mode

Initialise the Flask() object in order to use for rendering web pages. Also initialise the global variables frm and status to be used for storing image and state of the camera.
@app.route('/') defines the response that the web browser will give when url to the home page of the application is entered in the browser. In this case, the url returns a template file "index.html" that we had saved inside the templates folder.

Connecting to the Camera


@app.route('/connect', methods=['POST', 'GET'])
def conn():
    pp = request.form['cam']
    print(pp)

    return Response(loadCam(pp), mimetype='multipart/x-mixed-replace; boundary=frame')


def loadCam(cam):
    global frm
    if cam == '0': cam = 0
    cap=cv2.VideoCapture(cam)
    while True:
        st, frm = cap.read()
        web_frm = cv2.imencode('.jpg',frm)[1].tobytes()
        if status:
        # do comparison
            tr = Thread(target= comp, args=(frm,))
            tr.start()
        if cv2.waitKey(2) == 27 : break
        yield(b'--frame\r\n'b'Content-Type: image/jpeg\r\n\r\n' + web_frm +b'\r\n')

Enter fullscreen mode Exit fullscreen mode

When the user hit the connect button on the template page, the pp variable gets the camera's address that was supplied and use this as an argument to the loadcam() function. Using the capability of opencv (https://docs.opencv.org/4.x/d6/d00/tutorial_py_root.html), the application is able to render the live content of the camera to the webpage. Take note of the condition if status: which if true, the application will implement comp() function that does the comparison of the initial scene with the subsequent frames.

Taking the picture of initial state of in the camera's view

@app.route('/cap')
def snap():
    cv2.imwrite('static/snap.jpg', frm)
    return Response('New snap taken at: '+ str(datetime.datetime.now()))
Enter fullscreen mode Exit fullscreen mode

When the user click on the "Take Screen Snap" button on the page, the code saves the current frame as a picture (snap.jpg) inside the static folder

Start Monitoring for change in the scene

@app.route('/start')
def st():
    global status
    status = True
    return Response('Started Monitoring at: ' + str(datetime.datetime.now()))

def comp(img):
    try:
        img_org = cv2.imread('static/snap.jpg')
        # mse = summation(img_org - img)/(img_org.shape[0] * img_org.shape[1])
        mse =  np.sum((img_org.astype("float") - img.astype('float')) ** 2)
        mse /= (img_org.shape[0] * img_org.shape[1])
        if int(mse) > 300:
            cv2.imwrite('static/' + str(time.time()) +'.jpg', img)
        print (mse)
Enter fullscreen mode Exit fullscreen mode

This code changes the status value to true. This makes the loadcam() function to call comp() function in parallel thread. The comp() saves the labeled frames with the current time as the file name (in .jpg) format.

Stop Monitoring for change in the scene

@app.route('/stop')
def stp():
    global status
    status = False
    return Response('Stopped Monitoring at: ' + str(datetime.datetime.now()))
Enter fullscreen mode Exit fullscreen mode

This code sets the value of status to false and the loadcam() stops calling the comp() function.

Show the Captured changes in the camera scenes

@app.route('/shows')
def carry():
    fs = os.listdir('static/')
    fls = [f for f in fs if f !='snap.jpg' and f !='.DS_Store']
    return render_template('views.html', pics=fls)
Enter fullscreen mode Exit fullscreen mode

This code takes the names of the saved pictures from the static folder (excluding the initial state) and return it as a list to the view.html file which shows all the pictures. The view.html is also displayed via <iframe> html tag inside the home page.

Delete the captured changes

@app.route('/delete')
def rmv():
    fs = os.listdir('static/')
    for f in fs:
        if f != 'snap.jpg':
            os.remove('static/'+ f)
    return redirect('/shows')
Enter fullscreen mode Exit fullscreen mode

Code deletes all the captured images inside the static folder (excluding the snap.jpg). The snap.jpg will always be overwritten anytime the user clicks on the "Take Screen Snap" button on the page.

Launch the python code as a webpage

if __name__=='__main__':
    app.run()
Enter fullscreen mode Exit fullscreen mode

app.run() calls the run method of the flask module which launched the application as a web server.

Download the Source Code here

Top comments (1)

Collapse
 
yusufomo99 profile image
Yusuf Aweda

Deeply detailed. Thanks