From 1dc5c01476905d5f8d613c5efa6be5fd116f745c Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Wed, 16 Jun 2021 12:33:20 +0800
Subject: [PATCH] fix benchmark in deploy with solov2 (#3391)

---
 deploy/python/infer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 84f52b07b..65cbce70d 100644
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -213,6 +213,7 @@ class DetectorSOLOv2(Detector):
             cpu_threads=cpu_threads,
             enable_mkldnn=enable_mkldnn)
         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):
         '''
-- 
GitLab