modify: 性能测试

YOLOv8/yolov8_detector.py

@@ -31,29 +31,34 @@ class Detector:
     """完整的YOLOv8检测器，包含可视化"""
     
     def __init__(self, 
+                 handle:sail.Handle,
                  config_dict: dict, 
                  is_vfs=False, 
                  vfs_path: str="./application.svfs", 
-                 dev_id=0, conf_thresh = 0.25, 
+                 conf_thresh = 0.25, 
                  nms_thresh = 0.7,
                  logger: Optional[Logger] = None):
         if logger is None:
             self.logger = getLogger()
         else:
             self.logger = logger
-        self.dev_id = dev_id
+        self.handle = handle
         yolov8_config = YOLOv8Config.from_dict(config_dict)
         # 加载bmodel
+        self.net = sail.Engine(self.handle)
         if is_vfs:
             vfs = SQLiteVFS(vfs_path)
             bmodel_bytes = vfs.read_file(yolov8_config.bmodel_path)
             bmodel_size = vfs.get_file_info(yolov8_config.bmodel_path)["file_size"] # type: ignore
-            self.net = sail.Engine(bmodel_bytes, bmodel_size, dev_id, sail.IOMode.SYSO)
+            self.net.load(bmodel_bytes, bmodel_size)
+            # self.net = sail.Engine(bmodel_bytes, bmodel_size, dev_id, sail.IOMode.SYSO)
         else:
-            self.net = sail.Engine(yolov8_config.bmodel_path, dev_id, sail.IOMode.SYSO)
+            self.net.load(yolov8_config.bmodel_path)
+            # self.net = sail.Engine(yolov8_config.bmodel_path, dev_id, sail.IOMode.SYSO)
+        graph_name = self.net.get_graph_names()[0]
+        self.net.set_io_mode(graph_name, sail.IOMode.DEVIO)
         self.logger.info(f"加载模型: {yolov8_config.bmodel_path}")
         
-        self.handle = sail.Handle(dev_id)
         self.bmcv = sail.Bmcv(self.handle)
         self.graph_name = self.net.get_graph_names()[0]
         
@@ -281,7 +286,7 @@ class Detector:
         self.logger.info(f"\n处理图像: {image_path}")
         
         # 1. 解码图像
-        decoder = sail.Decoder(image_path, True, self.dev_id)
+        decoder = sail.Decoder(image_path, True)
         bmimg = sail.BMImage()
         ret = decoder.read(self.handle, bmimg)