Browse Source

Add img/vid to gitignore; minor changes in video/image presence

master
Lennart Heimbs 4 years ago
parent
commit
cc9304365a
3 changed files with 10 additions and 5 deletions
  1. 3
    0
      .gitignore
  2. 4
    3
      camera/image_presence.py
  3. 3
    2
      camera/video_presence.py

+ 3
- 0
.gitignore View File

camera/venv camera/venv
camera/images camera/images
camera/videos camera/videos
*.jpg
*.h264
*.mp4

+ 4
- 3
camera/image_presence.py View File

print("[INFO] {}: {} original boxes, {} after suppression".format(filename, len(rects), len(pick))) print("[INFO] {}: {} original boxes, {} after suppression".format(filename, len(rects), len(pick)))


# show the output images # show the output images
#cv2.imshow("Before NMS", orig)
#cv2.imshow("After NMS", image)
#cv2.waitKey(0)
if len(pick):
#cv2.imshow("Before NMS", orig)
cv2.imshow("After NMS", image)
cv2.waitKey(0)

+ 3
- 2
camera/video_presence.py View File

# otherwise, we are reading from a video file # otherwise, we are reading from a video file
else: else:
vs = cv2.VideoCapture(args["video"]) vs = cv2.VideoCapture(args["video"])
#vs.set(cv2.CAP_PROP_FPS, 2)


"""" Analyzing video frames """ """" Analyzing video frames """
# loop over the frames of the video, and store corresponding information from each frame # loop over the frames of the video, and store corresponding information from each frame
# dilate the thresholded image to fill in holes, then find contours on thresholded image # dilate the thresholded image to fill in holes, then find contours on thresholded image
thresh = cv2.dilate(thresh, None, iterations=2) thresh = cv2.dilate(thresh, None, iterations=2)
thresh = np.uint8(thresh) thresh = np.uint8(thresh)
cnts, im2 = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
_, cnts, im2 = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
#cnts = cnts if imutils.is_cv2() else im2 #cnts = cnts if imutils.is_cv2() else im2
#print(len(cnts)) #print(len(cnts))
#if len(cnts) > 1: #if len(cnts) > 1:


trackbox = frame[y:y+h, x:x+w] trackbox = frame[y:y+h, x:x+w]
trackbox = cv2.resize(trackbox, (224, 224)) trackbox = cv2.resize(trackbox, (224, 224))
cv2.imshow('image',trackbox)
#cv2.imshow('image',trackbox)
blob = cv2.dnn.blobFromImage(cv2.resize(trackbox, (300, 300)),0.007843, (300, 300), 127.5) blob = cv2.dnn.blobFromImage(cv2.resize(trackbox, (300, 300)),0.007843, (300, 300), 127.5)
net.setInput(blob) net.setInput(blob)
detections = net.forward() detections = net.forward()

Loading…
Cancel
Save