Skip to content
This repository has been archived by the owner on Nov 26, 2019. It is now read-only.

Commit

Permalink
proper structure created (#51)
Browse files Browse the repository at this point in the history
  • Loading branch information
@johri-lab @NIteshx2
johri-lab authored and NIteshx2 committed Dec 26, 2018
1 parent c463993 commit 315297e
Show file tree
Hide file tree
Showing 2 changed files with 140 additions and 112 deletions.
244 changes: 136 additions & 108 deletions PythonCode/UltimateSecurityCam.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@

#if you get error while importing the google how to install <Package Name> in python 3.6

#initial values set
THRESHOLD = 40
camera = cv2.VideoCapture(0)

Expand All @@ -22,121 +23,148 @@
size = (int(camera.get(cv2.CAP_PROP_FRAME_WIDTH)),
int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT)))


dir_path = os.path.dirname(os.path.realpath(__file__))
#video file name
videofile = "basic_motion_detection.avi"

videoWriter = cv2.VideoWriter(os.path.join(str(dir_path),videofile),
cv2.VideoWriter_fourcc('D', 'I', 'V', 'X'),
fps, size)

initial = int(time.time())
final = initial + 4

while (final-initial):
#start timer on the frames
ret, frame = camera.read()
initailiztion_text = "Starting in " + str(final-initial) + "..."
cv2.putText(frame,initailiztion_text,(60,30),cv2.FONT_HERSHEY_TRIPLEX,1,(0,100,255),2)
cv2.imshow("Ultimate Security Camera",frame)


if cv2.waitKey(int(45)) &0xff == ord('q'):
break

elif int(time.time()) == (initial + 1):
initial = initial + 1
#print(str(final-initial) + "...")

maxcnts=0
start = time.time()
while (True):
ret, frame = camera.read()
# The first frame as the background
if background is None:
background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
background = cv2.GaussianBlur(background, (21,21), 0)
continue

gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
gray_frame = cv2.GaussianBlur(gray_frame, (21,21), 0)

# Compare the difference between each frame of image and the background
#print(background.shape, gray_frame.shape)
diff = cv2.absdiff(background, gray_frame)
diff = cv2.threshold(diff, THRESHOLD, 255, cv2.THRESH_BINARY)[1]
diff = cv2.dilate(diff, es, iterations=2)
# Calculate the outline of the target in the image
image, cnts, hierarchy = cv2.findContours(diff.copy(),
cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
detection_text = ("Detecting " + str(len(cnts)) + " Moving Objects")
detection_text_colour = (0,255,0) #set as green
if len(cnts) > 0:
#if breach detected
detection_text_colour = (0,0,255) #set to red
cameraSound.play()

for c in cnts:
if cv2.contourArea(c) < (background.shape[0]*background.shape[1])/204:
#minimum area to be calculated based on image size and camera megapixels
continue
# Calculate the bounding box
(x, y, w, h) = cv2.boundingRect(c)
cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,0), 2)

#maximum object detected
if len(cnts)>maxcnts: maxcnts=len(cnts)

#print(detection_text)
cv2.putText(frame,detection_text,(60,30),cv2.FONT_HERSHEY_DUPLEX,1,detection_text_colour,2)

diff = cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR) #3 channel gray scaled image

horizontal_stack = np.hstack((frame, diff)) #proper way to stack 3D arrays
merged_windows = np.concatenate((frame, diff), axis=1)

cv2.imshow("Ultimate Security Camera", merged_windows)

#cv2.imshow("contours", frame)
videoWriter.write(frame)
#cv2.imshow("dif", diff)
#cv2.imwrite('didff.jpg', diff)

keypress = cv2.waitKey(45)
if keypress:
if keypress &0xff == ord('q'):
break
elif keypress &0xff == ord('r'):
#reset the camera
background = None

cv2.destroyAllWindows()
camera.release()

end = time.time()

confirm = input("Do you wish to save the current configs? [Y/N]: ")
#if run with python2 use raw_input

data={"Date and Time":time.asctime(time.localtime(time.time())),
"Camera FPS":fps,
"Threshold":THRESHOLD,
"Max Objects":maxcnts,
"Video File":"basic_motion_detection.avi",
"Path":dir_path,
"Duration": '%0.2f' %(end-start)}

configfile = "config.txt"

if confirm.startswith('y' or 'Y'):
print("\nUpdating config file...")
with open(configfile,'w') as jfile:
json.dump(data, jfile, indent = 4)
print("Data updated to " + configfile + " successfully!")

class UltimateSecurityCam:
""" UltimateSecurityCam class identifies object movements and
detection of any kind of undesirable movement in the
surroundings
"""

def __init__(self):
pass

