Added answer generation blueprint

Author: anujanegi

.gitignore

@@ -8,3 +8,5 @@ __pycache__/
 # Test images
 images/*
 
+# External scripts
+external/

modules/answer_generator.py

@@ -0,0 +1,13 @@
+class AnswerGenerator:
+    @staticmethod
+    def predict(comprehension, question):
+        """
+        predict answer for a question based on the given data
+        :param comprehension: data to read for answering the question
+        :param question: question based on the comprehension
+        :return: answer for the question
+        """
+        pass
+
+ag = AnswerGenerator()
+print(ag.predict("There are two people.", "How many people are there?"))

modules/object_detection.py

@@ -12,15 +12,14 @@ class ObjectDetector:
 
     @staticmethod
     def predict(frame):
-
-        cv_net = cv2.dnn.readNetFromTensorflow(PATH_TO_MODEL_WEIGHTS, PATH_TO_GRAPH)
-        labels = coco_label_reader(PATH_TO_LABELS)
-
         """
         predict objects in the frame
         :param frame: input image as numpy array
         :return: objects and bounding boxes
         """
+        cv_net = cv2.dnn.readNetFromTensorflow(PATH_TO_MODEL_WEIGHTS, PATH_TO_GRAPH)
+        labels = coco_label_reader(PATH_TO_LABELS)
+
         rows, cols, _ = frame.shape
         blob = cv2.dnn.blobFromImage(frame, size=(rows, cols), swapRB=True, crop=False)
         cv_net.setInput(blob)