Commit 1a7a0d4e authored by Sikhin VC's avatar Sikhin VC

added engine file

parent ac0b8a6d
Pipeline #69026 failed with stage
No preview for this file type
......@@ -34,7 +34,7 @@ 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_47.engine"
engine_file_path = "data/jk_v5_cam_44.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))
......
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