From 7bc3ca8a0d25b55626fc0c42c68ac706868024a2 Mon Sep 17 00:00:00 2001
From: Schuman Zhang <sz@proadviser.com.au>
Date: Mon, 9 Jul 2018 22:07:53 +1000
Subject: [PATCH] added video writerr

---
 .gitignore                                    |   4 +-
 analytics/__pycache__/tracking.cpython-36.pyc | Bin 2382 -> 2674 bytes
 analytics/tracking.py                         |   2 +-
 detect_object.py                              |  42 ++++++++++++++
 objection_detection_app.py                    |  53 +++---------------
 .../__pycache__/visualization.cpython-36.pyc  | Bin 14242 -> 0 bytes
 utils/__pycache__/webcam.cpython-36.pyc       | Bin 6840 -> 6835 bytes
 utils/webcam.py                               |   1 +
 video_writer.py                               |  19 +++++++
 9 files changed, 73 insertions(+), 48 deletions(-)
 create mode 100644 detect_object.py
 delete mode 100644 utils/__pycache__/visualization.cpython-36.pyc
 create mode 100644 video_writer.py

diff --git a/.gitignore b/.gitignore
index fda7dbf..b45d290 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
 tf_models/
-*.csv
\ No newline at end of file
+*.csv
+*.mp4
+__pycache__/
\ No newline at end of file
diff --git a/analytics/__pycache__/tracking.cpython-36.pyc b/analytics/__pycache__/tracking.cpython-36.pyc
index 4947a22757099a7c3d644d5a0e661c9c2e914149..ac28ea20a263e9986d80c1733c01ab173da4690a 100644
GIT binary patch
delta 1106
zcmZWpJ8u&~5Z>K;`251eHi=E%yp*dT1Q7}#AcxSPAOT8*DNc@eo%lHW%-$s-;v^TQ
zjYM<}NK{bKA@M(`(YBOy)F3f)juK1kYQK5y&g{(W-S^_}a(yYE&(t2*UX-$g{3Iho
z2mT(c;1&QUoO+}~>y#qTJXU83)#_S8^}7C$aE<G43D<=&(CX&VlQ(@0&f-9d4r{Q2
zM*wXS5eJ+C(vXFktGh--BQ{_h7HG{d<ysMOO?0}=xN(TuG-%RpW<c;7AZIywN^Em|
zP0@X2v}B_rRECRCG{d&v4bIwBWm;|F@h(^bJ9qS_x}|i#C%TIHfzo@8a7XE`-v+tW
zQ(80FQ<>FvGu)KzP)KF&4{0``JEFlssrRJVvo)m$!fVMKWTVopwkI46@`cFpc%JT-
zGngtG2{mDk!_35==m*OJm2Jp!Tw{ybaS&tz6LFtaXeoZr-Yr)_TQ_?`27dS1n$6@C
zZcF1qO_q_LMwmeu_26h#Iu##i%Y6nW97s<JUjYFMNL&h~CNAUjBO4?gg(9+bh-1JG
zX#zl;F(Aqs7Qo+^$+cM$UFDKWJKf#RmXM=2z#|1Xk`ILw`i>MmUxt_BzuF3|e7mDp
zC_Nu<7%R0&UKpe(d9}7lLI#~@d@5pYh8L=YkF^+|25oj=P_(AOT6A-8;XE%LQa%%D
z{2ZT+e;C_W=7xU2R*WdGRtcsPF-T{AeOM#RD1L5kK;W<Dqas#zdnp;eR^$aVrc*a>
zTmwcTK;DbUpTPf$lAOgIb8(QmN+;vbsq2f_8KryeAWX`n^q1lwP#TExB09B(WlJ^v
zZYW-d(uk>bd1j<@c;0fHX2bIwX91*OAzrm?dM5U)`+e-&aZu#YrF?*J>S6eVP=}uy
z-Y38ZXi+N-)3WqLZ>LnY>3iPs=OOWR$O6I=0{jAqGW{(V?8*LN0Q)q|1n#hIcYMC<
RiMte}0N+K9LLh1x{{Wo3={*1d

delta 835
zcmZWnyKWOf6rIODW*s{wjuGTRobZxXLgHZ_1rd%w3Jth`2&9d~YS~!_o3+=>tbqti
z6rq6%G!>sfO@*jvDfobq5CtV2U%=cA3YHmZ&YXGNGxyA$-}P^`JIzX^)J(2!fAtXh
zi6)w5_=f57oB@tFPEj8R7y~CM2}m*0fZjx$a_bf1R$}eZz<w9J*uo|T8jQsB8>Y)c
z2D)gBLL+7nb95K^q`m1AU5?X8$}rAGgIsDai{sHS8pOLk)|O1t&Icr1^#hN}CQBqC
zHH01Yh`eXOeOoxnqvub$W+Dn;hUxjykVknEPKwW}2lVZF$!rJCFcC5vY?%Z^1tM(&
zJ{1n|S%50QlxST=RvjNvE$hndNt?h6n~+gq4Y!L}AMYzp=~SPIuE{OWLXiwJkv9~v
zTe$S`x_uPm1-0$8))skn&&BA;=6#fto;@afX-s-0UR^+Zu0!~;`s^&Z$HtT&<}2#E
zv$NRPj~Em4va9XFgtj{(>L;q_-orEMi+igA=69M!`_)nncKPC}SrGun{TlEC_<x}$
zppIfry=G@|S$$#`&cM5AEA7f$fO%~_PIjfHX6m!TyLRJjkSEV`F|U@rQ!7(X!NM?(
z(liWDm}q%gg`Th8cpERF*ne*WRs*;M@Grh#P;5aB0FhviVP-RrF=iF(zN2U2EKQR*
u?`8uj;7e*6%clZv)K2!WXSRx?6WYfQ#5G&*XMB_<*PzN0K1dnE9e)5lNTD<U

diff --git a/analytics/tracking.py b/analytics/tracking.py
index 8a0c926..60eef4f 100644
--- a/analytics/tracking.py
+++ b/analytics/tracking.py
@@ -34,7 +34,7 @@ def update_person_status(self, class_names):
 
 
     def write_to_report(self, frame_number):
-        self.writer.writerow({'frame': frame_number, 'detections': self.class_names})
+        self.writer.writerow({'frame': frame_number, 'detections': self.class_counts})
 
 
     def __call__(self, context):
diff --git a/detect_object.py b/detect_object.py
new file mode 100644
index 0000000..902a503
--- /dev/null
+++ b/detect_object.py
@@ -0,0 +1,42 @@
+import os
+import numpy as np
+from utils import label_map_util
+from utils.webcam import draw_boxes_and_labels
+
+
+CWD_PATH = os.getcwd()
+PATH_TO_LABELS = os.path.join(CWD_PATH, 'detection', 'data', 'mscoco_label_map.pbtxt')
+
+NUM_CLASSES = 90
+# label map
+label_map = label_map_util.load_labelmap(PATH_TO_LABELS)
+categories = label_map_util.convert_label_map_to_categories(label_map, max_num_classes=NUM_CLASSES,
+                                                            use_display_name=True)
+category_index = label_map_util.create_category_index(categories)
+
+# pass in image_np, returns 
+def detect_objects(image_np, sess, detection_graph):
+
+    image_np_expanded = np.expand_dims(image_np, axis=0)
+    image_tensor = detection_graph.get_tensor_by_name('image_tensor:0')
+
+    boxes = detection_graph.get_tensor_by_name('detection_boxes:0')
+
+    scores = detection_graph.get_tensor_by_name('detection_scores:0')
+    classes = detection_graph.get_tensor_by_name('detection_classes:0')
+    num_detections = detection_graph.get_tensor_by_name('num_detections:0')
+
+    # Do the detection/model prediction here
+    (boxes, scores, classes, num_detections) = sess.run(
+        [boxes, scores, classes, num_detections],
+        feed_dict={image_tensor: image_np_expanded})
+    
+    rect_points, class_names, class_colors = draw_boxes_and_labels(
+        boxes=np.squeeze(boxes),
+        classes=np.squeeze(classes).astype(np.int32),
+        scores=np.squeeze(scores),
+        category_index=category_index,
+        min_score_thresh=.5
+    )
+
+    return dict(rect_points=rect_points, class_names=class_names, class_colors=class_colors)
\ No newline at end of file
diff --git a/objection_detection_app.py b/objection_detection_app.py
index 491b602..c267182 100644
--- a/objection_detection_app.py
+++ b/objection_detection_app.py
@@ -5,11 +5,12 @@
 import numpy as np
 import tensorflow as tf
 
-from utils.webcam import FPS, WebcamVideoStream, draw_boxes_and_labels
+from utils.webcam import FPS, WebcamVideoStream
 from queue import Queue
 from threading import Thread
-from utils import label_map_util
 from analytics.tracking import ObjectTracker
+from video_writer import VideoWriter
+from detect_object import detect_objects
 
 CWD_PATH = os.getcwd()
 
@@ -17,49 +18,6 @@
 PATH_TO_MODEL = os.path.join(CWD_PATH, 'detection', 'tf_models', MODEL_NAME, 'frozen_inference_graph.pb')
 PATH_TO_VIDEO = os.path.join(CWD_PATH, 'input.mp4')
 
-print('cv2', cv2.__version__)
-
-PATH_TO_LABELS = os.path.join(CWD_PATH, 'detection', 'data', 'mscoco_label_map.pbtxt')
-
-NUM_CLASSES = 90
-
-# label map
-label_map = label_map_util.load_labelmap(PATH_TO_LABELS)
-categories = label_map_util.convert_label_map_to_categories(label_map, max_num_classes=NUM_CLASSES,
-                                                            use_display_name=True)
-category_index = label_map_util.create_category_index(categories)
-
-# pass in image_np, returns 
-def detect_objects(image_np, sess, detection_graph):
-
-    image_np_expanded = np.expand_dims(image_np, axis=0)
-    # print('image_np_expanded')
-    # print(image_np)
-    image_tensor = detection_graph.get_tensor_by_name('image_tensor:0')
-
-    boxes = detection_graph.get_tensor_by_name('detection_boxes:0')
-
-    scores = detection_graph.get_tensor_by_name('detection_scores:0')
-    classes = detection_graph.get_tensor_by_name('detection_classes:0')
-    num_detections = detection_graph.get_tensor_by_name('num_detections:0')
-
-    # Do the detection/model prediction here
-    (boxes, scores, classes, num_detections) = sess.run(
-        [boxes, scores, classes, num_detections],
-        feed_dict={image_tensor: image_np_expanded})
-    
-    # print('num_detections:', num_detections)
-    # Visualization of the results of a detection.
-    # print('classes:', classes)
-    rect_points, class_names, class_colors = draw_boxes_and_labels(
-        boxes=np.squeeze(boxes),
-        classes=np.squeeze(classes).astype(np.int32),
-        scores=np.squeeze(scores),
-        category_index=category_index,
-        min_score_thresh=.5
-    )
-
-    return dict(rect_points=rect_points, class_names=class_names, class_colors=class_colors)
 
 def worker(input_q, output_q):
     # load the frozen tensorflow model into memory
@@ -104,6 +62,7 @@ def worker(input_q, output_q):
     video_capture = WebcamVideoStream(src=args.video_source,
                                       width=args.width,
                                       height=args.height).start()
+    writer = VideoWriter('output.mp4', (args.width, args.height))
 
     '''
     stream = cv2.VideoCapture(0)
@@ -131,8 +90,9 @@ def worker(input_q, output_q):
         else:
             data = output_q.get()
             context = {'frame': frame, 'class_names': data['class_names'], 'rec_points': data['rect_points'], 'class_colors': data['class_colors'],
-                        'width': args.width, 'height': args.height, 'frame_numer': fps.get_numFrames()}
+                        'width': args.width, 'height': args.height, 'frame_number': fps.get_numFrames()}
             new_frame = object_tracker(context)
+            writer(new_frame)
             cv2.imshow('Video', new_frame)
 
         print('[INFO] elapsed time: {:.2f}'.format(time.time() - t))
@@ -145,4 +105,5 @@ def worker(input_q, output_q):
     print('[INFO] approx. FPS: {:.2f}'.format(fps.fps()))
 
     video_capture.stop()
+    writer.close()
     cv2.destroyAllWindows()
diff --git a/utils/__pycache__/visualization.cpython-36.pyc b/utils/__pycache__/visualization.cpython-36.pyc
deleted file mode 100644
index c5ce605b447aa20db4a746edb3ff23ef059358cf..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 14242
zcmeHOU2q)7ao)ZEAK>sq@Yj^Jl5C5RDUkY!EyJ`VkRU1AAOI7vD6PaC7uS2Z1@?B2
zvwH*%-g6Q|j_o+ImH&SyapfU?%S+ypyyqoPE>~r%Qst^t>|&nskcX60iSzZ$-rfNs
zsmO6957D7^cBXfJdS-gMd%7FXjgMD;^5onb|8!Z?{!ttGeF^!O@dV$~HH|b=Bb_p?
zZtC)#F|+cW+sd1HnJa7+&7#bew#sH%<|<oNvnq3A=9qksn>EUs6L=;mM|mnxkxEpi
z3RP*0#;HaVG)Yr5O-E>kj?ytYPABLjouUuXX?lb{OplU5kJ01w1bu`)N*|*q=?u-%
zQ}l5<OHb1roul*g3_VNF(ew0k^a=VTy+EI$dHOV6pwG~Y^b);H3v`h#(IQ=@D|D4!
zp=<P6`W!9Mby}tsTBXm^4O*jh`U1U5H|aHcolJUz-lSXf^HitXWYGpSh{&cUZBmOI
z+M;c8sZAdF)S){hD4>wKbeDE$mwL2EU!;io^d<TQ`bGMs%`cf#^kv$AOEahGE&2-H
zN9dR7t4)o**38k@=~v#;&6yF%H|SRZIjSJdEWJ(NMEO`+{uaH1@^Sh${W|>yawq7I
z={M=OkUL3#Lcc@5i`*&tQ~E>tBji3rnKf<pJA407G4NY30y{MPrqS$rjnMJEzzADb
zXjsBFx`9nb(-+2FC+J$PvuAOIIoWQ#We4_1BflBio+0dp?cB8u%QKv|wP~L<I<`O$
zZ6om8_Q8G(-@{P0s+e{<(k1zZ=ZEMs><Z7YjJuZGMURdbgtkT8wr8|`a+<wiKZX+q
zw%eSm++%89zE?qNmKR*7VP8aRvvR3riBQ;>nweX;)3wYBr)^6(o^9qA_PPSEi?*|A
zn}v%`a0hK_7hTJ1v}{^%+rCF;g?oC{@m)JK%NM)C?cMZ!GRv1NvO_t_*n$^2XnM2d
zgf>33{D$vZ$mTEk!g9?TUS8AncWfc+m$)1_?zUOHB%F5OduHxZ&+^RDq9wLvZk%&h
zeV1(S1_qNKt_r(nmifcyTGqA`S2(*OEDy*h`9{kD!VF)t7<+R}=4}gYZO@dwu6M<q
zuJ2$)WW{7=rA6EBtUBJdS>@c?c26zfqEDL%+T|;burCUyfjU=Qo|&3(^@=a-AXGyX
zuXG!&z_HBoRSR3U;S1laTy6P5s4C}pQuvrot)MWCv+KM7ulSzbTeNq~;wyeA(J5c^
z$g#W|HktWrcYV<_^K$xTY00{4dt?i9e2@dmLCY*JSv#J>W^BpEDqLzg%_bJ3yyR@Q
z)Rt9bR&9?0%VnRD;1s#nZ5O+iw7V)SEWfJ2JE6Kz<0!CNK3OAK{&7_ptpAuQCTm+$
zC8iYF_kl77NXnsYyK*n8vVMY7Zpmqb);Lut<d;Cwo>{z(bLfSZIeFbCr`u+7zpluA
zOhHzJq>`kx+7%raM6bTB4KSTZ%8Z;Gd@2ESaB?=)h)JCxJtY7|QM_U0>y8&*61LUm
zY!LQt_#nH|b-&TD0>?9}%hp|M%U6_9Tybg1YT(qYxL_(~S-uv9wP6+#Dle-ycMXu7
z#`vn`+UbUwWm#0j&e<Vyvf>c6%Bt0|dKQN6ppk7gTC3e=)6A{fqKlLdLT=UVw#~|_
zzXKwZi%>{bD=R05$@DG?e+Pth!|z$iCaqaSE{+Mq^J<|gYnDez)l#~|oLg&w!py>&
zV|$*(Z}%?fxi~m|<+XJERaViUgre)7nOpNfpVhUIT`8{(2}TY-AeeO<tf;u&!aiWb
z3+sN{3VpLYT%kh33i3BwwiU*BF*~cs*IL`(Mx&&!s(Q=r8yRN3_~r00|Gk7Kco|8k
zZRufVD;sJ1`o6xEBc%OInBRi>5JAD%&wOjTpN%qGr6^0;{fw^N)F}6P?S}Rm-Vj>(
zy}w$+84ptVVw@3n2fPLZ4pC*{gW+}CogTy&4(F^9`o`+=RV7>&#AYxrWf#jmDW2vJ
z);D&Xuw?`-t798)v>*do;aR{aY@IcpyQO;JwthG4bi;bb3S09Ar##M1%W1S!x9Gy}
zx@2tF#*T182u9U5&fd6sG1hnDERb5BEyww5EF|M{Q()~G<2d7W;(Ql}<b}A{@I7pP
z7-zy}oZoCZt{vxAI<^<*0xZI8F3!tI#MKde;<AL*JB}M?n+|<f6UVUfKi)j|g}@fU
zxuDVNwk>b31tD>6(GIplzjLk)Vh69NyS61foaTB*_*-@(49@u*{H{Z;;RV@W$I*4`
zA$ZKWZs@qdxr3TPrxzb*-0BJmIi<Q4)H~khStf!460LAdpVY+(<Qh;e<&P<aA4w_m
z$Q-1Uz8*r=+}HY<NZ%?%8CE9W26cc^ihxRr+1F6cf=04?-?{8Hd}1mYB?jmX*d>eU
z!3aWTT0cv&xFMC45&Gg`Y>5XAh+<?=QNko=^|-*gkrmFCMHv9Wr$tmb8AB3h17|lb
zt;u%RR>Ty;N}F~l6`?raaKSuhb7Gns%<#v8q6D*o1+S|~%I#Olajx=+Lh3#{#0@@)
zDglqJ6({wQUc*z;#iQIXD{Q*Qo3!3wT=~c7rU!fxg@fCc9Po0OkIIoAWuk19i}Fz+
zDn_OKvaW#@6e3QGQ5g&Y8!9!^to3)`APYpifg64o(%#!dW|vRdK(RE=jGptK<xcVe
z@3kFoUZ$)WAHc!!pp;_IpbdlLG<;tGJ;=nsI1^Y<aW`xz2(v?8u`9dc^l(?hI=!~F
z%fnf_53aK->u~zuI#4owF%P8h;>99i?Hs^<&RDdYR@V&;CxBi}Lli=AuWio`f_cLQ
zD}X|sl!r|u)p5h;ht0?dI<D1Y(y6<UBiuTW*9k>^xJ&>IRS5@>cQNI_ywIb+Gc0qt
zKG-H0=nfVPw}>z>ILk<ltSZ#^`fwZSBik@<Tx%NZqH7yxNX>QjtO11|SDKMpDH6QX
zXARz-vvLDCk0pjA5jI3E-(`T&@S0*6D=3H^Xv?bCw5O5Yr6Ae}e776gYOMrwzUA4{
z{f@a9$G7+tZ&rOkYRK%?!EP0UFEsp4FZAQQl4=!s;^Q3S9_JW$&Re^jN1B)WAD7cD
zj3*CoQ+(>d3n)3sVXh%b>HgT&B~n)~hk$v3cH$V=Mous2;%SspUcpjm0*_?zuYvJ1
zuV6)>Uyih`%$u2hrC*IoVikO%GNXZ0aH>aDq*+dzxi@qDvHp0!)}IJre~8AS@u(I}
zG;_>3yo+d;L;dV%y{NpEL%muvgLDGvc#<jz(oN<^d5Dv&uJtBa6`?Gl$NuC^%{!gb
z!s2#C{EZIb0)Ca>DnZ;#zOH$h*R-hCpW6LqRD>NPPh(%np41k#TTy>Hn%pW!)B8GR
zbfiBM!U{sgXl90IdKCDK0h6g{=7dJ2=;(}gLX)^1Ij%)hz$OD+%2KqIx5iQPD8spS
zbX3-yrb;wP)wgpHc7PB1ctPt=-T5Arr91@Sxb<lEJ^801%gcP?)Jf&jd6-Q2uqQC^
zI-Eue#)6I|!0r^6ur_=?AT3*-f&uYLUd@NywHn(-Xzzj{a9bD$LZ}@s%m$9@qDQH2
z$+P7_pE915g!`mI?E%`_>b0f0YpgUGN;Mg>q6hQrVVw_{@nApyKQm(%I(r{_FbABU
z&BS@B_QyG<)V(sU%a%JAhRu7dwl3dey>;{Ve~*9Ped!+GcV=~Q4zD;biGDVtN;w!A
z!h1i#Z?+bfCEeosx)bMA8bF7L^L$kc;~Y!4q|Ozzic2AcSlH{>aq-&9Wg&cFj=8=C
z6QmMraS@sz_aGI<I0wU<8<*2F7U%fj%^p+Q;YWd+_!uWoa&m@~Sx%nf<l~&Mf*$8s
z3dJR-U03tRZKAF&LO{@N;tnYIRupG(ZHNmSei-^~@jP0ZRUQr!f+2Brgp<S-`9=1Z
z)Jl3R;Lfw@c!y+-cpTl?c!D`3+Np|OfjAk{Pv|q58GTOA;Pz0{YndEMc<WE$>olMl
z)>OG|BhT&dW0}j36glUSNkvXR1fw0Or2BcW_ACSir}-$KsG>SpX~z0{Y%XtBa6pfO
zw#T}O<U>E5B6^s6LEhWYy)5HdoY1xnW1#O@?t%LYG|@Mf&l=C)nm5keUu_vtkeJ5T
zVO%|+mB6G6wmj6E*tUz?2NbSw2i8F2jq_)XXGZS1|M&XjA*J$=ayew|Jl(@T>s`7i
zrfWzgb5v42yc1qSD37qnvL4v60=|?SLPMI!tyB>_j0cxT2V|)#P3%r+wD~k)lVG27
zyodJChSu)DFlV?Ht{8#B`i2EPbmY?cDqF}Uub4L+zPNy>csMqrTM6|j(})~vdA2ZA
zNaNFd_7CWIc980z>Q?#$4w3i>5;WvnC)gbiF=Qq2(UFrcH%}(_)hQo;*mz2-`~!>;
z@KsrxVKxAras~PHfb*C1{`pN5U>?x=IKn}VaC}3Z4>M)9v%$xpS3yIg+;&#Xg&6|;
z%lfVvW&tmpgl-1i?X&$tQ|lL_oVWl@63~7z1XBQ;C{Ur9IjM5lG?z(pMVaIKZE?S(
zYj-X{6Jz#Kvi|nH*oWH3_pR)Fz%Jw|yDzZ~@1Tdy>BCR@{g3qr>_T4e6uS`5;*5!B
zIC%`o-sx*nZ*O;T<;SHR0*dvKgwF_8lI~@Z%;w{~G^)l~7uL(lkV`?GU}u#S+qfJ{
zL)LZfeyE!)qi9woz1LMIx}y{pzEKVhTVe^tzr+(TT$?FCEyzJ>C_y>kTuFaimv!VM
zQWkPRN4S87_cdYVVZu77FAPmr1AT#=s#)u=F>9qSY}>t#56w=h3jaH1t8~bR*|lLe
ztEUyu8qdC$yyyE_T3=q-h9)71mIGEKVZp-$HD*xXP=H~B)Q0rz@Gw3^5oaNGe_KvE
zg6t7OIY_1Ff75Wo*Vgt=W2reV<<<a!5fS+==~zxHNrH|GYB{8;eP5CI$OF1o((wCC
z-BKFXjwxyQIv^=2vm|74<|7yUIg+8et1K*E*S{{~INpLJTV#6#+d497i6!6dF#A=#
zsUtcg^HwP;a4oI}s0JwPB!CtZXmR%<<Rh%P`x!{j=V9+C!IqPU3Jv!SEEZ)>VckFp
zRt%JmqXY{F{;0tAmOC!wWKqh=-+uV9`6;Djy8IuUhxayQKHxm0J{o8;(nqt#<i*~G
zO)b)HjrDbS?zEHIuVwDAhfY`Gfq5`)(V4mPvA%uig7@Rf1s2Q=zukH9$J`nBT+sGD
z{s7{iyWsi_%MD&cX3*|^W5vQq;(@EV$@w=1fDsY$CSd=LC&B1lgB;;+qf`GHPSQ*=
z_2Dd&A!2<O%W)|khutrNhxiZfMos#EJVXyeu;=B_EdFyi1h%0Fw-B2bq)~#E{!efV
z_VtJtki_}0i<?E9-E%r|5&mbV1HY28GK-sBON!&yILB53aS@4GO;1;SceH>wh`|?p
z80LF}<(KHnWsxiLDmwdTJW6iYSkLEceNI1yzZpnxI19PXkDBuL!<QO4l=#PPsPumS
zLlnT~p!9D+;TJP(B4#C@uLOXx;(rFP48u^)<0*i0v)eEl<65BY<BRw#YRIcWScDH6
zkbW*KZ&jq3np;%yj78{|L)YHUpfn!Vq8!R44rJSe&w54!yF?p&%R~j~=boSndMXZ}
zRE@H1+Gdk5yx<cwhOZ@jorGfy%yU26=dq?>kI=WK!B8h?oNG!4Sq*1w!uqq6UAC=Y
zTUu8WbHqb9rSvKvuqceUyU#f7&BW4!2&<9&bF-;n<*}HPB5(;tgAxCm)cmB^`~ls}
zO1&ZND3RF%v%`6K^~iW8{d`}u!~0P~EVt9L=HZ{Qya48EcszS5cr3MaBty=nPNI3^
z{M;wiR3CP!UcoZMsVmsbW{?$H23zQ)0YF)UU}4~8O77=-r*BK}ZFvKSek^NyojJJ|
zdym}ay6^fZ*Y~++H<`vY?O$F>GElgmNE+C#4I2?u%VD4a3_HqvBlTOjmUGsw6DV%H
zif-Z@Vh<KnS3mGyj`gL)Cv_yw6QecsI}2v<aWO$7&RSkCuDJfD11?<mMO=}F%uDQL
zGz&07ip`A?GeXtg?O;FaFk=OA8PQt-OaR-kVj@g~X@o7Ea5+e)SDA%?A0VnkJb^Kl
zX`!6Bw|H%o-jtUc3J>(=MqEZ@gK~?RRe0DOg7c3NnPB)Rpu_;#2MePie~eE^Im2I4
z>xR;=U<cHKQ>d&}$8cZZ=M)}rq|8&`MIQ!N5?y?~a*t)ekGP_y{otjB^p_)h5)bE@
zw*}uuXX-A2(>*40S>(nYtzS#w6HzHR4-SP$7?>MM@94X5L4^gVObLv+5|pcd>rWyg
zCbL}%Zo|kh2~$LQJ11TV5jDdlc1xk}DRyLSm&IG)B3!2yO<mAnykOY3qlst=HOo-g
z@NHk`XbJ97`~_v<7DLTDuSAvLU!k6H+uT-VU)wH-@52-#A#6MWj?><0%J-+KuvLww
z!!fwFBtDXt!Qth7gi7F&2jP1E2s^zn`gn9ij$C<17iXwE7#BQLjuB3TwSD~<7IN_A
zrnb8tPJ$PL17aPnK*^ly&qPyj1s?4m>mTo*2-zieG&&X?k4`98V0Z*Zn*<WgL{&95
z9Cg*V5p?j5erFYGC*Rh_tg8?oD~Pdy3tQRM;FMBEGj>=FEWDCUDpM&(83j2199#mP
zC2etr2fu%E;5nEojbJr4h2QP4<(SQI(&Q&uN6<kO>A<Zlg9cc^;qV&C0=eS`JFIz-
z)L)Z$=~5GB8j>Rsfg6q|!J{A;N_N8*%QU*yFqVBd?IaHH3y<I^185zIUlNL|420lw
zPs6p}P@OSO<fGPTK(5zAR}T%Nd4D2j5NS1(IH{TIzQRUX!-rzSg>_aL(H~I#R$5o>
z;o)lm8!%Mx*YYg`oO~JFYtG<+0#*Mr3_vtN%T_zZHmHNE2@JX*hQUH397OHC3~K1O
z@Q#3hIJBBc!y~wO!d8lus>@5wH4V#kljev<P`*@2iJosX)KnQqg_#ojj1LIO-b7Gr
z-9aQQqDq*8AgdA+Ge^=4YRC`lW)QsnD*B>@C|uM@QA+E|-i+H0-Iny0!kPAs+nnB>
z<!e75px$r@G(+TdayS@04&FvEwJkZbba*^quUHF#n|DMf;gTegTW2xAa0g&{;(bYR
z9%!Er?;XP)G=n_6(}x);c3L(_LL{4o!=++aVEeHJf)iC_Vv62~`(@NQgRR8b8tx+^
zIXDOe(lhkyEV1UHSnN9OZksi)4GafrO{r@ThL@15qEl{_4x3Wb$PkWKm*i6qTP&P0
zg5${%Bosu<P*8+&aPVl!)Y9Q-*#D-7kZz|@2Sc&vQ5WIy;GlyboO$C4u5k9OBt+rk
zjsZ#>EF9L$W}T4Z3P#)@lFFa9l8!mlA2vp{OzhmkLQBvNT(m<rshINy!U-HS6hpE|
z&dhN7d?Uil)3_(zFW4eSP=5X(y4*m8gYw!lbCv6_dX(nQW$_$jf1Km<u$Nsvx3E|I
zVt?NFV*jJ}2p{$;a$zMvWEqg>J$ei=d_#k0>WrP8iL20UTorj4z^SSnzkHsB&*Nkv
z6vXF|6YH{w?Gv{-R|ugQ1z}u(PZkmJaSmr95UiiXS#~xTtPbKxDV|tcUszsTxUpEj
zw6e5vV@=#Z6~QVH;z&Wgc-n#Z6azV!XZGaKM2qlL#f0L5VnK0jz!VhgF>43Uy1Gg7
z{n4Cy7jc`D!2JL*n&X3wmlPG(4sMru3~Mis#z+KO+v=E;30!K0;4&TaWCA&ej{G!}
zd`ejcL&X<({#ksAD{_u?-cE6iS3z#m{haBCbj;cGgd~R;T)nP5cZZZtgOyI+K5b5U
zy4l-Z(NE**%MqBYgif;(n#rL&jkEwiI7epIkgFg!lbu1W=?r_oNB;_X>6rKi`kl=R
z121_mSU*_Es4oq!Ra{=aBJOg<9VAzzg@5*RT&&m0Z`A8z570O-xwX71AV_`?izBsv
zvzC_S)j({cfjOQ67eVRL@Gt`7KnL+@9#hgBA_cs_7r1!rh#I$^uCHHPU%Fghx_o85
zetqG_)oaV<R03OHSxrD<pL^iQ3>m2^NmaTOq_<ER{n=6~%?MJ1m+LY#*o<5bN*BOW
zx~}~q2N4APZ*c57ke2|y2yh;IkjvDNACnM%b7}C$b@@vTDgWVHu2e0}sJ$6GoxIQU
Xd+dZvALa0}%Ct<6S03Z{;P1Zx$88iP

diff --git a/utils/__pycache__/webcam.cpython-36.pyc b/utils/__pycache__/webcam.cpython-36.pyc
index 10b13d00f8a8c3a89b04e46e17edb3c89bd484ef..e6ade014cb23e8a2afadc49201a4343d1793873e 100644
GIT binary patch
delta 506
zcmYk3OG{f(5XaBl+_!m&NlZjV6C=G9yJ$#JeAUDUZi15NB3Wq5xr>O;bG0oJq1CJf
znT0Erf?uFy)9;}mWFd$fzk;}NPC)St^E>mOIgjC-|Jg=1l?epwvxR&m+RNAjdp9(S
zzmj_gr>XpxdB#<))fv~0m9j-j=LRW*l+8_U)mgcX+q{itKX2zD9<D<<@b9VwVB-_i
zAdEkhU5MhNvIz-X@f|`B-uSA}kCW<mNaJ7iH&~6N_6Oi2Ug_sBiaFx~rtsd_h6P+Q
zM=BBOhC_|5vKm{5b>)}=<E+RWrBHHwB1L7<O=udz-EhBot^>l9N$U-^j6#G0xu+7C
zqQgN97mWzsnRDPY($=*OC435v!Z2pS&mB8*sXT{)3K94KVyrRUp3|w%dgoF4GZn-z
zq3PRjx5QX0@~Xd@R&~!6<eQ!t!J+s7lyNaWK|G1yhNtDOoJ3w?2KN&V{KU(|a{n47
z9Ia3g<FYbAm~(<6BSleS7NcFCVnQ}A5)$(A*1cj4x4H(C^pnhlSM!5k3HAkt%x=?9
H57x35DqUxW

delta 494
zcmYk1&r4fD5XX03-jB`8OY+pxBwDN`7JVv$irOYB{%EbG5Q;x8uLm(q4i6K9`)cu^
zU~Qo1GEgYk;=drdc=QkGt%uT_3cVM4E``q87P`ZHcE0<aU1r<Um(#_mblP3*DLtF%
z7vh`vZ&_R~a>l1`iYp5!5bDX$eI(FCt0mBSWrQhKHae^vRvt=BwnUgl7gM}vunV*3
zw;;@3oNWSl^aGBdoA!)-$kA7$0ew_S9K&t;n24ZAGiDWr=$&~2sdlgR5@3wZ?DtTn
zlJfzU=!?^YC-hj}yXo>&2uEUDpx6+BQP~VY73s<-*ec83Q|{`g?{W=B+q20}Hf&MK
zzt7J-^ndp3>n}-$026M&FsM>{qiflQfZpa#?@q9x#u%MMy|$-H*Hs)Hc9PNan&NE4
zYL@2ngV3RBex7-r|Lia7TT2?t8Y}d=5Wpc_6t?c{@IhczDypoVImT+>t4S@^H6GAV
z@Azow#=9>ftTo4@X6XE_m8lyF8hWDZVbl)&GILWMIoD|7LA|yB?8E~M3ID<a4`d=U
NkQtG;q$l7{{01pVZPfq(

diff --git a/utils/webcam.py b/utils/webcam.py
index 8fb43bc..9856af6 100644
--- a/utils/webcam.py
+++ b/utils/webcam.py
@@ -78,6 +78,7 @@ def read(self):
     def stop(self):
         # stopping the thread
         self.stopped = True
+        self.stream.release()
 
 
 def standard_colors():
diff --git a/video_writer.py b/video_writer.py
new file mode 100644
index 0000000..e75bacb
--- /dev/null
+++ b/video_writer.py
@@ -0,0 +1,19 @@
+import cv2
+
+class VideoWriter(object):
+
+    def __init__(self, path, size):
+        self.path = path
+        self.size = size
+        self.writer = cv2.VideoWriter(self.path, 
+                         cv2.VideoWriter_fourcc('M','J','P','G'), 
+                         20.0, self.size, True)
+
+    def __call__(self, frame):
+        self.writer.write(frame)
+
+    def close():
+        self.writer.release()
+
+
+