X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;f=example-apps%2FROBO%2Fretail_app%2Fcamera_driver%2Fcapture_frame.py;h=eb66edcad96310b8d269fc954b2608663bf9d0a3;hb=02e02f4e44c603151068279471504c0053ee76f9;hp=ed2ae699c41fe50292b422b87ca239746c7d2993;hpb=293b5e72268f5111b0bdd92deec739fda71ae323;p=ealt-edge.git

diff --git a/example-apps/ROBO/retail_app/camera_driver/capture_frame.py b/example-apps/ROBO/retail_app/camera_driver/capture_frame.py
index ed2ae69..eb66edc 100644
--- a/example-apps/ROBO/retail_app/camera_driver/capture_frame.py
+++ b/example-apps/ROBO/retail_app/camera_driver/capture_frame.py
@@ -14,38 +14,40 @@
 # limitations under the License.
 #
 
+import cv2
+
 
 class VideoCamera(object):
     """
     opneCV to capture frame from a camera
     """
-    """
     def __init__(self, url):
-        # self.video = cv2.VideoCapture(url)
+        self.video = cv2.VideoCapture(url)
 
     def delete(self):
-        # self.video.release()
+        self.video.release()
 
     def get_frame(self):
-        # get a frane from camera url
-        # success, image = self.video.read()
-        return
-    """
+        """
+        get a frame from camera url
+        """
+        success, image = self.video.read()
+        return success, image
 
 
 class VideoFile(object):
     """
     opneCV to capture frame from a video stream
     """
-    """
     def __init__(self, video_name):
-        # self.video = cv2.VideoCapture("./test/resources/" + video_name)
+        self.video = cv2.VideoCapture("./test/resources/" + video_name)
 
     def delete(self):
-        # self.video.release()
+        self.video.release()
 
     def get_frame(self):
-        # get a frane from stream
-        # success, image = self.video.read()
-        return
-    """
+        """
+        get a frane from stream
+        """
+        success, image = self.video.read()
+        return success, image