提交 cc8795c6 编写于 作者: 赛佬的小迷弟's avatar 赛佬的小迷弟

change args to self.args

上级 9c813bb3
...@@ -175,7 +175,7 @@ class TextDetector(object): ...@@ -175,7 +175,7 @@ class TextDetector(object):
st = time.time() st = time.time()
if args.benchmark: if self.args.benchmark:
self.autolog.times.start() self.autolog.times.start()
data = transform(data, self.preprocess_op) data = transform(data, self.preprocess_op)
...@@ -186,7 +186,7 @@ class TextDetector(object): ...@@ -186,7 +186,7 @@ class TextDetector(object):
shape_list = np.expand_dims(shape_list, axis=0) shape_list = np.expand_dims(shape_list, axis=0)
img = img.copy() img = img.copy()
if args.benchmark: if self.args.benchmark:
self.autolog.times.stamp() self.autolog.times.stamp()
self.input_tensor.copy_from_cpu(img) self.input_tensor.copy_from_cpu(img)
...@@ -195,7 +195,7 @@ class TextDetector(object): ...@@ -195,7 +195,7 @@ class TextDetector(object):
for output_tensor in self.output_tensors: for output_tensor in self.output_tensors:
output = output_tensor.copy_to_cpu() output = output_tensor.copy_to_cpu()
outputs.append(output) outputs.append(output)
if args.benchmark: if self.args.benchmark:
self.autolog.times.stamp() self.autolog.times.stamp()
preds = {} preds = {}
...@@ -220,7 +220,7 @@ class TextDetector(object): ...@@ -220,7 +220,7 @@ class TextDetector(object):
else: else:
dt_boxes = self.filter_tag_det_res(dt_boxes, ori_im.shape) 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) self.autolog.times.end(stamp=True)
et = time.time() et = time.time()
return dt_boxes, et - st return dt_boxes, et - st
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册