sunnychenxiwang commited on
Commit
24c4def
1 Parent(s): 50e84c0

update all

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. __init__.py +0 -0
  2. pipeline/GroundingDINO +1 -0
  3. pipeline/__pycache__/claim_generate.cpython-39.pyc +0 -0
  4. pipeline/__pycache__/judge.cpython-39.pyc +0 -0
  5. pipeline/__pycache__/openai_wrapper.cpython-39.pyc +0 -0
  6. pipeline/__pycache__/query_generate.cpython-39.pyc +0 -0
  7. pipeline/__pycache__/run.cpython-39.pyc +0 -0
  8. pipeline/__pycache__/run_pipeline.cpython-39.pyc +0 -0
  9. pipeline/__pycache__/tool_execute.cpython-39.pyc +0 -0
  10. pipeline/claim_generate.py +31 -0
  11. pipeline/judge.py +74 -0
  12. pipeline/mmocr/.circleci/config.yml +34 -0
  13. pipeline/mmocr/.circleci/docker/Dockerfile +11 -0
  14. pipeline/mmocr/.circleci/test.yml +196 -0
  15. pipeline/mmocr/.codespellrc +5 -0
  16. pipeline/mmocr/.coveragerc +3 -0
  17. pipeline/mmocr/.dev_scripts/benchmark_full_models.txt +18 -0
  18. pipeline/mmocr/.dev_scripts/benchmark_options.py +7 -0
  19. pipeline/mmocr/.dev_scripts/benchmark_train_models.txt +9 -0
  20. pipeline/mmocr/.dev_scripts/covignore.cfg +18 -0
  21. pipeline/mmocr/.dev_scripts/diff_coverage_test.sh +43 -0
  22. pipeline/mmocr/.github/CODE_OF_CONDUCT.md +76 -0
  23. pipeline/mmocr/.github/CONTRIBUTING.md +1 -0
  24. pipeline/mmocr/.github/ISSUE_TEMPLATE/1-bug-report.yml +121 -0
  25. pipeline/mmocr/.github/ISSUE_TEMPLATE/2-feature_request.yml +39 -0
  26. pipeline/mmocr/.github/ISSUE_TEMPLATE/3-new-model.yml +51 -0
  27. pipeline/mmocr/.github/ISSUE_TEMPLATE/4-documentation.yml +48 -0
  28. pipeline/mmocr/.github/ISSUE_TEMPLATE/config.yml +12 -0
  29. pipeline/mmocr/.github/pull_request_template.md +33 -0
  30. pipeline/mmocr/.github/workflows/lint.yml +27 -0
  31. pipeline/mmocr/.github/workflows/merge_stage_test.yml +160 -0
  32. pipeline/mmocr/.github/workflows/pr_stage_test.yml +102 -0
  33. pipeline/mmocr/.github/workflows/publish-to-pypi.yml +26 -0
  34. pipeline/mmocr/.github/workflows/test_mim.yml +44 -0
  35. pipeline/mmocr/.gitignore +146 -0
  36. pipeline/mmocr/.owners.yml +9 -0
  37. pipeline/mmocr/.pre-commit-config.yaml +63 -0
  38. pipeline/mmocr/.pylintrc +621 -0
  39. pipeline/mmocr/.readthedocs.yml +9 -0
  40. pipeline/mmocr/CITATION.cff +9 -0
  41. pipeline/mmocr/LICENSE +203 -0
  42. pipeline/mmocr/MANIFEST.in +5 -0
  43. pipeline/mmocr/README.md +251 -0
  44. pipeline/mmocr/README_zh-CN.md +250 -0
  45. pipeline/mmocr/__init__.py +1 -0
  46. pipeline/mmocr/configs/backbone/oclip/README.md +41 -0
  47. pipeline/mmocr/configs/backbone/oclip/metafile.yml +13 -0
  48. pipeline/mmocr/configs/kie/_base_/datasets/wildreceipt-openset.py +26 -0
  49. pipeline/mmocr/configs/kie/_base_/datasets/wildreceipt.py +16 -0
  50. pipeline/mmocr/configs/kie/_base_/default_runtime.py +33 -0
__init__.py ADDED
File without changes
pipeline/GroundingDINO ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit 60d796825e1266e56f7e4e9e00e88de662b67bd3
pipeline/__pycache__/claim_generate.cpython-39.pyc ADDED
Binary file (1.5 kB). View file
 
pipeline/__pycache__/judge.cpython-39.pyc ADDED
Binary file (2.37 kB). View file
 
pipeline/__pycache__/openai_wrapper.cpython-39.pyc ADDED
Binary file (2.6 kB). View file
 
pipeline/__pycache__/query_generate.cpython-39.pyc ADDED
Binary file (3.78 kB). View file
 
pipeline/__pycache__/run.cpython-39.pyc ADDED
Binary file (1.7 kB). View file
 
pipeline/__pycache__/run_pipeline.cpython-39.pyc ADDED
Binary file (1.73 kB). View file
 
pipeline/__pycache__/tool_execute.cpython-39.pyc ADDED
Binary file (2.54 kB). View file
 
