basic setup

- now allows flag for stream source - setup environments to work for me - got linux environment working - setup infrastructure for rtmp server stream object.

Author: samcrane8

environment-linux.yaml

@@ -7,7 +7,7 @@ dependencies:
 - freetype=2.5.5=2
 - icu=54.1=0
 - jbig=2.1=0
-- jlaura::opencv3=3.0.0=py35_0
+- menpo::opencv3=3.1.0=py35_0
 - jpeg=9b=0
 - libpng=1.6.27=0
 - libtiff=4.0.6=3

environment.yml

@@ -7,7 +7,7 @@ dependencies:
 - freetype=2.5.5=2
 - icu=54.1=0
 - jbig=2.1=0
-- jlaura::opencv3=3.0.0=py35_0
+- menpo::opencv3=3.1.0=py35_0
 - jpeg=9b=0
 - libpng=1.6.27=0
 - libtiff=4.0.6=3

object_detection_app.py

@@ -85,6 +85,7 @@ def worker(input_q, output_q):
 
 if __name__ == '__main__':
  parser = argparse.ArgumentParser()
+ parser.add_argument('-str', '--stream', dest="is_stream", action="store_true")
  parser.add_argument('-src', '--source', dest='video_source', type=int,
  default=0, help='Device index of the camera.')
  parser.add_argument('-wd', '--width', dest='width', type=int,
@@ -104,6 +105,12 @@ def worker(input_q, output_q):
  output_q = Queue(maxsize=args.queue_size)
  pool = Pool(args.num_workers, worker, (input_q, output_q))
 
+
+ if (args.is_stream):
+ print('Reading from rtmp stream.')
+ else:
+ print('Reading from webcam.')
+
  video_capture = WebcamVideoStream(src=args.video_source,
  width=args.width,
  height=args.height).start()

utils/app_utils.py

@@ -41,6 +41,42 @@ def fps(self):
  return self._numFrames / self.elapsed()
 
 
+class RTMPVideoStream:
+ def __init__(self, src, width, height):
+ # initialize the video camera stream and read the first frame
+ # from the stream
+ self.stream = cv2.VideoCapture(src)
+ self.stream.set(cv2.CAP_PROP_FRAME_WIDTH, width)
+ self.stream.set(cv2.CAP_PROP_FRAME_HEIGHT, height)
+ (self.grabbed, self.frame) = self.stream.read()
+
+ # initialize the variable used to indicate if the thread should
+ # be stopped
+ self.stopped = False
+
+ def start(self):
+ # start the thread to read frames from the video stream
+ Thread(target=self.update, args=()).start()
+ return self
+
+ def update(self):
+ # keep looping infinitely until the thread is stopped
+ while True:
+ # if the thread indicator variable is set, stop the thread
+ if self.stopped:
+ return
+
+ # otherwise, read the next frame from the stream
+ (self.grabbed, self.frame) = self.stream.read()
+
+ def read(self):
+ # return the frame most recently read
+ return self.frame
+
+ def stop(self):
+ # indicate that the thread should be stopped
+ self.stopped = True
+
 class WebcamVideoStream:
  def __init__(self, src, width, height):
  # initialize the video camera stream and read the first frame