opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'PaddlePaddle:dygraph' into dygraph

Browse Source
This commit is contained in:
whjdark
2022-07-21 15:29:21 +08:00
committed by GitHub
parent 8a98fb0c86 db5d66f9de
commit 9c72b7724c
5 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/rec/rec_mtb_nrtr.yml
+1 -1
View File
@@ -82,7 +82,7 @@ Train:
Eval:
dataset:
name: LMDBDataSet
data_dir: ./train_data/data_lmdb_release/validation/
data_dir: ./train_data/data_lmdb_release/evaluaiton/
transforms:
- DecodeImage: # load image
img_mode: BGR
configs/rec/rec_r45_abinet.yml
+1 -1
View File
@@ -82,7 +82,7 @@ Train:
Eval:
dataset:
name: LMDBDataSet
data_dir: ./train_data/data_lmdb_release/validation/
data_dir: ./train_data/data_lmdb_release/evaluation/
transforms:
- DecodeImage: # load image
img_mode: RGB
configs/rec/rec_svtrnet.yml
+2 -2
View File
@@ -77,7 +77,7 @@ Metric:
Train:
dataset:
name: LMDBDataSet
data_dir: ./train_data/data_lmdb_release/training
data_dir: ./train_data/data_lmdb_release/training/
transforms:
- DecodeImage: # load image
img_mode: BGR
@@ -97,7 +97,7 @@ Train:
Eval:
dataset:
name: LMDBDataSet
data_dir: ./train_data/data_lmdb_release/validation
data_dir: ./train_data/data_lmdb_release/evaluation/
transforms:
- DecodeImage: # load image
img_mode: BGR
configs/rec/rec_vitstr_none_ce.yml
+1 -1
View File
@@ -81,7 +81,7 @@ Train:
Eval:
dataset:
name: LMDBDataSet
data_dir: ./train_data/data_lmdb_release/validation/
data_dir: ./train_data/data_lmdb_release/evaluaiton/
transforms:
- DecodeImage: # load image
img_mode: BGR
ppocr/modeling/heads/rec_abinet_head.py
+2 -1
View File
@@ -273,7 +273,8 @@ def _get_length(logit):
out = out.cast('int32')
out = out.argmax(-1)
out = out + 1
out = paddle.where(abn, out, paddle.to_tensor(logit.shape[1]))
len_seq = paddle.zeros_like(out) + logit.shape[1]
out = paddle.where(abn, out, len_seq)
return out