pipeline/claim_generate.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import json
2
+ import yaml
3
+
4
+ class ClaimGenerator:
5
+ def __init__(self, prompt_path, chat):
6
+ with open(prompt_path,"r",encoding='utf-8') as file:
7
+ self.prompt = yaml.load(file, yaml.FullLoader)
8
+ self.chat = chat
9
+
10
+ def get_response(self, text):
11
+ user_prompt = self.prompt["user"].format(text=text)
12
+ message = [
13
+ {"role": "system", "content": self.prompt["system"]},
14
+ {"role": "user", "content": user_prompt}
15
+ ]
16
+ response = self.chat.get_response(message=message)
17
+ try:
18
+ response = json.loads(response)
19
+ except Exception as e:
20
+ print(e) # 是否有问题
21
+
22
+ claim_list = []
23
+ cnt = 0
24
+ for seg in response:
25
+ for cla in seg["claims"]:
26
+ cnt=(lambda x:x+1)(cnt)
27
+ claim_list.append("claim{}: {}".format(str(cnt), cla["claim"]))
28
+ claim_list = "\n".join([claim for claim in claim_list])
29
+ return response, claim_list
30
+
31
+
pipeline/judge.py ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import json
2
+ import yaml
3
+ import base64
4
+ class Judger:
5
+ def __init__(self, prompt_path, chat, type):
6
+ with open(prompt_path,"r",encoding='utf-8') as file:
7
+ self.prompt = yaml.load(file, yaml.FullLoader)[type]
8
+ self.chat = chat
9
+
10
+
11
+ def encode_image(self, image_path):
12
+ with open(image_path, "rb") as image_file:
13
+ return base64.b64encode(image_file.read()).decode('utf-8')
14
+
15
+ def get_response(self, object_res, attribue_res, text_res, fact_res, claim_list, image_path):
16
+ input = '''
17
+ Here is the object detection expert model's result:
18
+ {object}
19
+
20
+ Here is the scene text recognition expert model's result:
21
+ {text}
22
+
23
+ Here is the external knowledge:
24
+ {fact}
25
+
26
+ Here is the claim list:
27
+ {claims}
28
+
29
+ Output:
30
+ '''
31
+
32
+ object_det = ""
33
+ text_det = ""
34
+ for object_name, box in zip(object_res["phrases"], object_res["boxes"]):
35
+ object_det += object_name + " " + str(box) + "\n"
36
+
37
+ if text_res != None:
38
+ for text_name, box in zip(text_res["phrases"], text_res["boxes"]):
39
+ text_det += text_name + " " + str(box) + "\n"
40
+ else:
41
+ text_det = "none information"
42
+
43
+
44
+ img1 = self.encode_image("/home/wcx/wcx/GroundingDINO/LVLM/cot/img_examples/animal.jpg")
45
+ img2 = self.encode_image("/home/wcx/wcx/GroundingDINO/LVLM/cot/img_examples/ball.jpg")
46
+ base64_source_image = self.encode_image(image_path)
47
+ content = [
48
+ {"type": "text", "text": self.prompt["user"]},
49
+ {"type": "image_url","image_url": f"data:image/jpeg;base64,{img1}"},
50
+ {"type": "text", "text": self.prompt["example1"]},
51
+ {"type": "image_url","image_url": f"data:image/jpeg;base64,{img2}"},
52
+ {"type": "text", "text": self.prompt["example2"]},
53
+ {"type": "image_url","image_url": f"data:image/jpeg;base64,{base64_source_image}"},
54
+ {"type": "text", "text": input.format(object=object_det,text=text_det,fact=fact_res,claims=claim_list)}
55
+ ]
56
+
57
+ message = [
58
+ {
59
+ 'role': 'system',
60
+ 'content': self.prompt["system"]
61
+ },
62
+ {
63
+ "role": "user",
64
+ "content": content,
65
+ }
66
+ ]
67
+
68
+ response = self.chat.get_response(message=message)
69
+ try:
70
+ response = json.loads(response)
71
+ except Exception as e:
72
+ print(e)
73
+ return response
74
+
pipeline/mmocr/.circleci/config.yml ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ version: 2.1
2
+
3
+ # this allows you to use CircleCI's dynamic configuration feature
4
+ setup: true
5
+
6
+ # the path-filtering orb is required to continue a pipeline based on
7
+ # the path of an updated fileset
8
+ orbs:
9
+ path-filtering: circleci/path-filtering@0.1.2
10
+
11
+ workflows:
12
+ # the always-run workflow is always triggered, regardless of the pipeline parameters.
13
+ always-run:
14
+ jobs:
15
+ # the path-filtering/filter job determines which pipeline
16
+ # parameters to update.
17
+ - path-filtering/filter:
18
+ name: check-updated-files
19
+ # 3-column, whitespace-delimited mapping. One mapping per
20
+ # line:
21
+ # <regex path-to-test> <parameter-to-set> <value-of-pipeline-parameter>
22
+ mapping: |
23
+ mmocr/.* lint_only false
24
+ requirements/.* lint_only false
25
+ tests/.* lint_only false
26
+ tools/.* lint_only false
27
+ configs/.* lint_only false
28
+ .circleci/.* lint_only false
29
+ base-revision: dev-1.x
30
+ # this is the path of the configuration we should trigger once
31
+ # path filtering and pipeline parameter value updates are
32
+ # complete. In this case, we are using the parent dynamic
33
+ # configuration itself.
34
+ config-path: .circleci/test.yml
pipeline/mmocr/.circleci/docker/Dockerfile ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ARG PYTORCH="1.8.1"
2
+ ARG CUDA="10.2"
3
+ ARG CUDNN="7"
4
+
5
+ FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
6
+
7
+ # To fix GPG key error when running apt-get update
8
+ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
9
+ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
10
+
11
+ RUN apt-get update && apt-get install -y ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx
pipeline/mmocr/.circleci/test.yml ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ version: 2.1
2
+
3
+ # the default pipeline parameters, which will be updated according to
4
+ # the results of the path-filtering orb
5
+ parameters:
6
+ lint_only:
7
+ type: boolean
8
+ default: true
9
+
10
+ jobs:
11
+ lint:
12
+ docker:
13
+ - image: cimg/python:3.7.4
14
+ steps:
15
+ - checkout
16
+ - run:
17
+ name: Install pre-commit hook
18
+ command: |
19
+ pip install pre-commit
20
+ pre-commit install
21
+ - run:
22
+ name: Linting
23
+ command: pre-commit run --all-files
24
+ - run:
25
+ name: Check docstring coverage
26
+ command: |
27
+ pip install interrogate
28
+ interrogate -v --ignore-init-method --ignore-module --ignore-nested-functions --ignore-magic --ignore-regex "__repr__" --fail-under 90 mmocr
29
+ build_cpu:
30
+ parameters:
31
+ # The python version must match available image tags in
32
+ # https://circleci.com/developer/images/image/cimg/python
33
+ python:
34
+ type: string
35
+ torch:
36
+ type: string
37
+ torchvision:
38
+ type: string
39
+ docker:
40
+ - image: cimg/python:<< parameters.python >>
41
+ resource_class: large
42
+ steps:
43
+ - checkout
44
+ - run:
45
+ name: Install Libraries
46
+ command: |
47
+ sudo apt-get update
48
+ sudo apt-get install -y ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx libjpeg-dev zlib1g-dev libtinfo-dev libncurses5 libgeos-dev
49
+ - run:
50
+ name: Configure Python & pip
51
+ command: |
52
+ pip install --upgrade pip
53
+ pip install wheel
54
+ - run:
55
+ name: Install PyTorch
56
+ command: |
57
+ python -V
58
+ pip install torch==<< parameters.torch >>+cpu torchvision==<< parameters.torchvision >>+cpu -f https://download.pytorch.org/whl/torch_stable.html
59
+ - run:
60
+ name: Install mmocr dependencies
61
+ command: |
62
+ pip install git+https://github.com/open-mmlab/mmengine.git@main
63
+ pip install -U openmim
64
+ mim install 'mmcv >= 2.0.0rc1'
65
+ pip install git+https://github.com/open-mmlab/mmdetection.git@dev-3.x
66
+ pip install -r requirements/tests.txt
67
+ - run:
68
+ name: Build and install
69
+ command: |
70
+ pip install -e .
71
+ - run:
72
+ name: Run unittests
73
+ command: |
74
+ coverage run --branch --source mmocr -m pytest tests/
75
+ coverage xml
76
+ coverage report -m
77
+ build_cuda:
78
+ parameters:
79
+ torch:
80
+ type: string
81
+ cuda:
82
+ type: enum
83
+ enum: ["10.1", "10.2", "11.1", "11.7"]
84
+ cudnn:
85
+ type: integer
86
+ default: 7
87
+ machine:
88
+ image: ubuntu-2004-cuda-11.4:202110-01
89
+ # docker_layer_caching: true
90
+ resource_class: gpu.nvidia.small
91
+ steps:
92
+ - checkout
93
+ - run:
94
+ # Cloning repos in VM since Docker doesn't have access to the private key
95
+ name: Clone Repos
96
+ command: |
97
+ git clone -b main --depth 1 https://github.com/open-mmlab/mmengine.git /home/circleci/mmengine
98
+ git clone -b dev-3.x --depth 1 https://github.com/open-mmlab/mmdetection.git /home/circleci/mmdetection
99
+ - run:
100
+ name: Build Docker image
101
+ command: |
102
+ docker build .circleci/docker -t mmocr:gpu --build-arg PYTORCH=<< parameters.torch >> --build-arg CUDA=<< parameters.cuda >> --build-arg CUDNN=<< parameters.cudnn >>
103
+ docker run --gpus all -t -d -v /home/circleci/project:/mmocr -v /home/circleci/mmengine:/mmengine -v /home/circleci/mmdetection:/mmdetection -w /mmocr --name mmocr mmocr:gpu
104
+ - run:
105
+ name: Install mmocr dependencies
106
+ command: |
107
+ docker exec mmocr pip install -e /mmengine
108
+ docker exec mmocr pip install -U openmim
109
+ docker exec mmocr mim install 'mmcv >= 2.0.0rc1'
110
+ docker exec mmocr pip install -e /mmdetection
111
+ docker exec mmocr pip install -r requirements/tests.txt
112
+ - run:
113
+ name: Build and install
114
+ command: |
115
+ docker exec mmocr pip install -e .
116
+ - run:
117
+ name: Run unittests
118
+ command: |
119
+ docker exec mmocr pytest tests/
120
+
121
+ workflows:
122
+ pr_stage_lint:
123
+ when: << pipeline.parameters.lint_only >>
124
+ jobs:
125
+ - lint:
126
+ name: lint
127
+ filters:
128
+ branches:
129
+ ignore:
130
+ - dev-1.x
131
+ - 1.x
132
+ - main
133
+ pr_stage_test:
134
+ when:
135
+ not:
136
+ << pipeline.parameters.lint_only >>
137
+ jobs:
138
+ - lint:
139
+ name: lint
140
+ filters:
141
+ branches:
142
+ ignore:
143
+ - dev-1.x
144
+ - test-1.x
145
+ - main
146
+ - build_cpu:
147
+ name: minimum_version_cpu
148
+ torch: 1.6.0
149
+ torchvision: 0.7.0
150
+ python: "3.7"
151
+ requires:
152
+ - lint
153
+ - build_cpu:
154
+ name: maximum_version_cpu
155
+ torch: 2.0.0
156
+ torchvision: 0.15.1
157
+ python: 3.9.0
158
+ requires:
159
+ - minimum_version_cpu
160
+ - hold:
161
+ type: approval
162
+ requires:
163
+ - maximum_version_cpu
164
+ - build_cuda:
165
+ name: mainstream_version_gpu
166
+ torch: 1.8.1
167
+ # Use double quotation mark to explicitly specify its type
168
+ # as string instead of number
169
+ cuda: "10.2"
170
+ requires:
171
+ - hold
172
+ - build_cuda:
173
+ name: mainstream_version_gpu
174
+ torch: 2.0.0
175
+ # Use double quotation mark to explicitly specify its type
176
+ # as string instead of number
177
+ cuda: "11.7"
178
+ cudnn: 8
179
+ requires:
180
+ - hold
181
+ merge_stage_test:
182
+ when:
183
+ not:
184
+ << pipeline.parameters.lint_only >>
185
+ jobs:
186
+ - build_cuda:
187
+ name: minimum_version_gpu
188
+ torch: 1.6.0
189
+ # Use double quotation mark to explicitly specify its type
190
+ # as string instead of number
191
+ cuda: "10.1"
192
+ filters:
193
+ branches:
194
+ only:
195
+ - dev-1.x
196
+ - main
pipeline/mmocr/.codespellrc ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ [codespell]
2
+ skip = *.ipynb
3
+ count =
4
+ quiet-level = 3
5
+ ignore-words-list = convertor,convertors,formating,nin,wan,datas,hist,ned
pipeline/mmocr/.coveragerc ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ [run]
2
+ omit =
3
+ */__init__.py
pipeline/mmocr/.dev_scripts/benchmark_full_models.txt ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py
2
+ textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py
3
+ textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py
4
+ textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py
5
+ textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_icdar2015.py
6
+ textdet/panet/panet_resnet18_fpem-ffm_600e_icdar2015.py
7
+ textdet/psenet/psenet_resnet50_fpnf_600e_icdar2015.py
8
+ textdet/textsnake/textsnake_resnet50_fpn-unet_1200e_ctw1500.py
9
+ textrecog/abinet/abinet-vision_20e_st-an_mj.py
10
+ textrecog/crnn/crnn_mini-vgg_5e_mj.py
11
+ textrecog/master/master_resnet31_12e_st_mj_sa.py
12
+ textrecog/nrtr/nrtr_resnet31-1by16-1by8_6e_st_mj.py
13
+ textrecog/robust_scanner/robustscanner_resnet31_5e_st-sub_mj-sub_sa_real.py
14
+ textrecog/sar/sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.py
15
+ textrecog/satrn/satrn_shallow-small_5e_st_mj.py
16
+ textrecog/satrn/satrn_shallow-small_5e_st_mj.py
17
+ textrecog/aster/aster_resnet45_6e_st_mj.py
18
+ textrecog/svtr/svtr-small_20e_st_mj.py
pipeline/mmocr/.dev_scripts/benchmark_options.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ # Copyright (c) OpenMMLab. All rights reserved.
2
+
3
+ third_part_libs = [
4
+ 'pip install -r ../requirements/albu.txt',
5
+ ]
6
+
7
+ default_floating_range = 0.5
pipeline/mmocr/.dev_scripts/benchmark_train_models.txt ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py
2
+ textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py
3
+ textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_icdar2015.py
4
+ textrecog/abinet/abinet-vision_20e_st-an_mj.py
5
+ textrecog/crnn/crnn_mini-vgg_5e_mj.py
6
+ textrecog/aster/aster_resnet45_6e_st_mj.py
7
+ textrecog/nrtr/nrtr_resnet31-1by16-1by8_6e_st_mj.py
8
+ textrecog/sar/sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.py
9
+ textrecog/svtr/svtr-small_20e_st_mj.py
pipeline/mmocr/.dev_scripts/covignore.cfg ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Each line should be the relative path to the root directory
2
+ # of this repo. Support regular expression as well.
3
+ # For example:
4
+ # mmocr/models/textdet/postprocess/utils.py
5
+ # .*/utils.py
6
+ .*/__init__.py
7
+
8
+ # It will be removed after all models have been refactored
9
+ mmocr/utils/bbox_utils.py
10
+
11
+ # Major part is covered, however, it's hard to cover model's output.
12
+ mmocr/models/textdet/detectors/mmdet_wrapper.py
13
+
14
+ # It will be removed after KieVisualizer and TextSpotterVisualizer
15
+ mmocr/visualization/visualize.py
16
+
17
+ # Add tests for data preparers later
18
+ mmocr/datasets/preparers
pipeline/mmocr/.dev_scripts/diff_coverage_test.sh ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ set -e
4
+
5
+ readarray -t IGNORED_FILES < $( dirname "$0" )/covignore.cfg
6
+
7
+ REUSE_COVERAGE_REPORT=${REUSE_COVERAGE_REPORT:-0}
8
+ REPO=${1:-"origin"}
9
+ BRANCH=${2:-"refactor_dev"}
10
+
11
+ git fetch $REPO $BRANCH
12
+
13
+ PY_FILES=""
14
+ for FILE_NAME in $(git diff --name-only ${REPO}/${BRANCH}); do
15
+ # Only test python files in mmocr/ existing in current branch, and not ignored in covignore.cfg
16
+ if [ ${FILE_NAME: -3} == ".py" ] && [ ${FILE_NAME:0:6} == "mmocr/" ] && [ -f "$FILE_NAME" ]; then
17
+ IGNORED=false
18
+ for IGNORED_FILE_NAME in "${IGNORED_FILES[@]}"; do
19
+ # Skip blank lines
20
+ if [ -z "$IGNORED_FILE_NAME" ]; then
21
+ continue
22
+ fi
23
+ if [ "${IGNORED_FILE_NAME::1}" != "#" ] && [[ "$FILE_NAME" =~ $IGNORED_FILE_NAME ]]; then
24
+ echo "Ignoring $FILE_NAME"
25
+ IGNORED=true
26
+ break
27
+ fi
28
+ done
29
+ if [ "$IGNORED" = false ]; then
30
+ PY_FILES="$PY_FILES $FILE_NAME"
31
+ fi
32
+ fi
33
+ done
34
+
35
+ # Only test the coverage when PY_FILES are not empty, otherwise they will test the entire project
36
+ if [ ! -z "${PY_FILES}" ]
37
+ then
38
+ if [ "$REUSE_COVERAGE_REPORT" == "0" ]; then
39
+ coverage run --branch --source mmocr -m pytest tests/
40
+ fi
41
+ coverage report --fail-under 90 -m $PY_FILES
42
+ interrogate -v --ignore-init-method --ignore-module --ignore-nested-functions --ignore-magic --ignore-regex "__repr__" --fail-under 95 $PY_FILES
43
+ fi
pipeline/mmocr/.github/CODE_OF_CONDUCT.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Contributor Covenant Code of Conduct
2
+
3
+ ## Our Pledge
4
+
5
+ In the interest of fostering an open and welcoming environment, we as
6
+ contributors and maintainers pledge to making participation in our project and
7
+ our community a harassment-free experience for everyone, regardless of age, body
8
+ size, disability, ethnicity, sex characteristics, gender identity and expression,
9
+ level of experience, education, socio-economic status, nationality, personal
10
+ appearance, race, religion, or sexual identity and orientation.
11
+
12
+ ## Our Standards
13
+
14
+ Examples of behavior that contributes to creating a positive environment
15
+ include:
16
+
17
+ - Using welcoming and inclusive language
18
+ - Being respectful of differing viewpoints and experiences
19
+ - Gracefully accepting constructive criticism
20
+ - Focusing on what is best for the community
21
+ - Showing empathy towards other community members
22
+
23
+ Examples of unacceptable behavior by participants include:
24
+
25
+ - The use of sexualized language or imagery and unwelcome sexual attention or
26
+ advances
27
+ - Trolling, insulting/derogatory comments, and personal or political attacks
28
+ - Public or private harassment
29
+ - Publishing others' private information, such as a physical or electronic
30
+ address, without explicit permission
31
+ - Other conduct which could reasonably be considered inappropriate in a
32
+ professional setting
33
+
34
+ ## Our Responsibilities
35
+
36
+ Project maintainers are responsible for clarifying the standards of acceptable
37
+ behavior and are expected to take appropriate and fair corrective action in
38
+ response to any instances of unacceptable behavior.
39
+
40
+ Project maintainers have the right and responsibility to remove, edit, or
41
+ reject comments, commits, code, wiki edits, issues, and other contributions
42
+ that are not aligned to this Code of Conduct, or to ban temporarily or
43
+ permanently any contributor for other behaviors that they deem inappropriate,
44
+ threatening, offensive, or harmful.
45
+
46
+ ## Scope
47
+
48
+ This Code of Conduct applies both within project spaces and in public spaces
49
+ when an individual is representing the project or its community. Examples of
50
+ representing a project or community include using an official project e-mail
51
+ address, posting via an official social media account, or acting as an appointed
52
+ representative at an online or offline event. Representation of a project may be
53
+ further defined and clarified by project maintainers.
54
+
55
+ ## Enforcement
56
+
57
+ Instances of abusive, harassing, or otherwise unacceptable behavior may be
58
+ reported by contacting the project team at chenkaidev@gmail.com. All
59
+ complaints will be reviewed and investigated and will result in a response that
60
+ is deemed necessary and appropriate to the circumstances. The project team is
61
+ obligated to maintain confidentiality with regard to the reporter of an incident.
62
+ Further details of specific enforcement policies may be posted separately.
63
+
64
+ Project maintainers who do not follow or enforce the Code of Conduct in good
65
+ faith may face temporary or permanent repercussions as determined by other
66
+ members of the project's leadership.
67
+
68
+ ## Attribution
69
+
70
+ This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4,
71
+ available at https://www.contributor-covenant.org/version/1/4/code-of-conduct.html
72
+
73
+ For answers to common questions about this code of conduct, see
74
+ https://www.contributor-covenant.org/faq
75
+
76
+ [homepage]: https://www.contributor-covenant.org
pipeline/mmocr/.github/CONTRIBUTING.md ADDED
@@ -0,0 +1 @@
 
 
1
+ We appreciate all contributions to improve MMOCR. Please read [Contribution Guide](/docs/en/notes/contribution_guide.md) for step-by-step instructions to make a contribution to MMOCR, and [CONTRIBUTING.md](https://github.com/open-mmlab/mmcv/blob/master/CONTRIBUTING.md) in MMCV for more details about the contributing guideline.
pipeline/mmocr/.github/ISSUE_TEMPLATE/1-bug-report.yml ADDED
@@ -0,0 +1,121 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: "🐞 Bug report"
2
+ description: "Create a report to help us reproduce and fix the bug"
3
+ labels: kind/bug
4
+ title: "[Bug] "
5
+
6
+ body:
7
+ - type: markdown
8
+ attributes:
9
+ value: |
10
+ ## Note
11
+ For general usage questions or idea discussions, please post it to our [**Forum**](https://github.com/open-mmlab/mmocr/discussions)
12
+ If this issue is about installing MMCV, please file an issue at [MMCV](https://github.com/open-mmlab/mmcv/issues/new/choose).
13
+ If it's anything about model deployment, please raise it to [MMDeploy](https://github.com/open-mmlab/mmdeploy)
14
+
15
+ Please fill in as **much** of the following form as you're able to. **The clearer the description, the shorter it will take to solve it.**
16
+
17
+ - type: checkboxes
18
+ attributes:
19
+ label: Prerequisite
20
+ description: Please check the following items before creating a new issue.
21
+ options:
22
+ - label: I have searched [Issues](https://github.com/open-mmlab/mmocr/issues) and [Discussions](https://github.com/open-mmlab/mmocr/discussions) but cannot get the expected help.
23
+ required: true
24
+ # - label: I have read the [FAQ documentation](https://mmocr.readthedocs.io/en/1.x/notes/4_faq.html) but cannot get the expected help.
25
+ # required: true
26
+ - label: The bug has not been fixed in the [latest version (0.x)](https://github.com/open-mmlab/mmocr) or [latest version (1.x)](https://github.com/open-mmlab/mmocr/tree/dev-1.x).
27
+ required: true
28
+
29
+ - type: dropdown
30
+ id: task
31
+ attributes:
32
+ label: Task
33
+ description: The problem arises when
34
+ options:
35
+ - I'm using the official example scripts/configs for the officially supported tasks/models/datasets.
36
+ - I have modified the scripts/configs, or I'm working on my own tasks/models/datasets.
37
+ validations:
38
+ required: true
39
+
40
+ - type: dropdown
41
+ id: branch
42
+ attributes:
43
+ label: Branch
44
+ description: The problem arises when I'm working on
45
+ options:
46
+ - main branch https://github.com/open-mmlab/mmocr
47
+ - 1.x branch https://github.com/open-mmlab/mmocr/tree/dev-1.x
48
+ validations:
49
+ required: true
50
+
51
+ - type: textarea
52
+ attributes:
53
+ label: Environment
54
+ description: |
55
+ Please run `python mmocr/utils/collect_env.py` to collect necessary environment information and copy-paste it here.
56
+ You may add additional information that may be helpful for locating the problem, such as
57
+ - How you installed PyTorch \[e.g., pip, conda, source\]
58
+ - Other environment variables that may be related (such as `$PATH`, `$LD_LIBRARY_PATH`, `$PYTHONPATH`, etc.)
59
+ validations:
60
+ required: true
61
+
62
+ - type: textarea
63
+ attributes:
64
+ label: Reproduces the problem - code sample
65
+ description: |
66
+ Please provide a code sample that reproduces the problem you ran into. It can be a Colab link or just a code snippet.
67
+ placeholder: |
68
+ ```python
69
+ # Sample code to reproduce the problem
70
+ ```
71
+ validations:
72
+ required: true
73
+
74
+ - type: textarea
75
+ attributes:
76
+ label: Reproduces the problem - command or script
77
+ description: |
78
+ What command or script did you run?
79
+ placeholder: |
80
+ ```shell
81
+ The command or script you run.
82
+ ```
83
+ validations:
84
+ required: true
85
+
86
+ - type: textarea
87
+ attributes:
88
+ label: Reproduces the problem - error message
89
+ description: |
90
+ Please provide the error message or logs you got, with the full traceback.
91
+
92
+ Tip: You can attach images or log files by dragging them into the text area..
93
+ placeholder: |
94
+ ```
95
+ The error message or logs you got, with the full traceback.
96
+ ```
97
+ validations:
98
+ required: true
99
+
100
+ - type: textarea
101
+ attributes:
102
+ label: Additional information
103
+ description: |
104
+ Tell us anything else you think we should know.
105
+
106
+ Tip: You can attach images or log files by dragging them into the text area.
107
+ placeholder: |
108
+ 1. What's your expected result?
109
+ 2. What dataset did you use?
110
+ 3. What do you think might be the reason?
111
+
112
+ - type: markdown
113
+ attributes:
114
+ value: |
115
+ ## Acknowledgement
116
+ Thanks for taking the time to fill out this report.
117
+
118
+ If you have already identified the reason, we strongly appreciate you creating a new PR to fix it [**Here**](https://github.com/open-mmlab/mmocr/pulls)!
119
+ Please refer to [**Contribution Guide**](https://mmocr.readthedocs.io/en/dev-1.x/notes/contribution_guide.html) for contributing.
120
+
121
+ Welcome to join our [**Community**](https://mmocr.readthedocs.io/en/latest/contact.html) to discuss together. 👬
pipeline/mmocr/.github/ISSUE_TEMPLATE/2-feature_request.yml ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: 🚀 Feature request
2
+ description: Suggest an idea for this project
3
+ labels: [feature-request]
4
+ title: "[Feature] "
5
+
6
+ body:
7
+ - type: markdown
8
+ attributes:
9
+ value: |
10
+ ## Note
11
+ For general usage questions or idea discussions, please post it to our [**Forum**](https://github.com/open-mmlab/mmocr/discussions)
12
+
13
+ Please fill in as **much** of the following form as you're able to. **The clearer the description, the shorter it will take to solve it.**
14
+
15
+ - type: textarea
16
+ attributes:
17
+ label: What is the feature?
18
+ description: Tell us more about the feature and how this feature can help.
19
+ placeholder: |
20
+ E.g., It is inconvenient when \[....\].
21
+ validations:
22
+ required: true
23
+
24
+ - type: textarea
25
+ attributes:
26
+ label: Any other context?
27
+ description: |
28
+ Have you considered any alternative solutions or features? If so, what are they? Also, feel free to add any other context or screenshots about the feature request here.
29
+
30
+ - type: markdown
31
+ attributes:
32
+ value: |
33
+ ## Acknowledgement
34
+ Thanks for taking the time to fill out this report.
35
+
36
+ We strongly appreciate you creating a new PR to implement it [**Here**](https://github.com/open-mmlab/mmocr/pulls)!
37
+ Please refer to [**Contribution Guide**](https://mmocr.readthedocs.io/en/dev-1.x/notes/contribution_guide.html) for contributing.
38
+
39
+ Welcome to join our [**Community**](https://mmocr.readthedocs.io/en/latest/contact.html) to discuss together. 👬
pipeline/mmocr/.github/ISSUE_TEMPLATE/3-new-model.yml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: "\U0001F31F New model/dataset/scheduler addition"
2
+ description: Submit a proposal/request to implement a new model / dataset / scheduler
3
+ labels: [ "feature-request" ]
4
+ title: "[New Models] "
5
+
6
+
7
+ body:
8
+ - type: markdown
9
+ attributes:
10
+ value: |
11
+ ## Note
12
+ For general usage questions or idea discussions, please post it to our [**Forum**](https://github.com/open-mmlab/mmocr/discussions)
13
+
14
+ Please fill in as **much** of the following form as you're able to. **The clearer the description, the shorter it will take to solve it.**
15
+
16
+ - type: textarea
17
+ id: description-request
18
+ validations:
19
+ required: true
20
+ attributes:
21
+ label: Model/Dataset/Scheduler description
22
+ description: |
23
+ Put any and all important information relative to the model/dataset/scheduler
24
+
25
+ - type: checkboxes
26
+ attributes:
27
+ label: Open source status
28
+ description: |
29
+ Please provide the open-source status, which would be very helpful
30
+ options:
31
+ - label: "The model implementation is available"
32
+ - label: "The model weights are available."
33
+
34
+ - type: textarea
35
+ id: additional-info
36
+ attributes:
37
+ label: Provide useful links for the implementation
38
+ description: |
39
+ Please provide information regarding the implementation, the weights, and the authors.
40
+ Please mention the authors by @gh-username if you're aware of their usernames.
41
+
42
+ - type: markdown
43
+ attributes:
44
+ value: |
45
+ ## Acknowledgement
46
+ Thanks for taking the time to fill out this report.
47
+
48
+ We strongly appreciate you creating a new PR to implement it [**Here**](https://github.com/open-mmlab/mmocr/pulls)!
49
+ Please refer to [**Contribution Guide**](https://mmocr.readthedocs.io/en/dev-1.x/notes/contribution_guide.html) for contributing.
50
+
51
+ Welcome to join our [**Community**](https://mmocr.readthedocs.io/en/latest/contact.html) to discuss together. 👬
pipeline/mmocr/.github/ISSUE_TEMPLATE/4-documentation.yml ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: 📚 Documentation
2
+ description: Report an issue related to the documentation.
3
+ labels: "docs"
4
+ title: "[Docs] "
5
+
6
+ body:
7
+ - type: markdown
8
+ attributes:
9
+ value: |
10
+ ## Note
11
+ For general usage questions or idea discussions, please post it to our [**Forum**](https://github.com/open-mmlab/mmocr/discussions)
12
+ Please fill in as **much** of the following form as you're able to. **The clearer the description, the shorter it will take to solve it.**
13
+
14
+ - type: dropdown
15
+ id: branch
16
+ attributes:
17
+ label: Branch
18
+ description: This issue is related to the
19
+ options:
20
+ - master branch https://mmocr.readthedocs.io/en/latest/
21
+ - 1.x branch https://mmocr.readthedocs.io/en/dev-1.x/
22
+ validations:
23
+ required: true
24
+
25
+ - type: textarea
26
+ attributes:
27
+ label: 📚 The doc issue
28
+ description: >
29
+ A clear and concise description the issue.
30
+ validations:
31
+ required: true
32
+
33
+ - type: textarea
34
+ attributes:
35
+ label: Suggest a potential alternative/fix
36
+ description: >
37
+ Tell us how we could improve the documentation in this regard.
38
+
39
+ - type: markdown
40
+ attributes:
41
+ value: |
42
+ ## Acknowledgement
43
+ Thanks for taking the time to fill out this report.
44
+
45
+ If you have already identified the reason, we strongly appreciate you creating a new PR to fix it [**here**](https://github.com/open-mmlab/mmocr/pulls)!
46
+ Please refer to [**Contribution Guide**](https://mmocr.readthedocs.io/en/dev-1.x/notes/contribution_guide.html) for contributing.
47
+
48
+ Welcome to join our [**Community**](https://mmocr.readthedocs.io/en/latest/contact.html) to discuss together. 👬
pipeline/mmocr/.github/ISSUE_TEMPLATE/config.yml ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ blank_issues_enabled: false
2
+
3
+ contact_links:
4
+ - name: ❔ FAQ
5
+ url: https://mmocr.readthedocs.io/en/dev-1.x/get_started/faq.html
6
+ about: Is your question frequently asked?
7
+ - name: 💬 Forum
8
+ url: https://github.com/open-mmlab/mmocr/discussions
9
+ about: Ask general usage questions and discuss with other MMOCR community members
10
+ - name: 🌐 Explore OpenMMLab
11
+ url: https://openmmlab.com/
12
+ about: Get know more about OpenMMLab
pipeline/mmocr/.github/pull_request_template.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers.
2
+
3
+ ## Motivation
4
+
5
+ Please describe the motivation of this PR and the goal you want to achieve through this PR.
6
+
7
+ ## Modification
8
+
9
+ Please briefly describe what modification is made in this PR.
10
+
11
+ ## BC-breaking (Optional)
12
+
13
+ Does the modification introduce changes that break the backward-compatibility of the downstream repositories?
14
+ If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR.
15
+
16
+ ## Use cases (Optional)
17
+
18
+ If this PR introduces a new feature, it is better to list some use cases here, and update the documentation.
19
+
20
+ ## Checklist
21
+
22
+ **Before PR**:
23
+
24
+ - [ ] I have read and followed the workflow indicated in the [CONTRIBUTING.md](https://github.com/open-mmlab/mmocr/blob/main/.github/CONTRIBUTING.md) to create this PR.
25
+ - [ ] Pre-commit or linting tools indicated in [CONTRIBUTING.md](https://github.com/open-mmlab/mmocr/blob/main/.github/CONTRIBUTING.md) are used to fix the potential lint issues.
26
+ - [ ] Bug fixes are covered by unit tests, the case that causes the bug should be added in the unit tests.
27
+ - [ ] New functionalities are covered by complete unit tests. If not, please add more unit test to ensure the correctness.
28
+ - [ ] The documentation has been modified accordingly, including docstring or example tutorials.
29
+
30
+ **After PR**:
31
+
32
+ - [ ] If the modification has potential influence on downstream or other related projects, this PR should be tested with some of those projects.
33
+ - [ ] CLA has been signed and all committers have signed the CLA in this PR.
pipeline/mmocr/.github/workflows/lint.yml ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: lint
2
+
3
+ on: [push, pull_request]
4
+
5
+ concurrency:
6
+ group: ${{ github.workflow }}-${{ github.ref }}
7
+ cancel-in-progress: true
8
+
9
+ jobs:
10
+ lint:
11
+ runs-on: ubuntu-latest
12
+ steps:
13
+ - uses: actions/checkout@v2
14
+ - name: Set up Python 3.7
15
+ uses: actions/setup-python@v2
16
+ with:
17
+ python-version: 3.7
18
+ - name: Install pre-commit hook
19
+ run: |
20
+ pip install pre-commit
21
+ pre-commit install
22
+ - name: Linting
23
+ run: pre-commit run --all-files
24
+ - name: Check docstring coverage
25
+ run: |
26
+ pip install interrogate
27
+ interrogate -v --ignore-init-method --ignore-module --ignore-nested-functions --ignore-regex "__repr__" --fail-under 90 mmocr
pipeline/mmocr/.github/workflows/merge_stage_test.yml ADDED
@@ -0,0 +1,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: merge_stage_test
2
+
3
+ on:
4
+ push:
5
+ paths-ignore:
6
+ - 'README.md'
7
+ - 'README_zh-CN.md'
8
+ - 'docs/**'
9
+ - 'demo/**'
10
+ - '.dev_scripts/**'
11
+ - '.circleci/**'
12
+ - 'projects/**'
13
+ branches:
14
+ - dev-1.x
15
+
16
+ concurrency:
17
+ group: ${{ github.workflow }}-${{ github.ref }}
18
+ cancel-in-progress: true
19
+
20
+ jobs:
21
+ build_cpu_py:
22
+ runs-on: ubuntu-22.04
23
+ strategy:
24
+ matrix:
25
+ python-version: [3.8, 3.9]
26
+ torch: [1.8.1]
27
+ include:
28
+ - torch: 1.8.1
29
+ torchvision: 0.9.1
30
+ steps:
31
+ - uses: actions/checkout@v3
32
+ - name: Set up Python ${{ matrix.python-version }}
33
+ uses: actions/setup-python@v4
34
+ with:
35
+ python-version: ${{ matrix.python-version }}
36
+ - name: Upgrade pip
37
+ run: pip install pip --upgrade
38
+ - name: Install PyTorch
39
+ run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
40
+ - name: Install MMEngine
41
+ run: pip install git+https://github.com/open-mmlab/mmengine.git@main
42
+ - name: Install MMCV
43
+ run: |
44
+ pip install -U openmim
45
+ mim install 'mmcv >= 2.0.0rc1'
46
+ - name: Install MMDet
47
+ run: pip install git+https://github.com/open-mmlab/mmdetection.git@dev-3.x
48
+ - name: Install other dependencies
49
+ run: pip install -r requirements/tests.txt
50
+ - name: Build and install
51
+ run: rm -rf .eggs && pip install -e .
52
+ - name: Run unittests and generate coverage report
53
+ run: |
54
+ coverage run --branch --source mmocr -m pytest tests/
55
+ coverage xml
56
+ coverage report -m
57
+
58
+ build_cpu_pt:
59
+ runs-on: ubuntu-22.04
60
+ strategy:
61
+ matrix:
62
+ python-version: [3.7]
63
+ torch: [1.6.0, 1.7.1, 1.8.1, 1.9.1, 1.10.1, 1.11.0, 1.12.1, 1.13.0]
64
+ include:
65
+ - torch: 1.6.0
66
+ torchvision: 0.7.0
67
+ - torch: 1.7.1
68
+ torchvision: 0.8.2
69
+ - torch: 1.8.1
70
+ torchvision: 0.9.1
71
+ - torch: 1.9.1
72
+ torchvision: 0.10.1
73
+ - torch: 1.10.1
74
+ torchvision: 0.11.2
75
+ - torch: 1.11.0
76
+ torchvision: 0.12.0
77
+ - torch: 1.12.1
78
+ torchvision: 0.13.1
79
+ - torch: 1.13.0
80
+ torchvision: 0.14.0
81
+ - torch: 2.0.0
82
+ torchvision: 0.15.1
83
+ python-version: 3.8
84
+ steps:
85
+ - uses: actions/checkout@v3
86
+ - name: Set up Python ${{ matrix.python-version }}
87
+ uses: actions/setup-python@v4
88
+ with:
89
+ python-version: ${{ matrix.python-version }}
90
+ - name: Upgrade pip
91
+ run: pip install pip --upgrade
92
+ - name: Install PyTorch
93
+ run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
94
+ - name: Install MMEngine
95
+ run: pip install git+https://github.com/open-mmlab/mmengine.git@main
96
+ - name: Install MMCV
97
+ run: |
98
+ pip install -U openmim
99
+ mim install 'mmcv >= 2.0.0rc1'
100
+ - name: Install MMDet
101
+ run: pip install git+https://github.com/open-mmlab/mmdetection.git@dev-3.x
102
+ - name: Install other dependencies
103
+ run: pip install -r requirements/tests.txt
104
+ - name: Build and install
105
+ run: rm -rf .eggs && pip install -e .
106
+ - name: Run unittests and generate coverage report
107
+ run: |
108
+ coverage run --branch --source mmocr -m pytest tests/
109
+ coverage xml
110
+ coverage report -m
111
+ # Only upload coverage report for python3.7 && pytorch1.8.1 cpu
112
+ - name: Upload coverage to Codecov
113
+ if: ${{matrix.torch == '1.8.1' && matrix.python-version == '3.7'}}
114
+ uses: codecov/codecov-action@v1.0.14
115
+ with:
116
+ file: ./coverage.xml
117
+ flags: unittests
118
+ env_vars: OS,PYTHON
119
+ name: codecov-umbrella
120
+ fail_ci_if_error: false
121
+
122
+
123
+ build_windows:
124
+ runs-on: windows-2022
125
+ strategy:
126
+ matrix:
127
+ python: [3.7]
128
+ platform: [cpu, cu111]
129
+ torch: [1.8.1]
130
+ torchvision: [0.9.1]
131
+ include:
132
+ - python-version: 3.8
133
+ platform: cu117
134
+ torch: 2.0.0
135
+ torchvision: 0.15.1
136
+ steps:
137
+ - uses: actions/checkout@v2
138
+ - name: Set up Python ${{ matrix.python }}
139
+ uses: actions/setup-python@v2
140
+ with:
141
+ python-version: ${{ matrix.python }}
142
+ - name: Upgrade pip
143
+ run: python -m pip install --upgrade pip
144
+ - name: Install lmdb
145
+ run: pip install lmdb
146
+ - name: Install PyTorch
147
+ run: pip install torch==${{matrix.torch}}+${{matrix.platform}} torchvision==${{matrix.torchvision}}+${{matrix.platform}} -f https://download.pytorch.org/whl/${{matrix.platform}}/torch_stable.html
148
+ - name: Install mmocr dependencies
149
+ run: |
150
+ pip install git+https://github.com/open-mmlab/mmengine.git@main
151
+ pip install -U openmim
152
+ mim install 'mmcv >= 2.0.0rc1'
153
+ pip install git+https://github.com/open-mmlab/mmdetection.git@dev-3.x
154
+ pip install -r requirements/tests.txt
155
+ - name: Build and install
156
+ run: |
157
+ pip install -e .
158
+ - name: Run unittests and generate coverage report
159
+ run: |
160
+ pytest tests/
pipeline/mmocr/.github/workflows/pr_stage_test.yml ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: pr_stage_test
2
+
3
+ on:
4
+ pull_request:
5
+ paths-ignore:
6
+ - 'README.md'
7
+ - 'README_zh-CN.md'
8
+ - 'docs/**'
9
+ - 'demo/**'
10
+ - '.dev_scripts/**'
11
+ - '.circleci/**'
12
+ - 'projects/**'
13
+
14
+ concurrency:
15
+ group: ${{ github.workflow }}-${{ github.ref }}
16
+ cancel-in-progress: true
17
+
18
+ jobs:
19
+ build_cpu:
20
+ runs-on: ubuntu-22.04
21
+ strategy:
22
+ matrix:
23
+ python-version: [3.7]
24
+ include:
25
+ - torch: 1.8.1
26
+ torchvision: 0.9.1
27
+ steps:
28
+ - uses: actions/checkout@v3
29
+ - name: Set up Python ${{ matrix.python-version }}
30
+ uses: actions/setup-python@v4
31
+ with:
32
+ python-version: ${{ matrix.python-version }}
33
+ - name: Upgrade pip
34
+ run: pip install pip --upgrade
35
+ - name: Install PyTorch
36
+ run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
37
+ - name: Install MMEngine
38
+ run: pip install git+https://github.com/open-mmlab/mmengine.git@main
39
+ - name: Install MMCV
40
+ run: |
41
+ pip install -U openmim
42
+ mim install 'mmcv >= 2.0.0rc1'
43
+ - name: Install MMDet
44
+ run: pip install git+https://github.com/open-mmlab/mmdetection.git@dev-3.x
45
+ - name: Install other dependencies
46
+ run: pip install -r requirements/tests.txt
47
+ - name: Build and install
48
+ run: rm -rf .eggs && pip install -e .
49
+ - name: Run unittests and generate coverage report
50
+ run: |
51
+ coverage run --branch --source mmocr -m pytest tests/
52
+ coverage xml
53
+ coverage report -m
54
+ # Upload coverage report for python3.7 && pytorch1.8.1 cpu
55
+ - name: Upload coverage to Codecov
56
+ uses: codecov/codecov-action@v1.0.14
57
+ with:
58
+ file: ./coverage.xml
59
+ flags: unittests
60
+ env_vars: OS,PYTHON
61
+ name: codecov-umbrella
62
+ fail_ci_if_error: false
63
+
64
+
65
+ build_windows:
66
+ runs-on: windows-2022
67
+ strategy:
68
+ matrix:
69
+ python: [3.7]
70
+ platform: [cpu, cu111]
71
+ torch: [1.8.1]
72
+ torchvision: [0.9.1]
73
+ include:
74
+ - python-version: 3.8
75
+ platform: cu117
76
+ torch: 2.0.0
77
+ torchvision: 0.15.1
78
+ steps:
79
+ - uses: actions/checkout@v3
80
+ - name: Set up Python ${{ matrix.python }}
81
+ uses: actions/setup-python@v4
82
+ with:
83
+ python-version: ${{ matrix.python }}
84
+ - name: Upgrade pip
85
+ run: python -m pip install --upgrade pip
86
+ - name: Install lmdb
87
+ run: pip install lmdb
88
+ - name: Install PyTorch
89
+ run: pip install torch==${{matrix.torch}}+${{matrix.platform}} torchvision==${{matrix.torchvision}}+${{matrix.platform}} -f https://download.pytorch.org/whl/${{matrix.platform}}/torch_stable.html
90
+ - name: Install mmocr dependencies
91
+ run: |
92
+ pip install git+https://github.com/open-mmlab/mmengine.git@main
93
+ pip install -U openmim
94
+ mim install 'mmcv >= 2.0.0rc1'
95
+ pip install git+https://github.com/open-mmlab/mmdetection.git@dev-3.x
96
+ pip install -r requirements/tests.txt
97
+ - name: Build and install
98
+ run: |
99
+ pip install -e .
100
+ - name: Run unittests and generate coverage report
101
+ run: |
102
+ pytest tests/
pipeline/mmocr/.github/workflows/publish-to-pypi.yml ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: deploy
2
+
3
+ on: push
4
+
5
+ concurrency:
6
+ group: ${{ github.workflow }}-${{ github.ref }}
7
+ cancel-in-progress: true
8
+
9
+ jobs:
10
+ build-n-publish:
11
+ runs-on: ubuntu-latest
12
+ if: startsWith(github.event.ref, 'refs/tags')
13
+ steps:
14
+ - uses: actions/checkout@v2
15
+ - name: Set up Python 3.7
16
+ uses: actions/setup-python@v1
17
+ with:
18
+ python-version: 3.7
19
+ - name: Build MMOCR
20
+ run: |
21
+ pip install wheel
22
+ python setup.py sdist bdist_wheel
23
+ - name: Publish distribution to PyPI
24
+ run: |
25
+ pip install twine
26
+ twine upload dist/* -u __token__ -p ${{ secrets.pypi_password }}
pipeline/mmocr/.github/workflows/test_mim.yml ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: test-mim
2
+
3
+ on:
4
+ push:
5
+ paths:
6
+ - 'model-index.yml'
7
+ - 'configs/**'
8
+
9
+ pull_request:
10
+ paths:
11
+ - 'model-index.yml'
12
+ - 'configs/**'
13
+
14
+ concurrency:
15
+ group: ${{ github.workflow }}-${{ github.ref }}
16
+ cancel-in-progress: true
17
+
18
+ jobs:
19
+ build_cpu:
20
+ runs-on: ubuntu-18.04
21
+ strategy:
22
+ matrix:
23
+ python-version: [3.7]
24
+ torch: [1.8.0]
25
+ include:
26
+ - torch: 1.8.0
27
+ torch_version: torch1.8
28
+ torchvision: 0.9.0
29
+ steps:
30
+ - uses: actions/checkout@v2
31
+ - name: Set up Python ${{ matrix.python-version }}
32
+ uses: actions/setup-python@v2
33
+ with:
34
+ python-version: ${{ matrix.python-version }}
35
+ - name: Upgrade pip
36
+ run: pip install pip --upgrade
37
+ - name: Install PyTorch
38
+ run: pip install torch==${{matrix.torch}}+cpu torchvision==${{matrix.torchvision}}+cpu -f https://download.pytorch.org/whl/torch_stable.html
39
+ - name: Install openmim
40
+ run: pip install openmim
41
+ - name: Build and install
42
+ run: rm -rf .eggs && mim install -e .
43
+ - name: test commands of mim
44
+ run: mim search mmocr
pipeline/mmocr/.gitignore ADDED
@@ -0,0 +1,146 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+ *.ipynb
6
+
7
+ # C extensions
8
+ *.so
9
+
10
+ # Distribution / packaging
11
+ .Python
12
+ build/
13
+ develop-eggs/
14
+ dist/
15
+ downloads/
16
+ eggs/
17
+ .eggs/
18
+ lib/
19
+ lib64/
20
+ parts/
21
+ sdist/
22
+ var/
23
+ wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .coverage
43
+ .coverage.*
44
+ .cache
45
+ nosetests.xml
46
+ coverage.xml
47
+ *.cover
48
+ .hypothesis/
49
+ .pytest_cache/
50
+
51
+ # Translations
52
+ *.mo
53
+ *.pot
54
+
55
+ # Django stuff:
56
+ *.log
57
+ local_settings.py
58
+ db.sqlite3
59
+
60
+ # Flask stuff:
61
+ instance/
62
+ .webassets-cache
63
+
64
+ # Scrapy stuff:
65
+ .scrapy
66
+
67
+ # Sphinx documentation
68
+ docs/en/_build/
69
+ docs/zh_cn/_build/
70
+ docs/*/api/generated/
71
+
72
+ # PyBuilder
73
+ target/
74
+
75
+ # Jupyter Notebook
76
+ .ipynb_checkpoints
77
+
78
+ # pyenv
79
+ .python-version
80
+
81
+ # celery beat schedule file
82
+ celerybeat-schedule
83
+
84
+ # SageMath parsed files
85
+ *.sage.py
86
+
87
+ # Environments
88
+ .env
89
+ .venv
90
+ env/
91
+ venv/
92
+ ENV/
93
+ env.bak/
94
+ venv.bak/
95
+
96
+ # Spyder project settings
97
+ .spyderproject
98
+ .spyproject
99
+
100
+ # Rope project settings
101
+ .ropeproject
102
+
103
+ # mkdocs documentation
104
+ /site
105
+
106
+ # mypy
107
+ .mypy_cache/
108
+
109
+ # cython generated cpp
110
+ !data/dict
111
+ /data
112
+ .vscode
113
+ .idea
114
+
115
+ # custom
116
+ *.pkl
117
+ *.pkl.json
118
+ *.log.json
119
+ work_dirs/
120
+ exps/
121
+ *~
122
+ show_dir/
123
+
124
+ # Pytorch
125
+ *.pth
126
+
127
+ # demo
128
+ !tests/data
129
+ tests/results
130
+
131
+ #temp files
132
+ .DS_Store
133
+
134
+ checkpoints
135
+
136
+ htmlcov
137
+ *.swp
138
+ log.txt
139
+ workspace.code-workspace
140
+ results
141
+ mmocr/core/font.TTF
142
+ mmocr/.mim
143
+ workdirs/
144
+ .history/
145
+ .dev/
146
+ data/
pipeline/mmocr/.owners.yml ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ assign:
2
+ strategy:
3
+ random
4
+ # daily-shift-based
5
+ scedule:
6
+ '*/1 * * * *'
7
+ assignees:
8
+ - gaotongxiao
9
+ - Harold-lkk
pipeline/mmocr/.pre-commit-config.yaml ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exclude: ^tests/data/
2
+ repos:
3
+ - repo: https://github.com/PyCQA/flake8
4
+ rev: 5.0.4
5
+ hooks:
6
+ - id: flake8
7
+ - repo: https://github.com/zhouzaida/isort
8
+ rev: 5.12.1
9
+ hooks:
10
+ - id: isort
11
+ - repo: https://github.com/pre-commit/mirrors-yapf
12
+ rev: v0.32.0
13
+ hooks:
14
+ - id: yapf
15
+ - repo: https://github.com/codespell-project/codespell
16
+ rev: v2.2.1
17
+ hooks:
18
+ - id: codespell
19
+ - repo: https://github.com/pre-commit/pre-commit-hooks
20
+ rev: v4.3.0
21
+ hooks:
22
+ - id: trailing-whitespace
23
+ exclude: |
24
+ (?x)^(
25
+ dicts/|
26
+ projects/.*?/dicts/
27
+ )
28
+ - id: check-yaml
29
+ - id: end-of-file-fixer
30
+ exclude: |
31
+ (?x)^(
32
+ dicts/|
33
+ projects/.*?/dicts/
34
+ )
35
+ - id: requirements-txt-fixer
36
+ - id: double-quote-string-fixer
37
+ - id: check-merge-conflict
38
+ - id: fix-encoding-pragma
39
+ args: ["--remove"]
40
+ - id: mixed-line-ending
41
+ args: ["--fix=lf"]
42
+ - id: mixed-line-ending
43
+ args: ["--fix=lf"]
44
+ - repo: https://github.com/executablebooks/mdformat
45
+ rev: 0.7.9
46
+ hooks:
47
+ - id: mdformat
48
+ args: ["--number", "--table-width", "200"]
49
+ additional_dependencies:
50
+ - mdformat-openmmlab
51
+ - mdformat_frontmatter
52
+ - linkify-it-py
53
+ - repo: https://github.com/myint/docformatter
54
+ rev: v1.3.1
55
+ hooks:
56
+ - id: docformatter
57
+ args: ["--in-place", "--wrap-descriptions", "79"]
58
+ - repo: https://github.com/open-mmlab/pre-commit-hooks
59
+ rev: v0.2.0 # Use the ref you want to point at
60
+ hooks:
61
+ - id: check-algo-readme
62
+ - id: check-copyright
63
+ args: ["mmocr", "tests", "tools"] # these directories will be checked
pipeline/mmocr/.pylintrc ADDED
@@ -0,0 +1,621 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [MASTER]
2
+
3
+ # A comma-separated list of package or module names from where C extensions may
4
+ # be loaded. Extensions are loading into the active Python interpreter and may
5
+ # run arbitrary code.
6
+ extension-pkg-whitelist=
7
+
8
+ # Specify a score threshold to be exceeded before program exits with error.
9
+ fail-under=10.0
10
+
11
+ # Add files or directories to the blacklist. They should be base names, not
12
+ # paths.
13
+ ignore=CVS,configs
14
+
15
+ # Add files or directories matching the regex patterns to the blacklist. The
16
+ # regex matches against base names, not paths.
17
+ ignore-patterns=
18
+
19
+ # Python code to execute, usually for sys.path manipulation such as
20
+ # pygtk.require().
21
+ #init-hook=
22
+
23
+ # Use multiple processes to speed up Pylint. Specifying 0 will auto-detect the
24
+ # number of processors available to use.
25
+ jobs=1
26
+
27
+ # Control the amount of potential inferred values when inferring a single
28
+ # object. This can help the performance when dealing with large functions or
29
+ # complex, nested conditions.
30
+ limit-inference-results=100
31
+
32
+ # List of plugins (as comma separated values of python module names) to load,
33
+ # usually to register additional checkers.
34
+ load-plugins=
35
+
36
+ # Pickle collected data for later comparisons.
37
+ persistent=yes
38
+
39
+ # When enabled, pylint would attempt to guess common misconfiguration and emit
40
+ # user-friendly hints instead of false-positive error messages.
41
+ suggestion-mode=yes
42
+
43
+ # Allow loading of arbitrary C extensions. Extensions are imported into the
44
+ # active Python interpreter and may run arbitrary code.
45
+ unsafe-load-any-extension=no
46
+
47
+
48
+ [MESSAGES CONTROL]
49
+
50
+ # Only show warnings with the listed confidence levels. Leave empty to show
51
+ # all. Valid levels: HIGH, INFERENCE, INFERENCE_FAILURE, UNDEFINED.
52
+ confidence=
53
+
54
+ # Disable the message, report, category or checker with the given id(s). You
55
+ # can either give multiple identifiers separated by comma (,) or put this
56
+ # option multiple times (only on the command line, not in the configuration
57
+ # file where it should appear only once). You can also use "--disable=all" to
58
+ # disable everything first and then reenable specific checks. For example, if
59
+ # you want to run only the similarities checker, you can use "--disable=all
60
+ # --enable=similarities". If you want to run only the classes checker, but have
61
+ # no Warning level messages displayed, use "--disable=all --enable=classes
62
+ # --disable=W".
63
+ disable=print-statement,
64
+ parameter-unpacking,
65
+ unpacking-in-except,
66
+ old-raise-syntax,
67
+ backtick,
68
+ long-suffix,
69
+ old-ne-operator,
70
+ old-octal-literal,
71
+ import-star-module-level,
72
+ non-ascii-bytes-literal,
73
+ raw-checker-failed,
74
+ bad-inline-option,
75
+ locally-disabled,
76
+ file-ignored,
77
+ suppressed-message,
78
+ useless-suppression,
79
+ deprecated-pragma,
80
+ use-symbolic-message-instead,
81
+ apply-builtin,
82
+ basestring-builtin,
83
+ buffer-builtin,
84
+ cmp-builtin,
85
+ coerce-builtin,
86
+ execfile-builtin,
87
+ file-builtin,
88
+ long-builtin,
89
+ raw_input-builtin,
90
+ reduce-builtin,
91
+ standarderror-builtin,
92
+ unicode-builtin,
93
+ xrange-builtin,
94
+ coerce-method,
95
+ delslice-method,
96
+ getslice-method,
97
+ setslice-method,
98
+ no-absolute-import,
99
+ old-division,
100
+ dict-iter-method,
101
+ dict-view-method,
102
+ next-method-called,
103
+ metaclass-assignment,
104
+ indexing-exception,
105
+ raising-string,
106
+ reload-builtin,
107
+ oct-method,
108
+ hex-method,
109
+ nonzero-method,
110
+ cmp-method,
111
+ input-builtin,
112
+ round-builtin,
113
+ intern-builtin,
114
+ unichr-builtin,
115
+ map-builtin-not-iterating,
116
+ zip-builtin-not-iterating,
117
+ range-builtin-not-iterating,
118
+ filter-builtin-not-iterating,
119
+ using-cmp-argument,
120
+ eq-without-hash,
121
+ div-method,
122
+ idiv-method,
123
+ rdiv-method,
124
+ exception-message-attribute,
125
+ invalid-str-codec,
126
+ sys-max-int,
127
+ bad-python3-import,
128
+ deprecated-string-function,
129
+ deprecated-str-translate-call,
130
+ deprecated-itertools-function,
131
+ deprecated-types-field,
132
+ next-method-defined,
133
+ dict-items-not-iterating,
134
+ dict-keys-not-iterating,
135
+ dict-values-not-iterating,
136
+ deprecated-operator-function,
137
+ deprecated-urllib-function,
138
+ xreadlines-attribute,
139
+ deprecated-sys-function,
140
+ exception-escape,
141
+ comprehension-escape,
142
+ no-member,
143
+ invalid-name,
144
+ too-many-branches,
145
+ wrong-import-order,
146
+ too-many-arguments,
147
+ missing-function-docstring,
148
+ missing-module-docstring,
149
+ too-many-locals,
150
+ too-few-public-methods,
151
+ abstract-method,
152
+ broad-except,
153
+ too-many-nested-blocks,
154
+ too-many-instance-attributes,
155
+ missing-class-docstring,
156
+ duplicate-code,
157
+ not-callable,
158
+ protected-access,
159
+ dangerous-default-value,
160
+ no-name-in-module,
161
+ logging-fstring-interpolation,
162
+ super-init-not-called,
163
+ redefined-builtin,
164
+ attribute-defined-outside-init,
165
+ arguments-differ,
166
+ cyclic-import,
167
+ bad-super-call,
168
+ too-many-statements
169
+
170
+ # Enable the message, report, category or checker with the given id(s). You can
171
+ # either give multiple identifier separated by comma (,) or put this option
172
+ # multiple time (only on the command line, not in the configuration file where
173
+ # it should appear only once). See also the "--disable" option for examples.
174
+ enable=c-extension-no-member
175
+
176
+
177
+ [REPORTS]
178
+
179
+ # Python expression which should return a score less than or equal to 10. You
180
+ # have access to the variables 'error', 'warning', 'refactor', and 'convention'
181
+ # which contain the number of messages in each category, as well as 'statement'
182
+ # which is the total number of statements analyzed. This score is used by the
183
+ # global evaluation report (RP0004).
184
+ evaluation=10.0 - ((float(5 * error + warning + refactor + convention) / statement) * 10)
185
+
186
+ # Template used to display messages. This is a python new-style format string
187
+ # used to format the message information. See doc for all details.
188
+ #msg-template=
189
+
190
+ # Set the output format. Available formats are text, parseable, colorized, json
191
+ # and msvs (visual studio). You can also give a reporter class, e.g.
192
+ # mypackage.mymodule.MyReporterClass.
193
+ output-format=text
194
+
195
+ # Tells whether to display a full report or only the messages.
196
+ reports=no
197
+
198
+ # Activate the evaluation score.
199
+ score=yes
200
+
201
+
202
+ [REFACTORING]
203
+
204
+ # Maximum number of nested blocks for function / method body
205
+ max-nested-blocks=5
206
+
207
+ # Complete name of functions that never returns. When checking for
208
+ # inconsistent-return-statements if a never returning function is called then
209
+ # it will be considered as an explicit return statement and no message will be
210
+ # printed.
211
+ never-returning-functions=sys.exit
212
+
213
+
214
+ [TYPECHECK]
215
+
216
+ # List of decorators that produce context managers, such as
217
+ # contextlib.contextmanager. Add to this list to register other decorators that
218
+ # produce valid context managers.
219
+ contextmanager-decorators=contextlib.contextmanager
220
+
221
+ # List of members which are set dynamically and missed by pylint inference
222
+ # system, and so shouldn't trigger E1101 when accessed. Python regular
223
+ # expressions are accepted.
224
+ generated-members=
225
+
226
+ # Tells whether missing members accessed in mixin class should be ignored. A
227
+ # mixin class is detected if its name ends with "mixin" (case insensitive).
228
+ ignore-mixin-members=yes
229
+
230
+ # Tells whether to warn about missing members when the owner of the attribute
231
+ # is inferred to be None.
232
+ ignore-none=yes
233
+
234
+ # This flag controls whether pylint should warn about no-member and similar
235
+ # checks whenever an opaque object is returned when inferring. The inference
236
+ # can return multiple potential results while evaluating a Python object, but
237
+ # some branches might not be evaluated, which results in partial inference. In
238
+ # that case, it might be useful to still emit no-member and other checks for
239
+ # the rest of the inferred objects.
240
+ ignore-on-opaque-inference=yes
241
+
242
+ # List of class names for which member attributes should not be checked (useful
243
+ # for classes with dynamically set attributes). This supports the use of
244
+ # qualified names.
245
+ ignored-classes=optparse.Values,thread._local,_thread._local
246
+
247
+ # List of module names for which member attributes should not be checked
248
+ # (useful for modules/projects where namespaces are manipulated during runtime
249
+ # and thus existing member attributes cannot be deduced by static analysis). It
250
+ # supports qualified module names, as well as Unix pattern matching.
251
+ ignored-modules=
252
+
253
+ # Show a hint with possible names when a member name was not found. The aspect
254
+ # of finding the hint is based on edit distance.
255
+ missing-member-hint=yes
256
+
257
+ # The minimum edit distance a name should have in order to be considered a
258
+ # similar match for a missing member name.
259
+ missing-member-hint-distance=1
260
+
261
+ # The total number of similar names that should be taken in consideration when
262
+ # showing a hint for a missing member.
263
+ missing-member-max-choices=1
264
+
265
+ # List of decorators that change the signature of a decorated function.
266
+ signature-mutators=
267
+
268
+
269
+ [SPELLING]
270
+
271
+ # Limits count of emitted suggestions for spelling mistakes.
272
+ max-spelling-suggestions=4
273
+
274
+ # Spelling dictionary name. Available dictionaries: none. To make it work,
275
+ # install the python-enchant package.
276
+ spelling-dict=
277
+
278
+ # List of comma separated words that should not be checked.
279
+ spelling-ignore-words=
280
+
281
+ # A path to a file that contains the private dictionary; one word per line.
282
+ spelling-private-dict-file=
283
+
284
+ # Tells whether to store unknown words to the private dictionary (see the
285
+ # --spelling-private-dict-file option) instead of raising a message.
286
+ spelling-store-unknown-words=no
287
+
288
+
289
+ [LOGGING]
290
+
291
+ # The type of string formatting that logging methods do. `old` means using %
292
+ # formatting, `new` is for `{}` formatting.
293
+ logging-format-style=old
294
+
295
+ # Logging modules to check that the string format arguments are in logging
296
+ # function parameter format.
297
+ logging-modules=logging
298
+
299
+
300
+ [VARIABLES]
301
+
302
+ # List of additional names supposed to be defined in builtins. Remember that
303
+ # you should avoid defining new builtins when possible.
304
+ additional-builtins=
305
+
306
+ # Tells whether unused global variables should be treated as a violation.
307
+ allow-global-unused-variables=yes
308
+
309
+ # List of strings which can identify a callback function by name. A callback
310
+ # name must start or end with one of those strings.
311
+ callbacks=cb_,
312
+ _cb
313
+
314
+ # A regular expression matching the name of dummy variables (i.e. expected to
315
+ # not be used).
316
+ dummy-variables-rgx=_+$|(_[a-zA-Z0-9_]*[a-zA-Z0-9]+?$)|dummy|^ignored_|^unused_
317
+
318
+ # Argument names that match this expression will be ignored. Default to name
319
+ # with leading underscore.
320
+ ignored-argument-names=_.*|^ignored_|^unused_
321
+
322
+ # Tells whether we should check for unused import in __init__ files.
323
+ init-import=no
324
+
325
+ # List of qualified module names which can have objects that can redefine
326
+ # builtins.
327
+ redefining-builtins-modules=six.moves,past.builtins,future.builtins,builtins,io
328
+
329
+
330
+ [FORMAT]
331
+
332
+ # Expected format of line ending, e.g. empty (any line ending), LF or CRLF.
333
+ expected-line-ending-format=
334
+
335
+ # Regexp for a line that is allowed to be longer than the limit.
336
+ ignore-long-lines=^\s*(# )?<?https?://\S+>?$
337
+
338
+ # Number of spaces of indent required inside a hanging or continued line.
339
+ indent-after-paren=4
340
+
341
+ # String used as indentation unit. This is usually " " (4 spaces) or "\t" (1
342
+ # tab).
343
+ indent-string=' '
344
+
345
+ # Maximum number of characters on a single line.
346
+ max-line-length=100
347
+
348
+ # Maximum number of lines in a module.
349
+ max-module-lines=1000
350
+
351
+ # Allow the body of a class to be on the same line as the declaration if body
352
+ # contains single statement.
353
+ single-line-class-stmt=no
354
+
355
+ # Allow the body of an if to be on the same line as the test if there is no
356
+ # else.
357
+ single-line-if-stmt=no
358
+
359
+
360
+ [STRING]
361
+
362
+ # This flag controls whether inconsistent-quotes generates a warning when the
363
+ # character used as a quote delimiter is used inconsistently within a module.
364
+ check-quote-consistency=no
365
+
366
+ # This flag controls whether the implicit-str-concat should generate a warning
367
+ # on implicit string concatenation in sequences defined over several lines.
368
+ check-str-concat-over-line-jumps=no
369
+
370
+
371
+ [SIMILARITIES]
372
+
373
+ # Ignore comments when computing similarities.
374
+ ignore-comments=yes
375
+
376
+ # Ignore docstrings when computing similarities.
377
+ ignore-docstrings=yes
378
+
379
+ # Ignore imports when computing similarities.
380
+ ignore-imports=no
381
+
382
+ # Minimum lines number of a similarity.
383
+ min-similarity-lines=4
384
+
385
+
386
+ [MISCELLANEOUS]
387
+
388
+ # List of note tags to take in consideration, separated by a comma.
389
+ notes=FIXME,
390
+ XXX,
391
+ TODO
392
+
393
+ # Regular expression of note tags to take in consideration.
394
+ #notes-rgx=
395
+
396
+
397
+ [BASIC]
398
+
399
+ # Naming style matching correct argument names.
400
+ argument-naming-style=snake_case
401
+
402
+ # Regular expression matching correct argument names. Overrides argument-
403
+ # naming-style.
404
+ #argument-rgx=
405
+
406
+ # Naming style matching correct attribute names.
407
+ attr-naming-style=snake_case
408
+
409
+ # Regular expression matching correct attribute names. Overrides attr-naming-
410
+ # style.
411
+ #attr-rgx=
412
+
413
+ # Bad variable names which should always be refused, separated by a comma.
414
+ bad-names=foo,
415
+ bar,
416
+ baz,
417
+ toto,
418
+ tutu,
419
+ tata
420
+
421
+ # Bad variable names regexes, separated by a comma. If names match any regex,
422
+ # they will always be refused
423
+ bad-names-rgxs=
424
+
425
+ # Naming style matching correct class attribute names.
426
+ class-attribute-naming-style=any
427
+
428
+ # Regular expression matching correct class attribute names. Overrides class-
429
+ # attribute-naming-style.
430
+ #class-attribute-rgx=
431
+
432
+ # Naming style matching correct class names.
433
+ class-naming-style=PascalCase
434
+
435
+ # Regular expression matching correct class names. Overrides class-naming-
436
+ # style.
437
+ #class-rgx=
438
+
439
+ # Naming style matching correct constant names.
440
+ const-naming-style=UPPER_CASE
441
+
442
+ # Regular expression matching correct constant names. Overrides const-naming-
443
+ # style.
444
+ #const-rgx=
445
+
446
+ # Minimum line length for functions/classes that require docstrings, shorter
447
+ # ones are exempt.
448
+ docstring-min-length=-1
449
+
450
+ # Naming style matching correct function names.
451
+ function-naming-style=snake_case
452
+
453
+ # Regular expression matching correct function names. Overrides function-
454
+ # naming-style.
455
+ #function-rgx=
456
+
457
+ # Good variable names which should always be accepted, separated by a comma.
458
+ good-names=i,
459
+ j,
460
+ k,
461
+ ex,
462
+ Run,
463
+ _,
464
+ x,
465
+ y,
466
+ w,
467
+ h,
468
+ a,
469
+ b
470
+
471
+ # Good variable names regexes, separated by a comma. If names match any regex,
472
+ # they will always be accepted
473
+ good-names-rgxs=
474
+
475
+ # Include a hint for the correct naming format with invalid-name.
476
+ include-naming-hint=no
477
+
478
+ # Naming style matching correct inline iteration names.
479
+ inlinevar-naming-style=any
480
+
481
+ # Regular expression matching correct inline iteration names. Overrides
482
+ # inlinevar-naming-style.
483
+ #inlinevar-rgx=
484
+
485
+ # Naming style matching correct method names.
486
+ method-naming-style=snake_case
487
+
488
+ # Regular expression matching correct method names. Overrides method-naming-
489
+ # style.
490
+ #method-rgx=
491
+
492
+ # Naming style matching correct module names.
493
+ module-naming-style=snake_case
494
+
495
+ # Regular expression matching correct module names. Overrides module-naming-
496
+ # style.
497
+ #module-rgx=
498
+
499
+ # Colon-delimited sets of names that determine each other's naming style when
500
+ # the name regexes allow several styles.
501
+ name-group=
502
+
503
+ # Regular expression which should only match function or class names that do
504
+ # not require a docstring.
505
+ no-docstring-rgx=^_
506
+
507
+ # List of decorators that produce properties, such as abc.abstractproperty. Add
508
+ # to this list to register other decorators that produce valid properties.
509
+ # These decorators are taken in consideration only for invalid-name.
510
+ property-classes=abc.abstractproperty
511
+
512
+ # Naming style matching correct variable names.
513
+ variable-naming-style=snake_case
514
+
515
+ # Regular expression matching correct variable names. Overrides variable-
516
+ # naming-style.
517
+ #variable-rgx=
518
+
519
+
520
+ [DESIGN]
521
+
522
+ # Maximum number of arguments for function / method.
523
+ max-args=5
524
+
525
+ # Maximum number of attributes for a class (see R0902).
526
+ max-attributes=7
527
+
528
+ # Maximum number of boolean expressions in an if statement (see R0916).
529
+ max-bool-expr=5
530
+
531
+ # Maximum number of branch for function / method body.
532
+ max-branches=12
533
+
534
+ # Maximum number of locals for function / method body.
535
+ max-locals=15
536
+
537
+ # Maximum number of parents for a class (see R0901).
538
+ max-parents=7
539
+
540
+ # Maximum number of public methods for a class (see R0904).
541
+ max-public-methods=20
542
+
543
+ # Maximum number of return / yield for function / method body.
544
+ max-returns=6
545
+
546
+ # Maximum number of statements in function / method body.
547
+ max-statements=50
548
+
549
+ # Minimum number of public methods for a class (see R0903).
550
+ min-public-methods=2
551
+
552
+
553
+ [IMPORTS]
554
+
555
+ # List of modules that can be imported at any level, not just the top level
556
+ # one.
557
+ allow-any-import-level=
558
+
559
+ # Allow wildcard imports from modules that define __all__.
560
+ allow-wildcard-with-all=no
561
+
562
+ # Analyse import fallback blocks. This can be used to support both Python 2 and
563
+ # 3 compatible code, which means that the block might have code that exists
564
+ # only in one or another interpreter, leading to false positives when analysed.
565
+ analyse-fallback-blocks=no
566
+
567
+ # Deprecated modules which should not be used, separated by a comma.
568
+ deprecated-modules=optparse,tkinter.tix
569
+
570
+ # Create a graph of external dependencies in the given file (report RP0402 must
571
+ # not be disabled).
572
+ ext-import-graph=
573
+
574
+ # Create a graph of every (i.e. internal and external) dependencies in the
575
+ # given file (report RP0402 must not be disabled).
576
+ import-graph=
577
+
578
+ # Create a graph of internal dependencies in the given file (report RP0402 must
579
+ # not be disabled).
580
+ int-import-graph=
581
+
582
+ # Force import order to recognize a module as part of the standard
583
+ # compatibility libraries.
584
+ known-standard-library=
585
+
586
+ # Force import order to recognize a module as part of a third party library.
587
+ known-third-party=enchant
588
+
589
+ # Couples of modules and preferred modules, separated by a comma.
590
+ preferred-modules=
591
+
592
+
593
+ [CLASSES]
594
+
595
+ # List of method names used to declare (i.e. assign) instance attributes.
596
+ defining-attr-methods=__init__,
597
+ __new__,
598
+ setUp,
599
+ __post_init__
600
+
601
+ # List of member names, which should be excluded from the protected access
602
+ # warning.
603
+ exclude-protected=_asdict,
604
+ _fields,
605
+ _replace,
606
+ _source,
607
+ _make
608
+
609
+ # List of valid names for the first argument in a class method.
610
+ valid-classmethod-first-arg=cls
611
+
612
+ # List of valid names for the first argument in a metaclass class method.
613
+ valid-metaclass-classmethod-first-arg=cls
614
+
615
+
616
+ [EXCEPTIONS]
617
+
618
+ # Exceptions that will emit a warning when being caught. Defaults to
619
+ # "BaseException, Exception".
620
+ overgeneral-exceptions=BaseException,
621
+ Exception
pipeline/mmocr/.readthedocs.yml ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ version: 2
2
+
3
+ formats: all
4
+
5
+ python:
6
+ version: 3.7
7
+ install:
8
+ - requirements: requirements/docs.txt
9
+ - requirements: requirements/readthedocs.txt
pipeline/mmocr/CITATION.cff ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ cff-version: 1.2.0
2
+ message: "If you use this software, please cite it as below."
3
+ title: "OpenMMLab Text Detection, Recognition and Understanding Toolbox"
4
+ authors:
5
+ - name: "MMOCR Contributors"
6
+ version: 0.3.0
7
+ date-released: 2020-08-15
8
+ repository-code: "https://github.com/open-mmlab/mmocr"
9
+ license: Apache-2.0
pipeline/mmocr/LICENSE ADDED
@@ -0,0 +1,203 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Copyright (c) MMOCR Authors. All rights reserved.
2
+
3
+ Apache License
4
+ Version 2.0, January 2004
5
+ http://www.apache.org/licenses/
6
+
7
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
8
+
9
+ 1. Definitions.
10
+
11
+ "License" shall mean the terms and conditions for use, reproduction,
12
+ and distribution as defined by Sections 1 through 9 of this document.
13
+
14
+ "Licensor" shall mean the copyright owner or entity authorized by
15
+ the copyright owner that is granting the License.
16
+
17
+ "Legal Entity" shall mean the union of the acting entity and all
18
+ other entities that control, are controlled by, or are under common
19
+ control with that entity. For the purposes of this definition,
20
+ "control" means (i) the power, direct or indirect, to cause the
21
+ direction or management of such entity, whether by contract or
22
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
23
+ outstanding shares, or (iii) beneficial ownership of such entity.
24
+
25
+ "You" (or "Your") shall mean an individual or Legal Entity
26
+ exercising permissions granted by this License.
27
+
28
+ "Source" form shall mean the preferred form for making modifications,
29
+ including but not limited to software source code, documentation
30
+ source, and configuration files.
31
+
32
+ "Object" form shall mean any form resulting from mechanical
33
+ transformation or translation of a Source form, including but
34
+ not limited to compiled object code, generated documentation,
35
+ and conversions to other media types.
36
+
37
+ "Work" shall mean the work of authorship, whether in Source or
38
+ Object form, made available under the License, as indicated by a
39
+ copyright notice that is included in or attached to the work
40
+ (an example is provided in the Appendix below).
41
+
42
+ "Derivative Works" shall mean any work, whether in Source or Object
43
+ form, that is based on (or derived from) the Work and for which the
44
+ editorial revisions, annotations, elaborations, or other modifications
45
+ represent, as a whole, an original work of authorship. For the purposes
46
+ of this License, Derivative Works shall not include works that remain
47
+ separable from, or merely link (or bind by name) to the interfaces of,
48
+ the Work and Derivative Works thereof.
49
+
50
+ "Contribution" shall mean any work of authorship, including
51
+ the original version of the Work and any modifications or additions
52
+ to that Work or Derivative Works thereof, that is intentionally
53
+ submitted to Licensor for inclusion in the Work by the copyright owner
54
+ or by an individual or Legal Entity authorized to submit on behalf of
55
+ the copyright owner. For the purposes of this definition, "submitted"
56
+ means any form of electronic, verbal, or written communication sent
57
+ to the Licensor or its representatives, including but not limited to
58
+ communication on electronic mailing lists, source code control systems,
59
+ and issue tracking systems that are managed by, or on behalf of, the
60
+ Licensor for the purpose of discussing and improving the Work, but
61
+ excluding communication that is conspicuously marked or otherwise
62
+ designated in writing by the copyright owner as "Not a Contribution."
63
+
64
+ "Contributor" shall mean Licensor and any individual or Legal Entity
65
+ on behalf of whom a Contribution has been received by Licensor and
66
+ subsequently incorporated within the Work.
67
+
68
+ 2. Grant of Copyright License. Subject to the terms and conditions of
69
+ this License, each Contributor hereby grants to You a perpetual,
70
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
71
+ copyright license to reproduce, prepare Derivative Works of,
72
+ publicly display, publicly perform, sublicense, and distribute the
73
+ Work and such Derivative Works in Source or Object form.
74
+
75
+ 3. Grant of Patent License. Subject to the terms and conditions of
76
+ this License, each Contributor hereby grants to You a perpetual,
77
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
78
+ (except as stated in this section) patent license to make, have made,
79
+ use, offer to sell, sell, import, and otherwise transfer the Work,
80
+ where such license applies only to those patent claims licensable
81
+ by such Contributor that are necessarily infringed by their
82
+ Contribution(s) alone or by combination of their Contribution(s)
83
+ with the Work to which such Contribution(s) was submitted. If You
84
+ institute patent litigation against any entity (including a
85
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
86
+ or a Contribution incorporated within the Work constitutes direct
87
+ or contributory patent infringement, then any patent licenses
88
+ granted to You under this License for that Work shall terminate
89
+ as of the date such litigation is filed.
90
+
91
+ 4. Redistribution. You may reproduce and distribute copies of the
92
+ Work or Derivative Works thereof in any medium, with or without
93
+ modifications, and in Source or Object form, provided that You
94
+ meet the following conditions:
95
+
96
+ (a) You must give any other recipients of the Work or
97
+ Derivative Works a copy of this License; and
98
+
99
+ (b) You must cause any modified files to carry prominent notices
100
+ stating that You changed the files; and
101
+
102
+ (c) You must retain, in the Source form of any Derivative Works
103
+ that You distribute, all copyright, patent, trademark, and
104
+ attribution notices from the Source form of the Work,
105
+ excluding those notices that do not pertain to any part of
106
+ the Derivative Works; and
107
+
108
+ (d) If the Work includes a "NOTICE" text file as part of its
109
+ distribution, then any Derivative Works that You distribute must
110
+ include a readable copy of the attribution notices contained
111
+ within such NOTICE file, excluding those notices that do not
112
+ pertain to any part of the Derivative Works, in at least one
113
+ of the following places: within a NOTICE text file distributed
114
+ as part of the Derivative Works; within the Source form or
115
+ documentation, if provided along with the Derivative Works; or,
116
+ within a display generated by the Derivative Works, if and
117
+ wherever such third-party notices normally appear. The contents
118
+ of the NOTICE file are for informational purposes only and
119
+ do not modify the License. You may add Your own attribution
120
+ notices within Derivative Works that You distribute, alongside
121
+ or as an addendum to the NOTICE text from the Work, provided
122
+ that such additional attribution notices cannot be construed
123
+ as modifying the License.
124
+
125
+ You may add Your own copyright statement to Your modifications and
126
+ may provide additional or different license terms and conditions
127
+ for use, reproduction, or distribution of Your modifications, or
128
+ for any such Derivative Works as a whole, provided Your use,
129
+ reproduction, and distribution of the Work otherwise complies with
130
+ the conditions stated in this License.
131
+
132
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
133
+ any Contribution intentionally submitted for inclusion in the Work
134
+ by You to the Licensor shall be under the terms and conditions of
135
+ this License, without any additional terms or conditions.
136
+ Notwithstanding the above, nothing herein shall supersede or modify
137
+ the terms of any separate license agreement you may have executed
138
+ with Licensor regarding such Contributions.
139
+
140
+ 6. Trademarks. This License does not grant permission to use the trade
141
+ names, trademarks, service marks, or product names of the Licensor,
142
+ except as required for reasonable and customary use in describing the
143
+ origin of the Work and reproducing the content of the NOTICE file.
144
+
145
+ 7. Disclaimer of Warranty. Unless required by applicable law or
146
+ agreed to in writing, Licensor provides the Work (and each
147
+ Contributor provides its Contributions) on an "AS IS" BASIS,
148
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
149
+ implied, including, without limitation, any warranties or conditions
150
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
151
+ PARTICULAR PURPOSE. You are solely responsible for determining the
152
+ appropriateness of using or redistributing the Work and assume any
153
+ risks associated with Your exercise of permissions under this License.
154
+
155
+ 8. Limitation of Liability. In no event and under no legal theory,
156
+ whether in tort (including negligence), contract, or otherwise,
157
+ unless required by applicable law (such as deliberate and grossly
158
+ negligent acts) or agreed to in writing, shall any Contributor be
159
+ liable to You for damages, including any direct, indirect, special,
160
+ incidental, or consequential damages of any character arising as a
161
+ result of this License or out of the use or inability to use the
162
+ Work (including but not limited to damages for loss of goodwill,
163
+ work stoppage, computer failure or malfunction, or any and all
164
+ other commercial damages or losses), even if such Contributor
165
+ has been advised of the possibility of such damages.
166
+
167
+ 9. Accepting Warranty or Additional Liability. While redistributing
168
+ the Work or Derivative Works thereof, You may choose to offer,
169
+ and charge a fee for, acceptance of support, warranty, indemnity,
170
+ or other liability obligations and/or rights consistent with this
171
+ License. However, in accepting such obligations, You may act only
172
+ on Your own behalf and on Your sole responsibility, not on behalf
173
+ of any other Contributor, and only if You agree to indemnify,
174
+ defend, and hold each Contributor harmless for any liability
175
+ incurred by, or claims asserted against, such Contributor by reason
176
+ of your accepting any such warranty or additional liability.
177
+
178
+ END OF TERMS AND CONDITIONS
179
+
180
+ APPENDIX: How to apply the Apache License to your work.
181
+
182
+ To apply the Apache License to your work, attach the following
183
+ boilerplate notice, with the fields enclosed by brackets "[]"
184
+ replaced with your own identifying information. (Don't include
185
+ the brackets!) The text should be enclosed in the appropriate
186
+ comment syntax for the file format. We also recommend that a
187
+ file or class name and description of purpose be included on the
188
+ same "printed page" as the copyright notice for easier
189
+ identification within third-party archives.
190
+
191
+ Copyright 2021 MMOCR Authors. All rights reserved.
192
+
193
+ Licensed under the Apache License, Version 2.0 (the "License");
194
+ you may not use this file except in compliance with the License.
195
+ You may obtain a copy of the License at
196
+
197
+ http://www.apache.org/licenses/LICENSE-2.0
198
+
199
+ Unless required by applicable law or agreed to in writing, software
200
+ distributed under the License is distributed on an "AS IS" BASIS,
201
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
202
+ See the License for the specific language governing permissions and
203
+ limitations under the License.
pipeline/mmocr/MANIFEST.in ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ include requirements/*.txt
2
+ include mmocr/.mim/model-index.yml
3
+ include mmocr/.mim/dicts/*.txt
4
+ recursive-include mmocr/.mim/configs *.py *.yml
5
+ recursive-include mmocr/.mim/tools *.sh *.py
pipeline/mmocr/README.md ADDED
@@ -0,0 +1,251 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <div align="center">
2
+ <img src="resources/mmocr-logo.png" width="500px"/>
3
+ <div>&nbsp;</div>
4
+ <div align="center">
5
+ <b><font size="5">OpenMMLab website</font></b>
6
+ <sup>
7
+ <a href="https://openmmlab.com">
8
+ <i><font size="4">HOT</font></i>
9
+ </a>
10
+ </sup>
11
+ &nbsp;&nbsp;&nbsp;&nbsp;
12
+ <b><font size="5">OpenMMLab platform</font></b>
13
+ <sup>
14
+ <a href="https://platform.openmmlab.com">
15
+ <i><font size="4">TRY IT OUT</font></i>
16
+ </a>
17
+ </sup>
18
+ </div>
19
+ <div>&nbsp;</div>
20
+
21
+ [![build](https://github.com/open-mmlab/mmocr/workflows/build/badge.svg)](https://github.com/open-mmlab/mmocr/actions)
22
+ [![docs](https://readthedocs.org/projects/mmocr/badge/?version=dev-1.x)](https://mmocr.readthedocs.io/en/dev-1.x/?badge=dev-1.x)
23
+ [![codecov](https://codecov.io/gh/open-mmlab/mmocr/branch/main/graph/badge.svg)](https://codecov.io/gh/open-mmlab/mmocr)
24
+ [![license](https://img.shields.io/github/license/open-mmlab/mmocr.svg)](https://github.com/open-mmlab/mmocr/blob/main/LICENSE)
25
+ [![PyPI](https://badge.fury.io/py/mmocr.svg)](https://pypi.org/project/mmocr/)
26
+ [![Average time to resolve an issue](https://isitmaintained.com/badge/resolution/open-mmlab/mmocr.svg)](https://github.com/open-mmlab/mmocr/issues)
27
+ [![Percentage of issues still open](https://isitmaintained.com/badge/open/open-mmlab/mmocr.svg)](https://github.com/open-mmlab/mmocr/issues)
28
+ <a href="https://console.tiyaro.ai/explore?q=mmocr&pub=mmocr"> <img src="https://tiyaro-public-docs.s3.us-west-2.amazonaws.com/assets/try_on_tiyaro_badge.svg"></a>
29
+
30
+ [📘Documentation](https://mmocr.readthedocs.io/en/dev-1.x/) |
31
+ [🛠️Installation](https://mmocr.readthedocs.io/en/dev-1.x/get_started/install.html) |
32
+ [👀Model Zoo](https://mmocr.readthedocs.io/en/dev-1.x/modelzoo.html) |
33
+ [🆕Update News](https://mmocr.readthedocs.io/en/dev-1.x/notes/changelog.html) |
34
+ [🤔Reporting Issues](https://github.com/open-mmlab/mmocr/issues/new/choose)
35
+
36
+ </div>
37
+
38
+ <div align="center">
39
+
40
+ English | [简体中文](README_zh-CN.md)
41
+
42
+ </div>
43
+ <div align="center">
44
+ <a href="https://openmmlab.medium.com/" style="text-decoration:none;">
45
+ <img src="https://user-images.githubusercontent.com/25839884/219255827-67c1a27f-f8c5-46a9-811d-5e57448c61d1.png" width="3%" alt="" /></a>
46
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
47
+ <a href="https://discord.gg/raweFPmdzG" style="text-decoration:none;">
48
+ <img src="https://user-images.githubusercontent.com/25839884/218347213-c080267f-cbb6-443e-8532-8e1ed9a58ea9.png" width="3%" alt="" /></a>
49
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
50
+ <a href="https://twitter.com/OpenMMLab" style="text-decoration:none;">
51
+ <img src="https://user-images.githubusercontent.com/25839884/218346637-d30c8a0f-3eba-4699-8131-512fb06d46db.png" width="3%" alt="" /></a>
52
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
53
+ <a href="https://www.youtube.com/openmmlab" style="text-decoration:none;">
54
+ <img src="https://user-images.githubusercontent.com/25839884/218346691-ceb2116a-465a-40af-8424-9f30d2348ca9.png" width="3%" alt="" /></a>
55
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
56
+ <a href="https://space.bilibili.com/1293512903" style="text-decoration:none;">
57
+ <img src="https://user-images.githubusercontent.com/25839884/219026751-d7d14cce-a7c9-4e82-9942-8375fca65b99.png" width="3%" alt="" /></a>
58
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
59
+ <a href="https://www.zhihu.com/people/openmmlab" style="text-decoration:none;">
60
+ <img src="https://user-images.githubusercontent.com/25839884/219026120-ba71e48b-6e94-4bd4-b4e9-b7d175b5e362.png" width="3%" alt="" /></a>
61
+ </div>
62
+
63
+ ## Latest Updates
64
+
65
+ **The default branch is now `main` and the code on the branch has been upgraded to v1.0.0. The old `main` branch (v0.6.3) code now exists on the `0.x` branch.** If you have been using the `main` branch and encounter upgrade issues, please read the [Migration Guide](https://mmocr.readthedocs.io/en/dev-1.x/migration/overview.html) and notes on [Branches](https://mmocr.readthedocs.io/en/dev-1.x/migration/branches.html) .
66
+
67
+ v1.0.0 was released in 2023-04-06. Major updates from 1.0.0rc6 include:
68
+
69
+ 1. Support for SCUT-CTW1500, SynthText, and MJSynth datasets in Dataset Preparer
70
+ 2. Updated FAQ and documentation
71
+ 3. Deprecation of file_client_args in favor of backend_args
72
+ 4. Added a new MMOCR tutorial notebook
73
+
74
+ To know more about the updates in MMOCR 1.0, please refer to [What's New in MMOCR 1.x](https://mmocr.readthedocs.io/en/dev-1.x/migration/news.html), or
75
+ Read [Changelog](https://mmocr.readthedocs.io/en/dev-1.x/notes/changelog.html) for more details!
76
+
77
+ ## Introduction
78
+
79
+ MMOCR is an open-source toolbox based on PyTorch and mmdetection for text detection, text recognition, and the corresponding downstream tasks including key information extraction. It is part of the [OpenMMLab](https://openmmlab.com/) project.
80
+
81
+ The main branch works with **PyTorch 1.6+**.
82
+
83
+ <div align="center">
84
+ <img src="https://user-images.githubusercontent.com/24622904/187838618-1fdc61c0-2d46-49f9-8502-976ffdf01f28.png"/>
85
+ </div>
86
+
87
+ ### Major Features
88
+
89
+ - **Comprehensive Pipeline**
90
+
91
+ The toolbox supports not only text detection and text recognition, but also their downstream tasks such as key information extraction.
92
+
93
+ - **Multiple Models**
94
+
95
+ The toolbox supports a wide variety of state-of-the-art models for text detection, text recognition and key information extraction.
96
+
97
+ - **Modular Design**
98
+
99
+ The modular design of MMOCR enables users to define their own optimizers, data preprocessors, and model components such as backbones, necks and heads as well as losses. Please refer to [Overview](https://mmocr.readthedocs.io/en/dev-1.x/get_started/overview.html) for how to construct a customized model.
100
+
101
+ - **Numerous Utilities**
102
+
103
+ The toolbox provides a comprehensive set of utilities which can help users assess the performance of models. It includes visualizers which allow visualization of images, ground truths as well as predicted bounding boxes, and a validation tool for evaluating checkpoints during training. It also includes data converters to demonstrate how to convert your own data to the annotation files which the toolbox supports.
104
+
105
+ ## Installation
106
+
107
+ MMOCR depends on [PyTorch](https://pytorch.org/), [MMEngine](https://github.com/open-mmlab/mmengine), [MMCV](https://github.com/open-mmlab/mmcv) and [MMDetection](https://github.com/open-mmlab/mmdetection).
108
+ Below are quick steps for installation.
109
+ Please refer to [Install Guide](https://mmocr.readthedocs.io/en/dev-1.x/get_started/install.html) for more detailed instruction.
110
+
111
+ ```shell
112
+ conda create -n open-mmlab python=3.8 pytorch=1.10 cudatoolkit=11.3 torchvision -c pytorch -y
113
+ conda activate open-mmlab
114
+ pip3 install openmim
115
+ git clone https://github.com/open-mmlab/mmocr.git
116
+ cd mmocr
117
+ mim install -e .
118
+ ```
119
+
120
+ ## Get Started
121
+
122
+ Please see [Quick Run](https://mmocr.readthedocs.io/en/dev-1.x/get_started/quick_run.html) for the basic usage of MMOCR.
123
+
124
+ ## [Model Zoo](https://mmocr.readthedocs.io/en/dev-1.x/modelzoo.html)
125
+
126
+ Supported algorithms:
127
+
128
+ <details open>
129
+ <summary>BackBone</summary>
130
+
131
+ - [x] [oCLIP](configs/backbone/oclip/README.md) (ECCV'2022)
132
+
133
+ </details>
134
+
135
+ <details open>
136
+ <summary>Text Detection</summary>
137
+
138
+ - [x] [DBNet](configs/textdet/dbnet/README.md) (AAAI'2020) / [DBNet++](configs/textdet/dbnetpp/README.md) (TPAMI'2022)
139
+ - [x] [Mask R-CNN](configs/textdet/maskrcnn/README.md) (ICCV'2017)
140
+ - [x] [PANet](configs/textdet/panet/README.md) (ICCV'2019)
141
+ - [x] [PSENet](configs/textdet/psenet/README.md) (CVPR'2019)
142
+ - [x] [TextSnake](configs/textdet/textsnake/README.md) (ECCV'2018)
143
+ - [x] [DRRG](configs/textdet/drrg/README.md) (CVPR'2020)
144
+ - [x] [FCENet](configs/textdet/fcenet/README.md) (CVPR'2021)
145
+
146
+ </details>
147
+
148
+ <details open>
149
+ <summary>Text Recognition</summary>
150
+
151
+ - [x] [ABINet](configs/textrecog/abinet/README.md) (CVPR'2021)
152
+ - [x] [ASTER](configs/textrecog/aster/README.md) (TPAMI'2018)
153
+ - [x] [CRNN](configs/textrecog/crnn/README.md) (TPAMI'2016)
154
+ - [x] [MASTER](configs/textrecog/master/README.md) (PR'2021)
155
+ - [x] [NRTR](configs/textrecog/nrtr/README.md) (ICDAR'2019)
156
+ - [x] [RobustScanner](configs/textrecog/robust_scanner/README.md) (ECCV'2020)
157
+ - [x] [SAR](configs/textrecog/sar/README.md) (AAAI'2019)
158
+ - [x] [SATRN](configs/textrecog/satrn/README.md) (CVPR'2020 Workshop on Text and Documents in the Deep Learning Era)
159
+ - [x] [SVTR](configs/textrecog/svtr/README.md) (IJCAI'2022)
160
+
161
+ </details>
162
+
163
+ <details open>
164
+ <summary>Key Information Extraction</summary>
165
+
166
+ - [x] [SDMG-R](configs/kie/sdmgr/README.md) (ArXiv'2021)
167
+
168
+ </details>
169
+
170
+ <details open>
171
+ <summary>Text Spotting</summary>
172
+
173
+ - [x] [ABCNet](projects/ABCNet/README.md) (CVPR'2020)
174
+ - [x] [ABCNetV2](projects/ABCNet/README_V2.md) (TPAMI'2021)
175
+ - [x] [SPTS](projects/SPTS/README.md) (ACM MM'2022)
176
+
177
+ </details>
178
+
179
+ Please refer to [model_zoo](https://mmocr.readthedocs.io/en/dev-1.x/modelzoo.html) for more details.
180
+
181
+ ## Projects
182
+
183
+ [Here](projects/README.md) are some implementations of SOTA models and solutions built on MMOCR, which are supported and maintained by community users. These projects demonstrate the best practices based on MMOCR for research and product development. We welcome and appreciate all the contributions to OpenMMLab ecosystem.
184
+
185
+ ## Contributing
186
+
187
+ We appreciate all contributions to improve MMOCR. Please refer to [CONTRIBUTING.md](.github/CONTRIBUTING.md) for the contributing guidelines.
188
+
189
+ ## Acknowledgement
190
+
191
+ MMOCR is an open-source project that is contributed by researchers and engineers from various colleges and companies. We appreciate all the contributors who implement their methods or add new features, as well as users who give valuable feedbacks.
192
+ We hope the toolbox and benchmark could serve the growing research community by providing a flexible toolkit to reimplement existing methods and develop their own new OCR methods.
193
+
194
+ ## Citation
195
+
196
+ If you find this project useful in your research, please consider cite:
197
+
198
+ ```bibtex
199
+ @article{mmocr2021,
200
+ title={MMOCR: A Comprehensive Toolbox for Text Detection, Recognition and Understanding},
201
+ author={Kuang, Zhanghui and Sun, Hongbin and Li, Zhizhong and Yue, Xiaoyu and Lin, Tsui Hin and Chen, Jianyong and Wei, Huaqiang and Zhu, Yiqin and Gao, Tong and Zhang, Wenwei and Chen, Kai and Zhang, Wayne and Lin, Dahua},
202
+ journal= {arXiv preprint arXiv:2108.06543},
203
+ year={2021}
204
+ }
205
+ ```
206
+
207
+ ## License
208
+
209
+ This project is released under the [Apache 2.0 license](LICENSE).
210
+
211
+ ## OpenMMLab Family
212
+
213
+ - [MMEngine](https://github.com/open-mmlab/mmengine): OpenMMLab foundational library for training deep learning models
214
+ - [MMCV](https://github.com/open-mmlab/mmcv): OpenMMLab foundational library for computer vision.
215
+ - [MIM](https://github.com/open-mmlab/mim): MIM installs OpenMMLab packages.
216
+ - [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab image classification toolbox and benchmark.
217
+ - [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab detection toolbox and benchmark.
218
+ - [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab's next-generation platform for general 3D object detection.
219
+ - [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab rotated object detection toolbox and benchmark.
220
+ - [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab semantic segmentation toolbox and benchmark.
221
+ - [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab text detection, recognition, and understanding toolbox.
222
+ - [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab pose estimation toolbox and benchmark.
223
+ - [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 3D human parametric model toolbox and benchmark.
224
+ - [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab self-supervised learning toolbox and benchmark.
225
+ - [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab model compression toolbox and benchmark.
226
+ - [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab fewshot learning toolbox and benchmark.
227
+ - [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab's next-generation action understanding toolbox and benchmark.
228
+ - [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab video perception toolbox and benchmark.
229
+ - [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab optical flow toolbox and benchmark.
230
+ - [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab image and video editing toolbox.
231
+ - [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab image and video generative models toolbox.
232
+ - [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab model deployment framework.
233
+
234
+ ## Welcome to the OpenMMLab community
235
+
236
+ Scan the QR code below to follow the OpenMMLab team's [**Zhihu Official Account**](https://www.zhihu.com/people/openmmlab) and join the OpenMMLab team's [**QQ Group**](https://jq.qq.com/?_wv=1027&k=aCvMxdr3), or join the official communication WeChat group by adding the WeChat, or join our [**Slack**](https://join.slack.com/t/mmocrworkspace/shared_invite/zt-1ifqhfla8-yKnLO_aKhVA2h71OrK8GZw)
237
+
238
+ <div align="center">
239
+ <img src="https://raw.githubusercontent.com/open-mmlab/mmcv/master/docs/en/_static/zhihu_qrcode.jpg" height="400" /> <img src="https://raw.githubusercontent.com/open-mmlab/mmcv/master/docs/en/_static/qq_group_qrcode.jpg" height="400" /> <img src="https://raw.githubusercontent.com/open-mmlab/mmcv/master/docs/en/_static/wechat_qrcode.jpg" height="400" />
240
+ </div>
241
+
242
+ We will provide you with the OpenMMLab community
243
+
244
+ - 📢 share the latest core technologies of AI frameworks
245
+ - 💻 Explaining PyTorch common module source Code
246
+ - 📰 News related to the release of OpenMMLab
247
+ - 🚀 Introduction of cutting-edge algorithms developed by OpenMMLab
248
+ 🏃 Get the more efficient answer and feedback
249
+ - 🔥 Provide a platform for communication with developers from all walks of life
250
+
251
+ The OpenMMLab community looks forward to your participation! 👬
pipeline/mmocr/README_zh-CN.md ADDED
@@ -0,0 +1,250 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <div align="center">
2
+ <img src="resources/mmocr-logo.png" width="500px"/>
3
+ <div>&nbsp;</div>
4
+ <div align="center">
5
+ <b><font size="5">OpenMMLab 官网</font></b>
6
+ <sup>
7
+ <a href="https://openmmlab.com">
8
+ <i><font size="4">HOT</font></i>
9
+ </a>
10
+ </sup>
11
+ &nbsp;&nbsp;&nbsp;&nbsp;
12
+ <b><font size="5">OpenMMLab 开放平台</font></b>
13
+ <sup>
14
+ <a href="https://platform.openmmlab.com">
15
+ <i><font size="4">TRY IT OUT</font></i>
16
+ </a>
17
+ </sup>
18
+ </div>
19
+ <div>&nbsp;</div>
20
+
21
+ [![build](https://github.com/open-mmlab/mmocr/workflows/build/badge.svg)](https://github.com/open-mmlab/mmocr/actions)
22
+ [![docs](https://readthedocs.org/projects/mmocr/badge/?version=dev-1.x)](https://mmocr.readthedocs.io/en/dev-1.x/?badge=dev-1.x)
23
+ [![codecov](https://codecov.io/gh/open-mmlab/mmocr/branch/main/graph/badge.svg)](https://codecov.io/gh/open-mmlab/mmocr)
24
+ [![license](https://img.shields.io/github/license/open-mmlab/mmocr.svg)](https://github.com/open-mmlab/mmocr/blob/main/LICENSE)
25
+ [![PyPI](https://badge.fury.io/py/mmocr.svg)](https://pypi.org/project/mmocr/)
26
+ [![Average time to resolve an issue](https://isitmaintained.com/badge/resolution/open-mmlab/mmocr.svg)](https://github.com/open-mmlab/mmocr/issues)
27
+ [![Percentage of issues still open](https://isitmaintained.com/badge/open/open-mmlab/mmocr.svg)](https://github.com/open-mmlab/mmocr/issues)
28
+ <a href="https://console.tiyaro.ai/explore?q=mmocr&pub=mmocr"> <img src="https://tiyaro-public-docs.s3.us-west-2.amazonaws.com/assets/try_on_tiyaro_badge.svg"></a>
29
+
30
+ [📘文档](https://mmocr.readthedocs.io/zh_CN/dev-1.x/) |
31
+ [🛠️安装](https://mmocr.readthedocs.io/zh_CN/dev-1.x/get_started/install.html) |
32
+ [👀模型库](https://mmocr.readthedocs.io/zh_CN/dev-1.x/modelzoo.html) |
33
+ [🆕更新日志](https://mmocr.readthedocs.io/en/dev-1.x/notes/changelog.html) |
34
+ [🤔报告问题](https://github.com/open-mmlab/mmocr/issues/new/choose)
35
+
36
+ </div>
37
+
38
+ <div align="center">
39
+
40
+ [English](/README.md) | 简体中文
41
+
42
+ </div>
43
+
44
+ <div align="center">
45
+ <a href="https://openmmlab.medium.com/" style="text-decoration:none;">
46
+ <img src="https://user-images.githubusercontent.com/25839884/219255827-67c1a27f-f8c5-46a9-811d-5e57448c61d1.png" width="3%" alt="" /></a>
47
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
48
+ <a href="https://discord.gg/raweFPmdzG" style="text-decoration:none;">
49
+ <img src="https://user-images.githubusercontent.com/25839884/218347213-c080267f-cbb6-443e-8532-8e1ed9a58ea9.png" width="3%" alt="" /></a>
50
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
51
+ <a href="https://twitter.com/OpenMMLab" style="text-decoration:none;">
52
+ <img src="https://user-images.githubusercontent.com/25839884/218346637-d30c8a0f-3eba-4699-8131-512fb06d46db.png" width="3%" alt="" /></a>
53
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
54
+ <a href="https://www.youtube.com/openmmlab" style="text-decoration:none;">
55
+ <img src="https://user-images.githubusercontent.com/25839884/218346691-ceb2116a-465a-40af-8424-9f30d2348ca9.png" width="3%" alt="" /></a>
56
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
57
+ <a href="https://space.bilibili.com/1293512903" style="text-decoration:none;">
58
+ <img src="https://user-images.githubusercontent.com/25839884/219026751-d7d14cce-a7c9-4e82-9942-8375fca65b99.png" width="3%" alt="" /></a>
59
+ <img src="https://user-images.githubusercontent.com/25839884/218346358-56cc8e2f-a2b8-487f-9088-32480cceabcf.png" width="3%" alt="" />
60
+ <a href="https://www.zhihu.com/people/openmmlab" style="text-decoration:none;">
61
+ <img src="https://user-images.githubusercontent.com/25839884/219026120-ba71e48b-6e94-4bd4-b4e9-b7d175b5e362.png" width="3%" alt="" /></a>
62
+ </div>
63
+
64
+ ## 近期更新
65
+
66
+ **默认分支目前为 `main`,且分支上的代码已经切换到 v1.0.0 版本。旧版 `main` 分支(v0.6.3)的代码现存在 `0.x` 分支上。** 如果您一直在使用 `main` 分支,并遇到升级问题,请阅读 [迁移指南](https://mmocr.readthedocs.io/zh_CN/dev-1.x/migration/overview.html) 和 [分支说明](https://mmocr.readthedocs.io/zh_CN/dev-1.x/migration/branches.html) 。
67
+
68
+ 最新的版本 v1.0.0 于 2023-04-06 发布。其相对于 1.0.0rc6 的主要更新如下:
69
+
70
+ 1. Dataset Preparer 中支持了 SCUT-CTW1500, SynthText 和 MJSynth 数据集;
71
+ 2. 更新了文档和 FAQ;
72
+ 3. 升级文件后端;使用了 `backend_args` 替换 `file_client_args`;
73
+ 4. 增加了 MMOCR 教程 notebook。
74
+
75
+ 如果需要了解 MMOCR 1.0 相对于 0.x 的升级内容,请阅读 [MMOCR 1.x 更新汇总](https://mmocr.readthedocs.io/zh_CN/dev-1.x/migration/news.html);或者阅读[更新日志](https://mmocr.readthedocs.io/zh_CN/dev-1.x/notes/changelog.html)以获取更多信息。
76
+
77
+ ## 简介
78
+
79
+ MMOCR 是基于 PyTorch 和 mmdetection 的开源工具箱,专注于文本检测,文本识别以及相应的下游任务,如关键信息提取。 它是 OpenMMLab 项目的一部分。
80
+
81
+ 主分支目前支持 **PyTorch 1.6 以上**的版本。
82
+
83
+ <div align="center">
84
+ <img src="https://user-images.githubusercontent.com/24622904/187838618-1fdc61c0-2d46-49f9-8502-976ffdf01f28.png"/>
85
+ </div>
86
+
87
+ ### 主要特性
88
+
89
+ -**全流程**
90
+
91
+ 该工具箱不仅支持文本检测和文本识别,还支持其下游任务,例如关键信息提取。
92
+
93
+ -**多种模型**
94
+
95
+ 该工具箱支持用于文本检测,文本识别和关键信息提取的各种最新模型。
96
+
97
+ -**模块化设计**
98
+
99
+ MMOCR 的模块化设计使用户可以定义自己的优化器,数据预处理器,模型组件如主干模块,颈部模块和头部模块,以及损失函数。有关如何构建自定义模型的信息,请参考[概览](https://mmocr.readthedocs.io/zh_CN/dev-1.x/get_started/overview.html)。
100
+
101
+ -**众多实用工具**
102
+
103
+ 该工具箱提供了一套全面的实用程序,可以帮助用户评估模型的性能。它包括可对图像,标注的真值以及预测结果进行可视化的可视化工具,以及用于在训练过程中评估模型的验证工具。它还包括数据转换器,演示了如何将用户自建的标注数据转换为 MMOCR 支持的标注文件。
104
+
105
+ ## 安装
106
+
107
+ MMOCR 依赖 [PyTorch](https://pytorch.org/), [MMEngine](https://github.com/open-mmlab/mmengine), [MMCV](https://github.com/open-mmlab/mmcv) 和 [MMDetection](https://github.com/open-mmlab/mmdetection),以下是安装的简要步骤。
108
+ 更详细的安装指南请参考 [安装文档](https://mmocr.readthedocs.io/zh_CN/dev-1.x/get_started/install.html)。
109
+
110
+ ```shell
111
+ conda create -n open-mmlab python=3.8 pytorch=1.10 cudatoolkit=11.3 torchvision -c pytorch -y
112
+ conda activate open-mmlab
113
+ pip3 install openmim
114
+ git clone https://github.com/open-mmlab/mmocr.git
115
+ cd mmocr
116
+ mim install -e .
117
+ ```
118
+
119
+ ## 快速入门
120
+
121
+ 请参考[快速入门](https://mmocr.readthedocs.io/zh_CN/dev-1.x/get_started/quick_run.html)文档学习 MMOCR 的基本使用。
122
+
123
+ ## [模型库](https://mmocr.readthedocs.io/zh_CN/dev-1.x/modelzoo.html)
124
+
125
+ 支持的算法:
126
+
127
+ <details open>
128
+ <summary>骨干网络</summary>
129
+
130
+ - [x] [oCLIP](configs/backbone/oclip/README.md) (ECCV'2022)
131
+
132
+ </details>
133
+
134
+ <details open>
135
+ <summary>文字检测</summary>
136
+
137
+ - [x] [DBNet](configs/textdet/dbnet/README.md) (AAAI'2020) / [DBNet++](configs/textdet/dbnetpp/README.md) (TPAMI'2022)
138
+ - [x] [Mask R-CNN](configs/textdet/maskrcnn/README.md) (ICCV'2017)
139
+ - [x] [PANet](configs/textdet/panet/README.md) (ICCV'2019)
140
+ - [x] [PSENet](configs/textdet/psenet/README.md) (CVPR'2019)
141
+ - [x] [TextSnake](configs/textdet/textsnake/README.md) (ECCV'2018)
142
+ - [x] [DRRG](configs/textdet/drrg/README.md) (CVPR'2020)
143
+ - [x] [FCENet](configs/textdet/fcenet/README.md) (CVPR'2021)
144
+
145
+ </details>
146
+
147
+ <details open>
148
+ <summary>文字识别</summary>
149
+
150
+ - [x] [ABINet](configs/textrecog/abinet/README.md) (CVPR'2021)
151
+ - [x] [ASTER](configs/textrecog/aster/README.md) (TPAMI'2018)
152
+ - [x] [CRNN](configs/textrecog/crnn/README.md) (TPAMI'2016)
153
+ - [x] [MASTER](configs/textrecog/master/README.md) (PR'2021)
154
+ - [x] [NRTR](configs/textrecog/nrtr/README.md) (ICDAR'2019)
155
+ - [x] [RobustScanner](configs/textrecog/robust_scanner/README.md) (ECCV'2020)
156
+ - [x] [SAR](configs/textrecog/sar/README.md) (AAAI'2019)
157
+ - [x] [SATRN](configs/textrecog/satrn/README.md) (CVPR'2020 Workshop on Text and Documents in the Deep Learning Era)
158
+ - [x] [SVTR](configs/textrecog/svtr/README.md) (IJCAI'2022)
159
+
160
+ </details>
161
+
162
+ <details open>
163
+ <summary>关键信息提取</summary>
164
+
165
+ - [x] [SDMG-R](configs/kie/sdmgr/README.md) (ArXiv'2021)
166
+
167
+ </details>
168
+
169
+ <details open>
170
+ <summary>端对端 OCR</summary>
171
+
172
+ - [x] [ABCNet](projects/ABCNet/README.md) (CVPR'2020)
173
+ - [x] [ABCNetV2](projects/ABCNet/README_V2.md) (TPAMI'2021)
174
+ - [x] [SPTS](projects/SPTS/README.md) (ACM MM'2022)
175
+
176
+ </details>
177
+
178
+ 请点击[模型库](https://mmocr.readthedocs.io/zh_CN/dev-1.x/modelzoo.html)查看更多关于上述算法的详细信息。
179
+
180
+ ## 社区项目
181
+
182
+ [这里](projects/README.md)有一些由社区用户支持和维护的基于 MMOCR 的 SOTA 模型和解决方案的实现。这些项目展示了基于 MMOCR 的研究和产品开发的最佳实践。
183
+ 我们欢迎并感谢对 OpenMMLab 生态系统的所有贡献。
184
+
185
+ ## 贡献指南
186
+
187
+ 我们感谢所有的贡献者为改进和提升 MMOCR 所作出的努力。请参考[贡献指南](.github/CONTRIBUTING.md)来了解参与项目贡献的相关指引。
188
+
189
+ ## 致谢
190
+
191
+ MMOCR 是一款由来自不同高校和企业的研发人员共同参与贡献的开源项目。我们感谢所有为项目提供算法复现和新功能支持的贡献者,以及提供宝贵反馈的用户。 我们希望此工具箱可以帮助大家来复现已有的方法和开发新的方法,从而为研究社区贡献力量。
192
+
193
+ ## 引用
194
+
195
+ 如果您发现此项目对您的研究有用,请考虑引用:
196
+
197
+ ```bibtex
198
+ @article{mmocr2021,
199
+ title={MMOCR: A Comprehensive Toolbox for Text Detection, Recognition and Understanding},
200
+ author={Kuang, Zhanghui and Sun, Hongbin and Li, Zhizhong and Yue, Xiaoyu and Lin, Tsui Hin and Chen, Jianyong and Wei, Huaqiang and Zhu, Yiqin and Gao, Tong and Zhang, Wenwei and Chen, Kai and Zhang, Wayne and Lin, Dahua},
201
+ journal= {arXiv preprint arXiv:2108.06543},
202
+ year={2021}
203
+ }
204
+ ```
205
+
206
+ ## 开源许可证
207
+
208
+ 该项目采用 [Apache 2.0 license](LICENSE) 开源许可证。
209
+
210
+ ## OpenMMLab 的其他项目
211
+
212
+ - [MMEngine](https://github.com/open-mmlab/mmengine): OpenMMLab 深度学习模型训练基础库
213
+ - [MMCV](https://github.com/open-mmlab/mmcv): OpenMMLab 计算机视觉基础库
214
+ - [MIM](https://github.com/open-mmlab/mim): MIM 是 OpenMMlab 项目、算法、模型的统一入口
215
+ - [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab 图像分类工具箱
216
+ - [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab 目标检测工具箱
217
+ - [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab 新一代通用 3D 目标检测平台
218
+ - [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab 旋转框检测工具箱与测试基准
219
+ - [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab 语义分割工具箱
220
+ - [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab 全流程文字检测识别理解工具箱
221
+ - [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab 姿态估计工具箱
222
+ - [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 人体参数化模型工具箱与测试基准
223
+ - [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab 自监督学习工具箱与测试基准
224
+ - [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab 模型压缩工具箱与测试基准
225
+ - [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab 少样本学习工具箱与测试基准
226
+ - [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab 新一代视频理解工具箱
227
+ - [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab 一体化视频目标感知平台
228
+ - [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab 光流估计工具箱与测试基准
229
+ - [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab 图像视频编辑工具箱
230
+ - [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab 图片视频生成模型工具箱
231
+ - [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab 模型部署框架
232
+
233
+ ## 欢迎加入 OpenMMLab 社区
234
+
235
+ 扫描下方的二维码可关注 OpenMMLab 团队的 [知乎官方账号](https://www.zhihu.com/people/openmmlab),加入 OpenMMLab 团队的 [官方交流 QQ 群](https://r.vansin.top/?r=join-qq),或通过添加微信“Open小喵Lab”加入官方交流微信群。
236
+
237
+ <div align="center">
238
+ <img src="https://raw.githubusercontent.com/open-mmlab/mmcv/master/docs/en/_static/zhihu_qrcode.jpg" height="400" /> <img src="https://cdn.vansin.top/OpenMMLab/q3.png" height="400" /> <img src="https://raw.githubusercontent.com/open-mmlab/mmcv/master/docs/en/_static/wechat_qrcode.jpg" height="400" />
239
+ </div>
240
+
241
+ 我们会在 OpenMMLab 社区为大家
242
+
243
+ - 📢 分享 AI 框架的前沿核心技术
244
+ - 💻 解读 PyTorch 常用模块源码
245
+ - 📰 发布 OpenMMLab 的相关新闻
246
+ - 🚀 介绍 OpenMMLab 开发的前沿算法
247
+ - 🏃 获取更高效的问题答疑和意见反馈
248
+ - 🔥 提供与各行各业开发者充分交流的平台
249
+
250
+ 干货满满 📘,等你来撩 💗,OpenMMLab 社区期待您的加入 👬
pipeline/mmocr/__init__.py ADDED
@@ -0,0 +1 @@
 
 
1
+ from mmocr import *
pipeline/mmocr/configs/backbone/oclip/README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # oCLIP
2
+
3
+ > [Language Matters: A Weakly Supervised Vision-Language Pre-training Approach for Scene Text Detection and Spotting](https://www.ecva.net/papers/eccv_2022/papers_ECCV/papers/136880282.pdf)
4
+
5
+ <!-- [ALGORITHM] -->
6
+
7
+ ## Abstract
8
+
9
+ Recently, Vision-Language Pre-training (VLP) techniques have greatly benefited various vision-language tasks by jointly learning visual and textual representations, which intuitively helps in Optical Character Recognition (OCR) tasks due to the rich visual and textual information in scene text images. However, these methods cannot well cope with OCR tasks because of the difficulty in both instance-level text encoding and image-text pair acquisition (i.e. images and captured texts in them). This paper presents a weakly supervised pre-training method, oCLIP, which can acquire effective scene text representations by jointly learning and aligning visual and textual information. Our network consists of an image encoder and a character-aware text encoder that extract visual and textual features, respectively, as well as a visual-textual decoder that models the interaction among textual and visual features for learning effective scene text representations. With the learning of textual features, the pre-trained model can attend texts in images well with character awareness. Besides, these designs enable the learning from weakly annotated texts (i.e. partial texts in images without text bounding boxes) which mitigates the data annotation constraint greatly. Experiments over the weakly annotated images in ICDAR2019-LSVT show that our pre-trained model improves F-score by +2.5% and +4.8% while transferring its weights to other text detection and spotting networks, respectively. In addition, the proposed method outperforms existing pre-training techniques consistently across multiple public datasets (e.g., +3.2% and +1.3% for Total-Text and CTW1500).
10
+
11
+ <div align=center>
12
+ <img src="https://user-images.githubusercontent.com/24622904/199475057-aa688422-518d-4d7a-86fc-1be0cc1b5dc6.png"/>
13
+ </div>
14
+
15
+ ## Models
16
+
17
+ | Backbone | Pre-train Data | Model |
18
+ | :-------: | :------------: | :-------------------------------------------------------------------------------: |
19
+ | ResNet-50 | SynthText | [Link](https://download.openmmlab.com/mmocr/backbone/resnet50-oclip-7ba0c533.pth) |
20
+
21
+ ```{note}
22
+ The model is converted from the official [oCLIP](https://github.com/bytedance/oclip.git).
23
+ ```
24
+
25
+ ## Supported Text Detection Models
26
+
27
+ | | [DBNet](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#dbnet) | [DBNet++](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#dbnetpp) | [FCENet](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#fcenet) | [TextSnake](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#fcenet) | [PSENet](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#psenet) | [DRRG](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#drrg) | [Mask R-CNN](https://mmocr.readthedocs.io/en/dev-1.x/textdet_models.html#mask-r-cnn) |
28
+ | :-------: | :------------------------------------------------------------------------: | :----------------------------------------------------------------------------: | :--------------------------------------------------------------------------: | :-----------------------------------------------------------------------------: | :--------------------------------------------------------------------------: | :----------------------------------------------------------------------: | :----------------------------------------------------------------------------------: |
29
+ | ICDAR2015 | ✓ | ✓ | ✓ | | ✓ | | ✓ |
30
+ | CTW1500 | | | ✓ | ✓ | ✓ | ✓ | ✓ |
31
+
32
+ ## Citation
33
+
34
+ ```bibtex
35
+ @article{xue2022language,
36
+ title={Language Matters: A Weakly Supervised Vision-Language Pre-training Approach for Scene Text Detection and Spotting},
37
+ author={Xue, Chuhui and Zhang, Wenqing and Hao, Yu and Lu, Shijian and Torr, Philip and Bai, Song},
38
+ journal={Proceedings of the European Conference on Computer Vision (ECCV)},
39
+ year={2022}
40
+ }
41
+ ```
pipeline/mmocr/configs/backbone/oclip/metafile.yml ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Collections:
2
+ - Name: oCLIP
3
+ Metadata:
4
+ Training Data: SynthText
5
+ Architecture:
6
+ - CLIPResNet
7
+ Paper:
8
+ URL: https://arxiv.org/abs/2203.03911
9
+ Title: 'Language Matters: A Weakly Supervised Vision-Language Pre-training Approach for Scene Text Detection and Spotting'
10
+ README: configs/backbone/oclip/README.md
11
+
12
+ Models:
13
+ Weights: https://download.openmmlab.com/mmocr/backbone/resnet50-oclip-7ba0c533.pth
pipeline/mmocr/configs/kie/_base_/datasets/wildreceipt-openset.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ wildreceipt_openset_data_root = 'data/wildreceipt/'
2
+
3
+ wildreceipt_openset_train = dict(
4
+ type='WildReceiptDataset',
5
+ data_root=wildreceipt_openset_data_root,
6
+ metainfo=dict(category=[
7
+ dict(id=0, name='bg'),
8
+ dict(id=1, name='key'),
9
+ dict(id=2, name='value'),
10
+ dict(id=3, name='other')
11
+ ]),
12
+ ann_file='openset_train.txt',
13
+ pipeline=None)
14
+
15
+ wildreceipt_openset_test = dict(
16
+ type='WildReceiptDataset',
17
+ data_root=wildreceipt_openset_data_root,
18
+ metainfo=dict(category=[
19
+ dict(id=0, name='bg'),
20
+ dict(id=1, name='key'),
21
+ dict(id=2, name='value'),
22
+ dict(id=3, name='other')
23
+ ]),
24
+ ann_file='openset_test.txt',
25
+ test_mode=True,
26
+ pipeline=None)
pipeline/mmocr/configs/kie/_base_/datasets/wildreceipt.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ wildreceipt_data_root = 'data/wildreceipt/'
2
+
3
+ wildreceipt_train = dict(
4
+ type='WildReceiptDataset',
5
+ data_root=wildreceipt_data_root,
6
+ metainfo=wildreceipt_data_root + 'class_list.txt',
7
+ ann_file='train.txt',
8
+ pipeline=None)
9
+
10
+ wildreceipt_test = dict(
11
+ type='WildReceiptDataset',
12
+ data_root=wildreceipt_data_root,
13
+ metainfo=wildreceipt_data_root + 'class_list.txt',
14
+ ann_file='test.txt',
15
+ test_mode=True,
16
+ pipeline=None)
pipeline/mmocr/configs/kie/_base_/default_runtime.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ default_scope = 'mmocr'
2
+ env_cfg = dict(
3
+ cudnn_benchmark=False,
4
+ mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0),
5
+ dist_cfg=dict(backend='nccl'),
6
+ )
7
+ randomness = dict(seed=None)
8
+
9
+ default_hooks = dict(
10
+ timer=dict(type='IterTimerHook'),
11
+ logger=dict(type='LoggerHook', interval=100),
12
+ param_scheduler=dict(type='ParamSchedulerHook'),
13
+ checkpoint=dict(type='CheckpointHook', interval=1),
14
+ sampler_seed=dict(type='DistSamplerSeedHook'),
15
+ sync_buffer=dict(type='SyncBuffersHook'),
16
+ visualization=dict(
17
+ type='VisualizationHook',
18
+ interval=1,
19
+ enable=False,
20
+ show=False,
21
+ draw_gt=False,
22
+ draw_pred=False),
23
+ )
24
+
25
+ # Logging
26
+ log_level = 'INFO'
27
+ log_processor = dict(type='LogProcessor', window_size=10, by_epoch=True)
28
+
29
+ load_from = None
30
+ resume = False
31
+
32
+ visualizer = dict(
33
+ type='KIELocalVisualizer', name='visualizer', is_openset=False)