Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
7eede6b4
Commit
7eede6b4
authored
Dec 01, 2020
by
WenmuZhou
Browse files
use cls_batch_num replace rec_batch_num
parent
1c43e8bb
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tools/infer/predict_cls.py
tools/infer/predict_cls.py
+1
-1
No files found.
tools/infer/predict_cls.py
View file @
7eede6b4
...
...
@@ -37,7 +37,7 @@ logger = get_logger()
class
TextClassifier
(
object
):
def
__init__
(
self
,
args
):
self
.
cls_image_shape
=
[
int
(
v
)
for
v
in
args
.
cls_image_shape
.
split
(
","
)]
self
.
cls_batch_num
=
args
.
rec
_batch_num
self
.
cls_batch_num
=
args
.
cls
_batch_num
self
.
cls_thresh
=
args
.
cls_thresh
self
.
use_zero_copy_run
=
args
.
use_zero_copy_run
postprocess_params
=
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment