added engine file

scripts/cement_counter.py

@@ -238,10 +238,14 @@ class CementBagCounter(ModelWrapper):
         """
         if not self.janus_metadata.get('metadata'):
             self.janus_metadata['metadata'] = EDGE_CONFIG["extra_fields"]
-
-        _coordinates = [self.janus_metadata['metadata'].get(coordinate_key) for coordinate_key in
-                        JanusDeploymentConstants.LINE_COORDINATES]
-        _alignment = self.janus_metadata['metadata'].get(JanusDeploymentConstants.ALIGNMENT_KEY)
+        _coordinates = []
+        for data in self.janus_metadata["metadata"]:
+            print("data", data)
+            for elem in data:
+                if data[elem] in JanusDeploymentConstants.LINE_COORDINATES:
+                    _coordinates.append(data["value"])
+                if data[elem] == JanusDeploymentConstants.ALIGNMENT_KEY:
+                    _alignment = data["value"]
 
         return _alignment, _coordinates
 
@@ -546,13 +550,13 @@ class CementBagCounter(ModelWrapper):
     def _predict(self, obj):
         try:
             frame = obj['frame']
-            logger.info(f"Frame Received with shape {frame.shape}")
-            dets, frame, _dets, class_name = self.inference(frame)
-            if len(dets) > 0:
-                frame, objects, boxs = self.kalman_tracker(dets, frame)
-                frame = self.update_bag_count(frame=frame, detection_objects=objects, class_name=class_name,
-                                              detections=_dets)
-                logger.debug("self.uncounted_objects --> {}".format(self.uncounted_objects))
+            # logger.info(f"Frame Received with shape {frame.shape}")
+            # dets, frame, _dets, class_name = self.inference(frame)
+            # if len(dets) > 0:
+            #     frame, objects, boxs = self.kalman_tracker(dets, frame)
+            #     frame = self.update_bag_count(frame=frame, detection_objects=objects, class_name=class_name,
+            #                                   detections=_dets)
+                # logger.debug("self.uncounted_objects --> {}".format(self.uncounted_objects))
             frame = self.draw_line_over_image(frame)
             obj['frame'] = cv2.resize(frame, (self.config.get('FRAME_WIDTH'), self.config.
                                               get('FRAME_HEIGHT')))