v/Lego-Boost-Python-Library
1
0
Fork 0
mirror of https://github.com/undera/pylgbst.git synced 2020-11-18 19:37:26 -08:00
Code Issues Releases Wiki Activity

Record video

Browse Source
This commit is contained in:
Andrey Pohilko 2018-08-01 13:55:59 +03:00
parent a18808374b
commit 3322ad8e4c
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -1,4 +1,5 @@
.idea .idea
*.iml *.iml
*.pyc *.pyc
build build
*.avi

17
examples/tracker/__init__.py
View File

@ -1,7 +1,6 @@
import json import json
import logging import logging
import os import os
import time
import traceback import traceback
from threading import Thread from threading import Thread
@ -28,14 +27,22 @@ class FaceTracker(MoveHub):
# cap.set(cv2.CAP_PROP_XI_AUTO_WB, 1) # cap.set(cv2.CAP_PROP_XI_AUTO_WB, 1)
# cap.set(cv2.CAP_PROP_XI_AEAG, 1) # cap.set(cv2.CAP_PROP_XI_AEAG, 1)
size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))
fourcc = cv2.VideoWriter_fourcc(*'XVID')
fps = cap.get(cv2.CAP_PROP_FPS)
video = cv2.VideoWriter('output.avi', fourcc, fps, size)
try: try:
while True: while True:
flag, img = cap.read() flag, img = cap.read()
self.cur_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) self.cur_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
logging.debug("Got frame") logging.debug("Got frame")
video.write(self.cur_img)
finally: finally:
logging.info("Releasing cam...") logging.info("Releasing cam...")
cap.release() cap.release()
video.release()
face_cascade = cv2.CascadeClassifier('/usr/share/opencv/haarcascades/' + 'haarcascade_frontalface_default.xml') face_cascade = cv2.CascadeClassifier('/usr/share/opencv/haarcascades/' + 'haarcascade_frontalface_default.xml')
@ -89,7 +96,7 @@ class FaceTracker(MoveHub):
vert = 0 vert = 0
self.motor_external.constant(horz) self.motor_external.constant(horz)
self.motor_AB.constant(-vert) self.motor_A.constant(-vert)
def main(self): def main(self):
thr = Thread(target=self.capture) thr = Thread(target=self.capture)
@ -106,8 +113,8 @@ class FaceTracker(MoveHub):
cur_face = (0, 0, 0, 0) cur_face = (0, 0, 0, 0)
while thr.isAlive(): while thr.isAlive():
bodies, weights = self._find_color() # bodies, weights = self._find_color()
#bodies, weights = self._find_faces() bodies, weights = self._find_faces()
if len(bodies): if len(bodies):
logging.debug("Bodies: %s / Weights: %s", bodies, weights) logging.debug("Bodies: %s / Weights: %s", bodies, weights)
@ -149,5 +156,3 @@ if __name__ == '__main__':
pass pass
# conn.disconnect() # conn.disconnect()
# obj = A()
# obj.main()