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

[Enhancement] Add ABINet_Vision api (#1041) 

* add ABINet_Vision api

* revert change

Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
pull/1065/head
Abdelrahman Ahmad 2022-06-09 05:47:09 +02:00 committed by GitHub
parent 376fe05e6c
commit 9d7818b564
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
mmocr/utils/ocr.py
View File

@ -323,6 +323,10 @@ class MMOCR:
'config': 'abinet/abinet_academic.py',
'ckpt': 'abinet/abinet_academic-f718abf6.pth'
},
'ABINet_Vision': {
'config': 'abinet/abinet_vision_only_academic.py',
'ckpt': 'abinet/abinet_vision_only_academic-e6b9ea89.pth'
},
'SEG': {
'config': 'seg/seg_r31_1by16_fpnocr_academic.py',
'ckpt': 'seg/seg_r31_1by16_fpnocr_academic-72235b11.pth'