added engine file

scripts/cement_counter.py

@@ -17,8 +17,7 @@ from scripts.utils.helpers import box_iou2
 from scripts.utils.image_utils import draw_circles_on_frame, resize_to_64_64
 from scripts.utils.infocenter import MongoLogger
 from scripts.utils.tracker import Tracker
-# from scripts.utils.yolov5_trt import YoloV5TRT
-# from sklearn.utils.linear_assignment_ import linear_assignment
+from scripts.utils.yolov5_trt import YoloV5TRT
 from scipy.optimize import linear_sum_assignment as linear_assignment
 from edge_engine.common.config import EDGE_CONFIG
 
@@ -35,10 +34,10 @@ class CementBagCounter(ModelWrapper):
         self.device_id = device_id
         self.rtp = pubs.rtp_write
         self.frame_skip = self.config.get('frame_skip', False)
-        engine_file_path = "data/jk_v5_cam_42.engine"
+        engine_file_path = "data/jk_v5_cam_47.engine"
         self.classes = {0: 'cement_bag'}
-        # self.yolo_v5_wrapper = YoloV5TRT(engine_file_path, model_config.get('conf_thresh', 0.7),
-        #                                  model_config.get('iou_thresh', 0.15))
+        self.yolo_v5_wrapper = YoloV5TRT(engine_file_path, model_config.get('conf_thresh', 0.7),
+                                         model_config.get('iou_thresh', 0.15))
 
         # TRT Additions stop