Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "31adeb41cde6c2f6020c62f913c628f91d151738"
Commit fa2d08f2 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

add elapse to cls

parent b8834933
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tools/infer/predict_cls.py
View file @ fa2d08f2
...@@ -112,6 +112,7 @@ class TextClassifier(object): ...@@ -112,6 +112,7 @@ class TextClassifier(object):
if '180' in label and score > self.cls_thresh: if '180' in label and score > self.cls_thresh:
img_list[indices[beg_img_no + rno]] = cv2.rotate( img_list[indices[beg_img_no + rno]] = cv2.rotate(
img_list[indices[beg_img_no + rno]], 1) img_list[indices[beg_img_no + rno]], 1)
elapse = time.time() - starttime
return img_list, cls_res, elapse return img_list, cls_res, elapse
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment