未验证 提交 1dc5c014 编写于 作者: G Guanghua Yu 提交者: GitHub

fix benchmark in deploy with solov2 (#3391)

上级 87848d4d
...@@ -213,6 +213,7 @@ class DetectorSOLOv2(Detector): ...@@ -213,6 +213,7 @@ class DetectorSOLOv2(Detector):
cpu_threads=cpu_threads, cpu_threads=cpu_threads,
enable_mkldnn=enable_mkldnn) enable_mkldnn=enable_mkldnn)
self.det_times = Timer() self.det_times = Timer()
self.cpu_mem, self.gpu_mem, self.gpu_util = 0, 0, 0
def predict(self, image, threshold=0.5, warmup=0, repeats=1): def predict(self, image, threshold=0.5, warmup=0, repeats=1):
''' '''
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册