Fix OpenCV stream to be rgb instead of bgr.

Author: datitran

object_detection_app.py

@@ -76,7 +76,8 @@ def worker(input_q, output_q):
  while True:
  fps.update()
  frame = input_q.get()
- output_q.put(detect_objects(frame, sess, detection_graph))
+ frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+ output_q.put(detect_objects(frame_rgb, sess, detection_graph))
 
  fps.stop()
  sess.close()
@@ -114,7 +115,8 @@ def worker(input_q, output_q):
 
  t = time.time()
 
- cv2.imshow('Video', output_q.get())
+ output_rgb = cv2.cvtColor(output_q.get(), cv2.COLOR_RGB2BGR)
+ cv2.imshow('Video', output_rgb)
  fps.update()
 
  print('[INFO] elapsed time: {:.2f}'.format(time.time() - t))

object_detection_multilayer.py

@@ -74,7 +74,8 @@ def worker(input_q, output_q):
  while True:
  fps.update()
  frame = input_q.get()
- output_q.put(detect_objects(frame, sess, detection_graph))
+ frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+ output_q.put(detect_objects(frame_rgb, sess, detection_graph))
 
  fps.stop()
  sess.close()