forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #982 from FLming/develop 

因Python3.6以前的字典类型是无序的而导致的错误
This commit is contained in:
littletomatodonkey 2020-10-21 19:52:22 +08:00 committed by GitHub
parent 5238c0bd47 2734733630
commit 606f869d01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
ppocr/modeling/architectures/rec_model.py
View File

@ -16,6 +16,8 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from collections import OrderedDict
from paddle import fluid
from ppocr.utils.utility import create_module
@ -215,16 +217,15 @@ class RecModel(object):
label = labels['label']
if self.loss_type == 'srn':
total_loss, img_loss, word_loss = self.loss(predicts, labels)
outputs = {
'total_loss': total_loss,
'img_loss': img_loss,
'word_loss': word_loss,
'decoded_out': decoded_out,
'label': label
}
outputs = OrderedDict([('total_loss', total_loss),
('img_loss', img_loss),
('word_loss', word_loss),
('decoded_out', decoded_out),
('label', label)])
else:
outputs = {'total_loss':loss, 'decoded_out':\
decoded_out, 'label':label}
outputs = OrderedDict([('total_loss', loss),
('decoded_out', decoded_out),
('label', label)])
return loader, outputs
# export_model
elif mode == "export":
@ -233,16 +234,15 @@ class RecModel(object):
predict = fluid.layers.softmax(predict)
if self.loss_type == "srn":
return [
image, labels, {
'decoded_out': decoded_out,
'predicts': predict
}
]
image, labels, OrderedDict([('decoded_out', decoded_out),
('predicts', predict)])]
return [image, {'decoded_out': decoded_out, 'predicts': predict}]
return [image, OrderedDict([('decoded_out', decoded_out),
('predicts', predict)])]
# eval or test
else:
predict = predicts['predict']
if self.loss_type == "ctc":
predict = fluid.layers.softmax(predict)
return loader, {'decoded_out': decoded_out, 'predicts': predict}
return loader, OrderedDict([('decoded_out', decoded_out),
('predicts', predict)])