ROC integration
Change-Id: Ica1d8b10e9c0318d89361ff85206aabfa5f58e7f
diff --git a/person_detection/person_detection.py b/person_detection/person_detection.py
index 47a3fcf..5d01794 100644
--- a/person_detection/person_detection.py
+++ b/person_detection/person_detection.py
@@ -17,11 +17,9 @@
from base_camera import BaseCamera
-import action
-
-Shape = namedtuple('Shape', ['n','c','h','w'])
class Camera(BaseCamera):
+ Shape = namedtuple('Shape', ['n','c','h','w'])
def __init__(self, device, args):
log.basicConfig(format="[ %(levelname)s ] %(message)s", level=log.INFO, stream=sys.stdout)
@@ -32,7 +30,7 @@
self.is_async_mode = True
self.device = device
- super(Camera, self).__init__(device, args.idle)
+ super(Camera, self).__init__(device, args.key, args.mbrlow, args.mbrhigh, args.devicegroup)
def __del__(self):
# stream.release()
@@ -70,7 +68,7 @@
log.info("Loading IR to the plugin...")
exec_net = IECore().load_network(network=net, device_name="CPU", num_requests=2)
# Read and pre-process input image
- shape = Shape(*net.inputs[input_blob].shape)
+ shape = Camera.Shape(*net.inputs[input_blob].shape)
del net
@@ -142,7 +140,7 @@
cv2.FONT_HERSHEY_COMPLEX, 0.6, black, 1)
if obj_count > 0:
- action.person_detected(self.device, obj_count)
+ self.person_detected(obj_count)
yield cv2.imencode('.jpg', frame)[1].tobytes()