def initial_window(self):
#initial window starts
initial = int(time.time())
final = initial + 4
while (final-initial):
#start timer on the frames
ret, frame = camera.read()
initailiztion_text = ("Starting in " + str(final-initial) + "...")
cv2.putText(frame,initailiztion_text,(60,30),
cv2.FONT_HERSHEY_TRIPLEX,1,(0,100,255),2)
cv2.imshow("Ultimate Security Camera",frame)

if cv2.waitKey(int(45)) &0xff == ord('q'):
break

elif int(time.time()) == (initial + 1):
initial = initial + 1
#print(str(final-initial) + "...")

def usc(self):
#main window opens and opject movement detection starts
maxcnts = 0
global background
start = time.time()
while (True):
ret, frame = camera.read()
# The first frame as the background
if background is None:
background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
background = cv2.GaussianBlur(background, (21,21), 0)
continue

gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
gray_frame = cv2.GaussianBlur(gray_frame, (21,21), 0)

# Compare the difference between each frame of image and the background
#print(background.shape, gray_frame.shape)
diff = cv2.absdiff(background, gray_frame)
diff = cv2.threshold(diff, THRESHOLD, 255, cv2.THRESH_BINARY)[1]
diff = cv2.dilate(diff, es, iterations=2)
# Calculate the outline of the target in the image
image, cnts, hierarchy = cv2.findContours(diff.copy(),
cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
detection_text = ("Detecting " + str(len(cnts)) + " Moving Objects")
detection_text_colour = (0,255,0) #set as green
if len(cnts) > 0:
#if breach detected
detection_text_colour = (0,0,255) #set to red
cameraSound.play()

for c in cnts:
if cv2.contourArea(c) < (background.shape[0]*background.shape[1])/204:
#minimum area to be calculated based on image size and camera megapixels
continue
# Calculate the bounding box
(x, y, w, h) = cv2.boundingRect(c)
cv2.rectangle(frame, (x,y), (x+w, y+h), (0,255,0), 2)

#maximum object detected
if len(cnts)>maxcnts: maxcnts=len(cnts)

#print(detection_text)
cv2.putText(frame,detection_text,(60,30),cv2.FONT_HERSHEY_DUPLEX,1,detection_text_colour,2)

diff = cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR) #3 channel gray scaled image

horizontal_stack = np.hstack((frame, diff)) #proper way to stack 3D arrays
merged_windows = np.concatenate((frame, diff), axis=1)

cv2.imshow("Ultimate Security Camera", merged_windows)

#cv2.imshow("contours", frame)
videoWriter.write(frame)
#cv2.imshow("dif", diff)
#cv2.imwrite('didff.jpg', diff)

keypress = cv2.waitKey(45)
if keypress:
if keypress &0xff == ord('q'):
break
elif keypress &0xff == ord('r'):
#reset the camera
background = None

cv2.destroyAllWindows()
camera.release()

end = time.time()
duration = end-start

data={"Date and Time":time.asctime(time.localtime(time.time())),
"Camera FPS":fps,
"Threshold":THRESHOLD,
"Max Objects recorded":maxcnts,
"Video File":videofile,
"Path":dir_path,
"Duration": '%0.2f' %(duration) + ' seconds'}
return data

def config(self,data):
#saves all necessary configurations

elif confirm.startswith('n' or 'N'):
pass
confirm = input("Do you wish to save the current configs? [Y/N]: ")
#if run with python2 use raw_input

configfile = "config.txt"

if confirm.startswith('y' or 'Y'):
print("\nUpdating config file...")
with open(configfile,'w') as jfile:
json.dump(data, jfile, indent = 4)
print("Data updated to " + configfile + " successfully!")

elif confirm.startswith('n' or 'N'):
pass

else:
print("Invalid input!")


else:
print("Invalid input!")

#Sequence of main program execution
start = UltimateSecurityCam()
start.initial_window()
data = start.usc()
start.config(data)
print("Program succesfully terminated!")
8 changes: 4 additions & 4 deletions PythonCode/config.txt
View file
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
{
"Date and Time": "Sun Dec 23 03:47:01 2018",
"Date and Time": "Wed Dec 26 05:17:21 2018",
"Camera FPS": 2,
"Threshold": 40,
"Max Objects": 17,
"Max Objects recorded": 3,
"Video File": "basic_motion_detection.avi",
"Path": "/home/johri/github/UltimateSecurityCam/latest 23-12/UltimateSecurityCam/PythonCode",
"Duration": "7.88"
"Path": "/home/johri/github/UltimateSecurityCam/latest 23-12/26-12-18/PythonCode",
"Duration": "13.76 seconds"
}

0 comments on commit 315297e

Please sign in to comment.