Commit 22109853 authored by Sikhin VC's avatar Sikhin VC

added engine file

parent 4401bfc8
Pipeline #68936 failed with stage
...@@ -550,13 +550,13 @@ class CementBagCounter(ModelWrapper): ...@@ -550,13 +550,13 @@ class CementBagCounter(ModelWrapper):
def _predict(self, obj): def _predict(self, obj):
try: try:
frame = obj['frame'] frame = obj['frame']
# logger.info(f"Frame Received with shape {frame.shape}") logger.info(f"Frame Received with shape {frame.shape}")
# dets, frame, _dets, class_name = self.inference(frame) dets, frame, _dets, class_name = self.inference(frame)
# if len(dets) > 0: if len(dets) > 0:
# frame, objects, boxs = self.kalman_tracker(dets, frame) frame, objects, boxs = self.kalman_tracker(dets, frame)
# frame = self.update_bag_count(frame=frame, detection_objects=objects, class_name=class_name, frame = self.update_bag_count(frame=frame, detection_objects=objects, class_name=class_name,
# detections=_dets) detections=_dets)
# logger.debug("self.uncounted_objects --> {}".format(self.uncounted_objects)) logger.debug("self.uncounted_objects --> {}".format(self.uncounted_objects))
frame = self.draw_line_over_image(frame) frame = self.draw_line_over_image(frame)
obj['frame'] = cv2.resize(frame, (self.config.get('FRAME_WIDTH'), self.config. obj['frame'] = cv2.resize(frame, (self.config.get('FRAME_WIDTH'), self.config.
get('FRAME_HEIGHT'))) get('FRAME_HEIGHT')))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment