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

save video with rectangle

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

15
examples/tracker/__init__.py
View File

@ -17,6 +17,7 @@ class FaceTracker(MoveHub):
def __init__(self, connection=None): def __init__(self, connection=None):
super(FaceTracker, self).__init__(connection) super(FaceTracker, self).__init__(connection)
self.cur_img = None self.cur_img = None
self.cur_face = None
def capture(self): def capture(self):
logging.info("Starting cam...") logging.info("Starting cam...")
@ -36,9 +37,12 @@ class FaceTracker(MoveHub):
try: try:
while True: while True:
flag, img = cap.read() flag, img = cap.read()
if self.cur_face is not None:
(x, y, w, h) = self.cur_face
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 0, 255,), 2)
video.write(img)
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()
@ -111,7 +115,6 @@ class FaceTracker(MoveHub):
pyplot.ion() pyplot.ion()
pyplot.show() pyplot.show()
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()
@ -126,15 +129,13 @@ class FaceTracker(MoveHub):
for n in range(0, len(bodies)): for n in range(0, len(bodies)):
items.append((bodies[n], weights[n])) items.append((bodies[n], weights[n]))
self.cur_face = None
for item in sorted(items, key=lambda i: i[1], reverse=True): for item in sorted(items, key=lambda i: i[1], reverse=True):
cur_face = item[0] self.cur_face = item[0]
self._auto_pan(cur_face) self._auto_pan(self.cur_face)
break break
(x, y, w, h) = cur_face
cv2.rectangle(self.cur_img, (x, y), (x + w, y + h), (255, 0, 0), 2)
plt.set_array(self.cur_img) plt.set_array(self.cur_img)
logging.debug("Updated frame") logging.debug("Updated frame")
pyplot.pause(0.02) pyplot.pause(0.02)