Skip to content

J
jk_edge_code_api_integration
Commit 22109853 authored by Sikhin VC's avatar Sikhin VC
Browse files
Options

added engine file

parent 4401bfc8
Pipeline #68936 failed with stage
    Hide whitespace changes
    Inline Side-by-side
    Showing with 7 additions and 7 deletions
    scripts/cement_counter.py
    View file @ 22109853
    ......@@ -550,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')))
    ......
    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