OrgLion-ML
/
deep-text-recognition-benchmark
mirror of https://github.com/clovaai/deep-text-recognition-benchmark.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Update test.py

This commit is contained in:
Kwon 2019-05-08 15:16:45 +09:00 committed by GitHub
parent 3772d75f40
commit 3a1ebfadc3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -135,6 +135,10 @@ def test(opt):
converter = CTCLabelConverter(opt.character) converter = CTCLabelConverter(opt.character)
else: else:
converter = AttnLabelConverter(opt.character) converter = AttnLabelConverter(opt.character)
if opt.rgb:
opt.input_channel = 3
opt.num_class = len(converter.character) opt.num_class = len(converter.character)
model = Model(opt) model = Model(opt)
print('model input parameters', opt.imgH, opt.imgW, opt.num_fiducial, opt.input_channel, opt.output_channel, print('model input parameters', opt.imgH, opt.imgW, opt.num_fiducial, opt.input_channel, opt.output_channel,