From 9d7818b564b2b410c69c50c98488a40a3f3635fd Mon Sep 17 00:00:00 2001
From: Abdelrahman Ahmad <47902062+Abdelrahman350@users.noreply.github.com>
Date: Thu, 9 Jun 2022 05:47:09 +0200
Subject: [PATCH] [Enhancement] Add ABINet_Vision api (#1041)
* add ABINet_Vision api
* revert change
Co-authored-by: gaotongxiao <gaotongxiao@gmail.com>
---
mmocr/utils/ocr.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/mmocr/utils/ocr.py b/mmocr/utils/ocr.py
index 4b7db05a..18bc454c 100755
--- a/mmocr/utils/ocr.py
+++ b/mmocr/utils/ocr.py
@@ -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'