(self, device)
| 28 | self.compiler = None |
| 29 | |
| 30 | def load_model(self, device): |
| 31 | if self.compiler == "onnx": |
| 32 | return self._load_onnx_model(device) |
| 33 | elif self.compiler == "trt": |
| 34 | return self._load_trt_model().to(device) |
| 35 | elif self.compiler == "deploy": |
| 36 | self.cfg.model.model.auxiliary = {} |
| 37 | return create_model(self.cfg.model, class_num=self.class_num, weight_path=self.cfg.weight).to(device) |
| 38 | |
| 39 | def _load_onnx_model(self, device): |
| 40 | from onnxruntime import InferenceSession |
no test coverage detected