fix bug
This commit is contained in:
parent
9cb3072033
commit
d179c7c446
|
@ -25,7 +25,7 @@ class CharacterOps(object):
|
|||
def __init__(self, config):
|
||||
self.character_type = config['character_type']
|
||||
self.loss_type = config['loss_type']
|
||||
self.max_text_len = 25
|
||||
self.max_text_len = config['max_text_length']
|
||||
if self.character_type == "en":
|
||||
self.character_str = "0123456789abcdefghijklmnopqrstuvwxyz"
|
||||
dict_character = list(self.character_str)
|
||||
|
|
|
@ -56,8 +56,8 @@ def parse_args():
|
|||
#params for text recognizer
|
||||
parser.add_argument("--rec_algorithm", type=str, default='CRNN')
|
||||
parser.add_argument("--rec_model_dir", type=str)
|
||||
parser.add_argument("--rec_image_shape", type=str, default="1, 64, 320")
|
||||
parser.add_argument("--rec_char_type", type=str, default='en')
|
||||
parser.add_argument("--rec_image_shape", type=str, default="3, 32, 320")
|
||||
parser.add_argument("--rec_char_type", type=str, default='ch')
|
||||
parser.add_argument("--rec_batch_num", type=int, default=30)
|
||||
parser.add_argument(
|
||||
"--rec_char_dict_path",
|
||||
|
|
Loading…
Reference in New Issue