kicap/object-detection-flask-opencv
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update project with minor fixes and improvements

Browse Source
This commit is contained in:
Diego Inácio
2022-06-24 14:53:57 -03:00
parent 7cb96779e4
commit 478e3d8837
5 changed files with 110 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
application.py
View File

@ -13,72 +13,79 @@ check_settings()
VIDEO = VideoStreaming()
@application.route('/')
@application.route("/")
def home():
TITLE = 'Object detection'
return render_template('index.html', TITLE=TITLE)
TITLE = "Object detection"
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
@application.route('/request_preview_switch')
# * Button requests
@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)

39
camera_settings.py
View File

@ -1,48 +1,51 @@
import os
import cv2
attrib_list = {
'exposure': cv2.CAP_PROP_EXPOSURE,
'contrast': cv2.CAP_PROP_CONTRAST
"exposure": cv2.CAP_PROP_EXPOSURE,
"contrast": cv2.CAP_PROP_CONTRAST
}
def check_settings():
VIDEO_CHECK = cv2.VideoCapture(0)
if not os.path.exists('camera_settings.log'):
f = open('camera_settings.log', 'w')
if not os.path.exists("camera_settings.log"):
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')
lines = f.read().split('\n')
f = open("camera_settings.log", "r")
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('* Checking camera settings *')
print('*'*28)
print("*"*28)
print("* Checking camera settings *")
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'):
print('"camera_settings.log" does not exist!')
print('Verify your camera settings!')
if not os.path.exists("camera_settings.log"):
print("'camera_settings.log' does not exist!")
print("Verify your camera settings!")
return False
else:
VIDEO_CHECK = cv2.VideoCapture(0)
f = open('camera_settings.log', 'r')
lines = f.read().split('\n')
f = open("camera_settings.log", "r")
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()

35
object_detection.py
View File

@ -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',
'models/yolov3.cfg'
os.path.join(MODELS_PATH, "yolov3.weights"),
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)
cv2.putText(snap, label, (W//2 - 100, H//2), font, 2, color, 2)
color = (255, 255, 255)
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")

18
static/script.js
View File

@ -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