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
ModelZoo
LPRNet_pytorch
Commits
7cb703c4
Commit
7cb703c4
authored
Mar 02, 2023
by
liuhy
Browse files
修改test代码
parent
679ae84d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
3 deletions
+1
-3
test.py
test.py
+1
-3
No files found.
test.py
View file @
7cb703c4
...
...
@@ -75,9 +75,7 @@ def validation(args):
if
__name__
==
'__main__'
:
parser
=
argparse
.
ArgumentParser
(
description
=
'parameters to vaildate net'
)
parser
.
add_argument
(
'--model'
,
default
=
'model/lprnet.pth'
,
help
=
'model path to vaildate'
)
# parser.add_argument('--model', default='weights/Final_LPRNet_model.pth', help='model path to vaildate')
# parser.add_argument('--imgpath', default='imgs/川JK0707.jpg', help='the image path')
parser
.
add_argument
(
'--imgpath'
,
default
=
'/code/lpr_ori/data/test'
,
help
=
'the image path'
)
parser
.
add_argument
(
'--imgpath'
,
default
=
'imgs'
,
help
=
'the image path'
)
parser
.
add_argument
(
'--device'
,
default
=
'cuda'
,
help
=
'Use cuda to vaildate model'
)
parser
.
add_argument
(
'--export_onnx'
,
default
=
False
,
help
=
'export model to onnx'
)
parser
.
add_argument
(
'--dynamic'
,
default
=
False
,
help
=
'use dynamic batch size'
)
...
...
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