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

fix conflic in program

Browse Source
This commit is contained in:
andyjpaddle
2021-08-31 12:19:11 +00:00
parent ea52f47a95
commit ea019601a7
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/program.py
+1 -5
View File
@@ -402,11 +402,7 @@ def preprocess(is_train=False):
alg = config['Architecture']['algorithm']
assert alg in [
'EAST', 'DB', 'SAST', 'Rosetta', 'CRNN', 'STARNet', 'RARE', 'SRN',
<<<<<<< HEAD
'CLS', 'PGNet', 'Distillation', 'TableAttn', 'SAR'
=======
'CLS', 'PGNet', 'Distillation', 'NRTR', 'TableAttn'
>>>>>>> 63ed5fcab30801626ecf55a89f5dc9faf79a16d2
'CLS', 'PGNet', 'Distillation', 'NRTR', 'TableAttn', 'SAR'
]
device = 'gpu:{}'.format(dist.ParallelEnv().dev_id) if use_gpu else 'cpu'