From cc8795c68aba7b7c5ef2e77fe559c04b74045a6f Mon Sep 17 00:00:00 2001
From: LuGu <2623587501@qq.com>
Date: Mon, 5 Jul 2021 20:55:38 +0800
Subject: [PATCH] change args to self.args

---
 tools/infer/predict_det.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index 4253964ec7..4e3ab18f6b 100755
--- a/tools/infer/predict_det.py
+++ b/tools/infer/predict_det.py
@@ -175,7 +175,7 @@ def __call__(self, img):
 
         st = time.time()
         
-        if args.benchmark:
+        if self.args.benchmark:
             self.autolog.times.start()
 
         data = transform(data, self.preprocess_op)
@@ -186,7 +186,7 @@ def __call__(self, img):
         shape_list = np.expand_dims(shape_list, axis=0)
         img = img.copy()
 
-        if args.benchmark:
+        if self.args.benchmark:
             self.autolog.times.stamp()
 
         self.input_tensor.copy_from_cpu(img)
@@ -195,7 +195,7 @@ def __call__(self, img):
         for output_tensor in self.output_tensors:
             output = output_tensor.copy_to_cpu()
             outputs.append(output)
-        if args.benchmark:
+        if self.args.benchmark:
             self.autolog.times.stamp()
 
         preds = {}
@@ -220,7 +220,7 @@ def __call__(self, img):
         else:
             dt_boxes = self.filter_tag_det_res(dt_boxes, ori_im.shape)
 
-        if args.benchmark:
+        if self.args.benchmark:
             self.autolog.times.end(stamp=True)
         et = time.time()
         return dt_boxes, et - st