mirror of
https://gitlab.com/Shinobi-Systems/ShinobiCE.git
synced 2025-03-09 15:40:15 +00:00
109 lines
3.5 KiB
JavaScript
109 lines
3.5 KiB
JavaScript
;
|
|
var argparse = require('argparse');
|
|
var datetime = require('datetime');
|
|
var imutils = require('imutils');
|
|
var math = require('math');
|
|
var cv2 = require('opencv4nodejs');
|
|
var np = require('numpy');
|
|
|
|
width = 800;
|
|
|
|
textIn = 0;
|
|
textOut = 0;
|
|
|
|
function testIntersectionIn(x, y) {
|
|
|
|
res = -450 * x + 400 * y + 157500;
|
|
if((res >= -550) && (res < 550))) {
|
|
console.log (str(res));
|
|
return true;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
function testIntersectionOut(x, y) {
|
|
res = -450 * x + 400 * y + 180000;
|
|
if (res >= -550) && (res <= 550))) {
|
|
console.log (str(res));
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
camera = cv2.VideoCapture('test2.mp4');
|
|
|
|
firstFrame = null;
|
|
|
|
// loop over the frames of the video
|
|
while (true) {
|
|
// grab the current frame and initialize the occupied/unoccupied
|
|
// text
|
|
(grabbed, frame) = camera.read();
|
|
text = 'Unoccupied';
|
|
}
|
|
// if the frame could not be grabbed, then we have reached the end
|
|
// of the video
|
|
if (!grabbed) {
|
|
break;
|
|
}
|
|
// resize the frame, convert it to grayscale, and blur it
|
|
frame = imutils.resize(frame, width=width);
|
|
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY);
|
|
gray = cv2.GaussianBlur(gray, (21, 21), 0);
|
|
|
|
// if the first frame is None, initialize it
|
|
if (firstFrame === null) {
|
|
firstFrame = gray;
|
|
continue;
|
|
}
|
|
// compute the absolute difference between the current frame and
|
|
// first frame
|
|
frameDelta = cv2.absdiff(firstFrame, gray);
|
|
thresh = cv2.threshold(frameDelta, 25, 255, cv2.THRESH_BINARY)[1];
|
|
// dilate the thresholded image to fill in holes, then find contours
|
|
// on thresholded image
|
|
thresh = cv2.dilate(thresh, null, iterations=2);
|
|
_, cnts, _ = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE);
|
|
// loop over the contours
|
|
for (c in cnts) {
|
|
// if the contour is too small, ignore it
|
|
if (cv2.contourArea(c) < 12000) {
|
|
continue;
|
|
// compute the bounding box for the contour, draw it on the frame,
|
|
// and update the text
|
|
(x, y, w, h) = cv2.boundingRect(c);
|
|
cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2);
|
|
}
|
|
cv2.line(frame, (width / 2, 0), (width, 450), (250, 0, 1), 2) #blue line;
|
|
cv2.line(frame, (width / 2 - 50, 0), (width - 50, 450), (0, 0, 255), 2)#red line;
|
|
}
|
|
|
|
rectagleCenterPont = ((x + x + w) /2, (y + y + h) /2);
|
|
cv2.circle(frame, rectagleCenterPont, 1, (0, 0, 255), 5);
|
|
|
|
if(testIntersectionIn((x + x + w) / 2, (y + y + h) / 2))) {
|
|
textIn += 1;
|
|
}
|
|
if(testIntersectionOut((x + x + w) / 2, (y + y + h) / 2))) {
|
|
textOut += 1;
|
|
}
|
|
// draw the text and timestamp on the frame
|
|
|
|
// show the frame and record if the user presses a key
|
|
// cv2.imshow("Thresh", thresh)
|
|
// cv2.imshow("Frame Delta", frameDelta)
|
|
|
|
if (cv2.waitKey(1) & 0xFF == ord('q')) {
|
|
break;
|
|
}
|
|
cv2.putText(frame, 'In: {}'.format(str(textIn)), (10, 50),
|
|
cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2);
|
|
cv2.putText(frame, 'Out: {}'.format(str(textOut)), (10, 70),
|
|
cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2);
|
|
cv2.putText(frame, datetime.datetime.now().strftime('%A %d %B %Y %I:%M:%S%p'),
|
|
(10, frame.shape[0] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.35, (0, 0, 255), 1);
|
|
cv2.imshow('Security Feed', frame);
|
|
|
|
|
|
// cleanup the camera and close any open windows
|
|
camera.release();
|
|
cv2.destroyAllWindows();
|