Update predict_det.py

tools/infer/predict_det.py

@@ -175,6 +175,7 @@ class TextDetector(object):
 
         st = time.time()
         
+        if args.benchmark:
             self.autolog.times.start()
 
         data = transform(data, self.preprocess_op)
@@ -185,6 +186,7 @@ class TextDetector(object):
         shape_list = np.expand_dims(shape_list, axis=0)
         img = img.copy()
 
+        if args.benchmark:
             self.autolog.times.stamp()
 
         self.input_tensor.copy_from_cpu(img)
@@ -193,7 +195,7 @@ class TextDetector(object):
         for output_tensor in self.output_tensors:
             output = output_tensor.copy_to_cpu()
             outputs.append(output)
-
+        if args.benchmark:
             self.autolog.times.stamp()
 
         preds = {}
@@ -218,6 +220,7 @@ class TextDetector(object):
         else:
             dt_boxes = self.filter_tag_det_res(dt_boxes, ori_im.shape)
 
+        if args.benchmark:
             self.autolog.times.end(stamp=True)
         et = time.time()
         return dt_boxes, et - st
@@ -260,5 +263,6 @@ if __name__ == "__main__":
         cv2.imwrite(img_path, src_im)
         logger.info("The visualized image saved in {}".format(img_path))
     
+    if args.benchmark:
         text_detector.autolog.report()