cameras.py

import tensorflow as tf
import cv2
import objectdetection

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
    detector = objectdetection.ObjectDetection('ssd_mobilenet_v2_coco_2018_03_29/frozen_inference_graph.pb', session=sess)
    cam = cv2.VideoCapture(0)
    while True:
        ret_val, img = cam.read()
        print(detector.predict([img]))