mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix #416: Fix undesired assignment to "pretrained" in test.py (#418) 

* fix ocr linting

* fix undesired assignment to pretrained key
pull/427/head
Tong Gao 2021-08-10 22:07:41 +08:00 committed by GitHub
parent f62b4513f5
commit 532e8f808d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmocr/utils/ocr.py
View File

@ -275,8 +275,7 @@ class MMOCR:
},
'RobustScanner': {
'config': 'robust_scanner/robustscanner_r31_academic.py',
'ckpt':
'robustscanner/robustscanner_r31_academic-5f05874f.pth'
'ckpt': 'robustscanner/robustscanner_r31_academic-5f05874f.pth'
},
'SEG': {
'config': 'seg/seg_r31_1by16_fpnocr_academic.py',

3
tools/test.py
View File

@ -127,7 +127,8 @@ def main():
# set cudnn_benchmark
if cfg.get('cudnn_benchmark', False):
torch.backends.cudnn.benchmark = True
cfg.model.pretrained = None
if cfg.model.get('pretrained'):
cfg.model.pretrained = None
if cfg.model.get('neck'):
if isinstance(cfg.model.neck, list):
for neck_cfg in cfg.model.neck: