Skip to content

GitLab

Commit 8d803e24 authored by liuhy's avatar liuhy
Browse files

修改train参数

parent 921cc06f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
train.py
View file @ 8d803e24
...@@ -223,7 +223,7 @@ def get_parser(): ...@@ -223,7 +223,7 @@ def get_parser():
parser.add_argument('--max_epoch', default=15, help='epoch to train the network') parser.add_argument('--max_epoch', default=15, help='epoch to train the network')
parser.add_argument('--img_size', default=[94, 24], help='the image size') parser.add_argument('--img_size', default=[94, 24], help='the image size')
parser.add_argument('--train_img_dirs', default="data/train", help='the train images path') parser.add_argument('--train_img_dirs', default="data/train", help='the train images path')
parser.add_argument('--test_img_dirs', default="data/test", help='the test images path') parser.add_argument('--test_img_dirs', default="imgs", help='the test images path')
parser.add_argument('--dropout_rate', default=0.5, help='dropout rate.') parser.add_argument('--dropout_rate', default=0.5, help='dropout rate.')
parser.add_argument('--learning_rate', default=0.1, help='base value of learning rate.') parser.add_argument('--learning_rate', default=0.1, help='base value of learning rate.')
parser.add_argument('--lpr_max_len', default=8, help='license plate number max length.') parser.add_argument('--lpr_max_len', default=8, help='license plate number max length.')
......
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