From cdd98d085fb99d41c651f7394505291e9f2a8c74 Mon Sep 17 00:00:00 2001 From: Double_V Date: Tue, 29 Jun 2021 10:01:24 +0800 Subject: [PATCH] Update predict_det.py --- tools/infer/predict_det.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index e1d88fae..4253964e 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -174,8 +174,9 @@ class TextDetector(object): data = {'image': img} st = time.time() - - self.autolog.times.start() + + if args.benchmark: + self.autolog.times.start() data = transform(data, self.preprocess_op) img, shape_list = data @@ -185,7 +186,8 @@ class TextDetector(object): shape_list = np.expand_dims(shape_list, axis=0) img = img.copy() - self.autolog.times.stamp() + if args.benchmark: + self.autolog.times.stamp() self.input_tensor.copy_from_cpu(img) self.predictor.run() @@ -193,8 +195,8 @@ class TextDetector(object): for output_tensor in self.output_tensors: output = output_tensor.copy_to_cpu() outputs.append(output) - - self.autolog.times.stamp() + if args.benchmark: + self.autolog.times.stamp() preds = {} if self.det_algorithm == "EAST": @@ -218,7 +220,8 @@ class TextDetector(object): else: dt_boxes = self.filter_tag_det_res(dt_boxes, ori_im.shape) - self.autolog.times.end(stamp=True) + if args.benchmark: + self.autolog.times.end(stamp=True) et = time.time() return dt_boxes, et - st @@ -259,6 +262,7 @@ if __name__ == "__main__": "det_res_{}".format(img_name_pure)) cv2.imwrite(img_path, src_im) logger.info("The visualized image saved in {}".format(img_path)) - - text_detector.autolog.report() + + if args.benchmark: + text_detector.autolog.report() -- GitLab