From b8f7ead74cb0200ad5c422e82724ca6b2eb1c543 Mon Sep 17 00:00:00 2001 From: Tong Gao <gaotongxiao@gmail.com> Date: Tue, 17 Aug 2021 17:39:30 +0800 Subject: [PATCH] [Enhancement] Add copyright info (#439) * add copyright info --- demo/batch_image_demo.py | 1 + demo/image_demo.py | 1 + demo/ner_demo.py | 1 + demo/ocr_image_demo.py | 1 + demo/webcam_demo.py | 1 + docs/conf.py | 1 + docs/stats.py | 1 + docs_zh_CN/conf.py | 1 + docs_zh_CN/stats.py | 1 + mmocr/__init__.py | 1 + mmocr/apis/__init__.py | 1 + mmocr/apis/inference.py | 1 + mmocr/apis/train.py | 1 + mmocr/core/__init__.py | 1 + mmocr/core/deployment/__init__.py | 1 + mmocr/core/deployment/deploy_utils.py | 1 + mmocr/core/evaluation/__init__.py | 1 + mmocr/core/evaluation/hmean.py | 1 + mmocr/core/evaluation/hmean_ic13.py | 1 + mmocr/core/evaluation/hmean_iou.py | 1 + mmocr/core/evaluation/kie_metric.py | 1 + mmocr/core/evaluation/ner_metric.py | 1 + mmocr/core/evaluation/ocr_metric.py | 1 + mmocr/core/evaluation/utils.py | 1 + mmocr/core/mask.py | 1 + mmocr/core/visualize.py | 1 + mmocr/datasets/__init__.py | 1 + mmocr/datasets/base_dataset.py | 1 + mmocr/datasets/builder.py | 1 + mmocr/datasets/icdar_dataset.py | 1 + mmocr/datasets/kie_dataset.py | 1 + mmocr/datasets/ner_dataset.py | 1 + mmocr/datasets/ocr_dataset.py | 1 + mmocr/datasets/ocr_seg_dataset.py | 1 + mmocr/datasets/pipelines/__init__.py | 1 + mmocr/datasets/pipelines/box_utils.py | 1 + mmocr/datasets/pipelines/crop.py | 7 ++++++- mmocr/datasets/pipelines/custom_format_bundle.py | 1 + mmocr/datasets/pipelines/dbnet_transforms.py | 1 + mmocr/datasets/pipelines/kie_transforms.py | 1 + mmocr/datasets/pipelines/loading.py | 1 + mmocr/datasets/pipelines/ner_transforms.py | 1 + mmocr/datasets/pipelines/ocr_seg_targets.py | 1 + mmocr/datasets/pipelines/ocr_transforms.py | 1 + mmocr/datasets/pipelines/test_time_aug.py | 1 + mmocr/datasets/pipelines/textdet_targets/__init__.py | 1 + .../pipelines/textdet_targets/base_textdet_targets.py | 1 + mmocr/datasets/pipelines/textdet_targets/dbnet_targets.py | 1 + mmocr/datasets/pipelines/textdet_targets/drrg_targets.py | 1 + mmocr/datasets/pipelines/textdet_targets/fcenet_targets.py | 1 + mmocr/datasets/pipelines/textdet_targets/panet_targets.py | 1 + mmocr/datasets/pipelines/textdet_targets/psenet_targets.py | 1 + .../pipelines/textdet_targets/textsnake_targets.py | 1 + mmocr/datasets/pipelines/transforms.py | 1 + mmocr/datasets/text_det_dataset.py | 1 + mmocr/datasets/uniform_concat_dataset.py | 1 + mmocr/datasets/utils/__init__.py | 1 + mmocr/datasets/utils/loader.py | 1 + mmocr/datasets/utils/parser.py | 1 + mmocr/models/__init__.py | 1 + mmocr/models/builder.py | 1 + mmocr/models/common/__init__.py | 1 + mmocr/models/common/backbones/__init__.py | 1 + mmocr/models/common/backbones/unet.py | 1 + mmocr/models/common/losses/__init__.py | 1 + mmocr/models/common/losses/dice_loss.py | 1 + mmocr/models/common/losses/focal_loss.py | 1 + mmocr/models/kie/__init__.py | 1 + mmocr/models/kie/extractors/__init__.py | 1 + mmocr/models/kie/extractors/sdmgr.py | 1 + mmocr/models/kie/heads/__init__.py | 1 + mmocr/models/kie/heads/sdmgr_head.py | 1 + mmocr/models/kie/losses/__init__.py | 1 + mmocr/models/kie/losses/sdmgr_loss.py | 1 + mmocr/models/ner/__init__.py | 1 + mmocr/models/ner/classifiers/__init__.py | 1 + mmocr/models/ner/classifiers/ner_classifier.py | 1 + mmocr/models/ner/convertors/__init__.py | 1 + mmocr/models/ner/convertors/ner_convertor.py | 1 + mmocr/models/ner/decoders/__init__.py | 1 + mmocr/models/ner/decoders/fc_decoder.py | 1 + mmocr/models/ner/encoders/__init__.py | 1 + mmocr/models/ner/encoders/bert_encoder.py | 1 + mmocr/models/ner/losses/__init__.py | 1 + mmocr/models/ner/losses/masked_cross_entropy_loss.py | 1 + mmocr/models/ner/losses/masked_focal_loss.py | 1 + mmocr/models/ner/utils/__init__.py | 1 + mmocr/models/textdet/__init__.py | 1 + mmocr/models/textdet/dense_heads/__init__.py | 1 + mmocr/models/textdet/dense_heads/db_head.py | 1 + mmocr/models/textdet/dense_heads/drrg_head.py | 1 + mmocr/models/textdet/dense_heads/fce_head.py | 1 + mmocr/models/textdet/dense_heads/head_mixin.py | 1 + mmocr/models/textdet/dense_heads/pan_head.py | 1 + mmocr/models/textdet/dense_heads/pse_head.py | 1 + mmocr/models/textdet/dense_heads/textsnake_head.py | 1 + mmocr/models/textdet/detectors/__init__.py | 1 + mmocr/models/textdet/detectors/dbnet.py | 1 + mmocr/models/textdet/detectors/drrg.py | 1 + mmocr/models/textdet/detectors/fcenet.py | 1 + mmocr/models/textdet/detectors/ocr_mask_rcnn.py | 1 + mmocr/models/textdet/detectors/panet.py | 1 + mmocr/models/textdet/detectors/psenet.py | 1 + .../models/textdet/detectors/single_stage_text_detector.py | 1 + mmocr/models/textdet/detectors/text_detector_mixin.py | 1 + mmocr/models/textdet/detectors/textsnake.py | 1 + mmocr/models/textdet/losses/__init__.py | 1 + mmocr/models/textdet/losses/db_loss.py | 1 + mmocr/models/textdet/losses/drrg_loss.py | 1 + mmocr/models/textdet/losses/fce_loss.py | 1 + mmocr/models/textdet/losses/pan_loss.py | 1 + mmocr/models/textdet/losses/pse_loss.py | 1 + mmocr/models/textdet/losses/textsnake_loss.py | 1 + mmocr/models/textdet/modules/__init__.py | 1 + mmocr/models/textdet/modules/gcn.py | 1 + mmocr/models/textdet/modules/local_graph.py | 1 + mmocr/models/textdet/modules/proposal_local_graph.py | 1 + mmocr/models/textdet/modules/utils.py | 1 + mmocr/models/textdet/necks/__init__.py | 1 + mmocr/models/textdet/necks/fpem_ffm.py | 1 + mmocr/models/textdet/necks/fpn_cat.py | 1 + mmocr/models/textdet/necks/fpn_unet.py | 1 + mmocr/models/textdet/necks/fpnf.py | 1 + mmocr/models/textdet/postprocess/__init__.py | 1 + mmocr/models/textdet/postprocess/wrapper.py | 1 + mmocr/models/textrecog/__init__.py | 1 + mmocr/models/textrecog/backbones/__init__.py | 1 + .../textrecog/backbones/nrtr_modality_transformer.py | 1 + mmocr/models/textrecog/backbones/resnet31_ocr.py | 1 + mmocr/models/textrecog/backbones/very_deep_vgg.py | 1 + mmocr/models/textrecog/convertors/__init__.py | 1 + mmocr/models/textrecog/convertors/attn.py | 1 + mmocr/models/textrecog/convertors/base.py | 1 + mmocr/models/textrecog/convertors/ctc.py | 1 + mmocr/models/textrecog/convertors/seg.py | 1 + mmocr/models/textrecog/decoders/__init__.py | 1 + mmocr/models/textrecog/decoders/base_decoder.py | 1 + mmocr/models/textrecog/decoders/crnn_decoder.py | 1 + .../textrecog/decoders/position_attention_decoder.py | 1 + mmocr/models/textrecog/decoders/robust_scanner_decoder.py | 1 + mmocr/models/textrecog/decoders/sar_decoder.py | 1 + mmocr/models/textrecog/decoders/sar_decoder_with_bs.py | 1 + .../textrecog/decoders/sequence_attention_decoder.py | 1 + mmocr/models/textrecog/decoders/transformer_decoder.py | 1 + mmocr/models/textrecog/encoders/__init__.py | 1 + mmocr/models/textrecog/encoders/base_encoder.py | 1 + .../models/textrecog/encoders/channel_reduction_encoder.py | 1 + mmocr/models/textrecog/encoders/sar_encoder.py | 1 + mmocr/models/textrecog/encoders/transformer_encoder.py | 1 + mmocr/models/textrecog/heads/__init__.py | 1 + mmocr/models/textrecog/heads/seg_head.py | 1 + mmocr/models/textrecog/layers/__init__.py | 1 + mmocr/models/textrecog/layers/conv_layer.py | 1 + .../models/textrecog/layers/dot_product_attention_layer.py | 1 + mmocr/models/textrecog/layers/lstm_layer.py | 1 + mmocr/models/textrecog/layers/position_aware_layer.py | 1 + .../models/textrecog/layers/robust_scanner_fusion_layer.py | 1 + mmocr/models/textrecog/layers/transformer_layer.py | 1 + mmocr/models/textrecog/losses/__init__.py | 1 + mmocr/models/textrecog/losses/ce_loss.py | 1 + mmocr/models/textrecog/losses/ctc_loss.py | 1 + mmocr/models/textrecog/losses/seg_loss.py | 1 + mmocr/models/textrecog/necks/__init__.py | 1 + mmocr/models/textrecog/necks/fpn_ocr.py | 1 + mmocr/models/textrecog/preprocessor/__init__.py | 1 + mmocr/models/textrecog/preprocessor/base_preprocessor.py | 1 + mmocr/models/textrecog/recognizer/__init__.py | 1 + mmocr/models/textrecog/recognizer/base.py | 1 + mmocr/models/textrecog/recognizer/crnn.py | 1 + .../textrecog/recognizer/encode_decode_recognizer.py | 1 + mmocr/models/textrecog/recognizer/nrtr.py | 1 + mmocr/models/textrecog/recognizer/robust_scanner.py | 1 + mmocr/models/textrecog/recognizer/sar.py | 1 + mmocr/models/textrecog/recognizer/seg_recognizer.py | 1 + mmocr/utils/__init__.py | 1 + mmocr/utils/box_util.py | 1 + mmocr/utils/check_argument.py | 1 + mmocr/utils/collect_env.py | 1 + mmocr/utils/data_convert_util.py | 1 + mmocr/utils/fileio.py | 1 + mmocr/utils/img_util.py | 1 + mmocr/utils/lmdb_util.py | 1 + mmocr/utils/logger.py | 1 + mmocr/utils/model.py | 1 + mmocr/utils/ocr.py | 1 + mmocr/utils/string_util.py | 1 + tests/test_apis/test_model_inference.py | 1 + tests/test_core/test_deploy_utils.py | 1 + tests/test_core/test_end2end_vis.py | 1 + tests/test_dataset/test_base_dataset.py | 1 + tests/test_dataset/test_crop.py | 1 + tests/test_dataset/test_dbnet_transforms.py | 1 + tests/test_dataset/test_detect_dataset.py | 1 + tests/test_dataset/test_icdar_dataset.py | 1 + tests/test_dataset/test_kie_dataset.py | 1 + tests/test_dataset/test_loader.py | 1 + tests/test_dataset/test_loading.py | 1 + tests/test_dataset/test_ner_dataset.py | 1 + tests/test_dataset/test_ocr_dataset.py | 1 + tests/test_dataset/test_ocr_seg_dataset.py | 1 + tests/test_dataset/test_ocr_seg_target.py | 1 + tests/test_dataset/test_ocr_transforms.py | 1 + tests/test_dataset/test_parser.py | 1 + tests/test_dataset/test_test_time_aug.py | 1 + tests/test_dataset/test_textdet_targets.py | 1 + tests/test_dataset/test_transforms.py | 1 + tests/test_dataset/test_uniform_concat_dataset.py | 1 + tests/test_metrics/test_eval_utils.py | 1 + tests/test_metrics/test_hmean_detect.py | 1 + tests/test_metrics/test_hmean_ic13.py | 1 + tests/test_metrics/test_hmean_iou.py | 1 + tests/test_models/test_detector.py | 1 + tests/test_models/test_kie_config.py | 1 + .../test_label_convertor/test_attn_label_convertor.py | 1 + .../test_label_convertor/test_ctc_label_convertor.py | 1 + tests/test_models/test_loss.py | 1 + tests/test_models/test_modules.py | 1 + tests/test_models/test_ner_model.py | 1 + tests/test_models/test_ocr_backbone.py | 1 + tests/test_models/test_ocr_decoder.py | 1 + tests/test_models/test_ocr_encoder.py | 1 + tests/test_models/test_ocr_head.py | 1 + tests/test_models/test_ocr_layer.py | 1 + tests/test_models/test_ocr_loss.py | 1 + tests/test_models/test_ocr_neck.py | 1 + tests/test_models/test_ocr_preprocessor.py | 1 + tests/test_models/test_panhead.py | 1 + tests/test_models/test_recog_config.py | 1 + tests/test_models/test_recognizer.py | 1 + tests/test_models/test_targets.py | 1 + tests/test_models/test_textdet_head.py | 1 + tests/test_models/test_textdet_neck.py | 1 + tests/test_tools/test_data_converter.py | 1 + tests/test_utils/test_box.py | 1 + tests/test_utils/test_check_argument.py | 1 + tests/test_utils/test_mask/test_mask_utils.py | 1 + tests/test_utils/test_model.py | 1 + tests/test_utils/test_ocr.py | 1 + tests/test_utils/test_string_util.py | 1 + tests/test_utils/test_text/test_text_utils.py | 1 + tests/test_utils/test_textio.py | 1 + tests/test_utils/test_version_utils.py | 1 + tests/test_utils/test_wrapper.py | 1 + tools/benchmark_processing.py | 1 + tools/data/textdet/coco_to_line_dict.py | 1 + tools/data/textdet/ctw1500_converter.py | 1 + tools/data/textdet/icdar_converter.py | 1 + tools/data/textdet/synthtext_converter.py | 1 + tools/data/textdet/textocr_converter.py | 1 + tools/data/textdet/totaltext_converter.py | 1 + tools/data/textrecog/seg_synthtext_converter.py | 1 + tools/data/textrecog/svt_converter.py | 1 + tools/data/textrecog/synthtext_converter.py | 1 + tools/data/textrecog/textocr_converter.py | 1 + tools/data/textrecog/totaltext_converter.py | 1 + tools/data/utils/txt2lmdb.py | 1 + tools/deployment/deploy_test.py | 1 + tools/deployment/onnx2tensorrt.py | 1 + tools/deployment/pytorch2onnx.py | 1 + tools/det_test_imgs.py | 1 + tools/kie_test_imgs.py | 1 + tools/publish_model.py | 1 + tools/recog_test_imgs.py | 1 + tools/test.py | 1 + tools/train.py | 1 + 265 files changed, 270 insertions(+), 1 deletion(-) diff --git a/demo/batch_image_demo.py b/demo/batch_image_demo.py index 20a37d11..e68c904d 100644 --- a/demo/batch_image_demo.py +++ b/demo/batch_image_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from argparse import ArgumentParser from pathlib import Path diff --git a/demo/image_demo.py b/demo/image_demo.py index 2e6d33f7..e13f2e93 100644 --- a/demo/image_demo.py +++ b/demo/image_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from argparse import ArgumentParser import mmcv diff --git a/demo/ner_demo.py b/demo/ner_demo.py index fb3998b6..2abadf2e 100755 --- a/demo/ner_demo.py +++ b/demo/ner_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from argparse import ArgumentParser from mmdet.apis import init_detector diff --git a/demo/ocr_image_demo.py b/demo/ocr_image_demo.py index 251e00ef..b18f7009 100644 --- a/demo/ocr_image_demo.py +++ b/demo/ocr_image_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from argparse import ArgumentParser import mmcv diff --git a/demo/webcam_demo.py b/demo/webcam_demo.py index f4c7cf2a..0d5a9f3a 100644 --- a/demo/webcam_demo.py +++ b/demo/webcam_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import cv2 diff --git a/docs/conf.py b/docs/conf.py index 2674095d..7df69489 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. # Configuration file for the Sphinx documentation builder. # # This file only contains a selection of the most common options. For a full diff --git a/docs/stats.py b/docs/stats.py index 8cf9c8db..a44b71a8 100755 --- a/docs/stats.py +++ b/docs/stats.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import functools as func import glob import re diff --git a/docs_zh_CN/conf.py b/docs_zh_CN/conf.py index 0ca65846..0583ebec 100644 --- a/docs_zh_CN/conf.py +++ b/docs_zh_CN/conf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. # Configuration file for the Sphinx documentation builder. # # This file only contains a selection of the most common options. For a full diff --git a/docs_zh_CN/stats.py b/docs_zh_CN/stats.py index 034aeb1b..6b716ba5 100755 --- a/docs_zh_CN/stats.py +++ b/docs_zh_CN/stats.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import functools as func import glob import re diff --git a/mmocr/__init__.py b/mmocr/__init__.py index 34936e1b..5b4511e0 100644 --- a/mmocr/__init__.py +++ b/mmocr/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import mmcv diff --git a/mmocr/apis/__init__.py b/mmocr/apis/__init__.py index 955b170b..02969203 100644 --- a/mmocr/apis/__init__.py +++ b/mmocr/apis/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .inference import model_inference from .train import train_detector diff --git a/mmocr/apis/inference.py b/mmocr/apis/inference.py index be979343..ced9db71 100644 --- a/mmocr/apis/inference.py +++ b/mmocr/apis/inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch from mmcv.ops import RoIPool diff --git a/mmocr/apis/train.py b/mmocr/apis/train.py index ecebeac4..e17a0143 100644 --- a/mmocr/apis/train.py +++ b/mmocr/apis/train.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch diff --git a/mmocr/core/__init__.py b/mmocr/core/__init__.py index 544daaea..53d36431 100644 --- a/mmocr/core/__init__.py +++ b/mmocr/core/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from . import evaluation from .mask import extract_boundary, points2boundary, seg2boundary from .visualize import (det_recog_show_result, imshow_edge_node, diff --git a/mmocr/core/deployment/__init__.py b/mmocr/core/deployment/__init__.py index 643d20a8..1754028f 100644 --- a/mmocr/core/deployment/__init__.py +++ b/mmocr/core/deployment/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .deploy_utils import (ONNXRuntimeDetector, ONNXRuntimeRecognizer, TensorRTDetector, TensorRTRecognizer) diff --git a/mmocr/core/deployment/deploy_utils.py b/mmocr/core/deployment/deploy_utils.py index 9650966d..7142c8ac 100644 --- a/mmocr/core/deployment/deploy_utils.py +++ b/mmocr/core/deployment/deploy_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import warnings from typing import Any, Iterable diff --git a/mmocr/core/evaluation/__init__.py b/mmocr/core/evaluation/__init__.py index f171ef18..ab18b39d 100644 --- a/mmocr/core/evaluation/__init__.py +++ b/mmocr/core/evaluation/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .hmean import eval_hmean from .hmean_ic13 import eval_hmean_ic13 from .hmean_iou import eval_hmean_iou diff --git a/mmocr/core/evaluation/hmean.py b/mmocr/core/evaluation/hmean.py index 30d66553..c46331ec 100644 --- a/mmocr/core/evaluation/hmean.py +++ b/mmocr/core/evaluation/hmean.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from operator import itemgetter import mmcv diff --git a/mmocr/core/evaluation/hmean_ic13.py b/mmocr/core/evaluation/hmean_ic13.py index d3c69467..1b4a59bb 100644 --- a/mmocr/core/evaluation/hmean_ic13.py +++ b/mmocr/core/evaluation/hmean_ic13.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import mmocr.utils as utils diff --git a/mmocr/core/evaluation/hmean_iou.py b/mmocr/core/evaluation/hmean_iou.py index 8ad0363f..318c9e6e 100644 --- a/mmocr/core/evaluation/hmean_iou.py +++ b/mmocr/core/evaluation/hmean_iou.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import mmocr.utils as utils diff --git a/mmocr/core/evaluation/kie_metric.py b/mmocr/core/evaluation/kie_metric.py index 00dc2387..2ba695b5 100644 --- a/mmocr/core/evaluation/kie_metric.py +++ b/mmocr/core/evaluation/kie_metric.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch diff --git a/mmocr/core/evaluation/ner_metric.py b/mmocr/core/evaluation/ner_metric.py index 4a3c3a07..d67c2f51 100644 --- a/mmocr/core/evaluation/ner_metric.py +++ b/mmocr/core/evaluation/ner_metric.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections import Counter diff --git a/mmocr/core/evaluation/ocr_metric.py b/mmocr/core/evaluation/ocr_metric.py index 04c59482..175bbfb7 100644 --- a/mmocr/core/evaluation/ocr_metric.py +++ b/mmocr/core/evaluation/ocr_metric.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import re from difflib import SequenceMatcher diff --git a/mmocr/core/evaluation/utils.py b/mmocr/core/evaluation/utils.py index d7c90aab..36de7d0c 100644 --- a/mmocr/core/evaluation/utils.py +++ b/mmocr/core/evaluation/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import Polygon as plg diff --git a/mmocr/core/mask.py b/mmocr/core/mask.py index c9b46d19..fd4689b8 100644 --- a/mmocr/core/mask.py +++ b/mmocr/core/mask.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np diff --git a/mmocr/core/visualize.py b/mmocr/core/visualize.py index 115ef0d3..5746f201 100644 --- a/mmocr/core/visualize.py +++ b/mmocr/core/visualize.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import os import shutil diff --git a/mmocr/datasets/__init__.py b/mmocr/datasets/__init__.py index 886d098e..0906ce52 100644 --- a/mmocr/datasets/__init__.py +++ b/mmocr/datasets/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.datasets.builder import DATASETS, build_dataloader, build_dataset from . import utils diff --git a/mmocr/datasets/base_dataset.py b/mmocr/datasets/base_dataset.py index df07f26b..5dc54e46 100644 --- a/mmocr/datasets/base_dataset.py +++ b/mmocr/datasets/base_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmcv.utils import print_log from mmdet.datasets.builder import DATASETS diff --git a/mmocr/datasets/builder.py b/mmocr/datasets/builder.py index e7bcf423..1e4cc66e 100644 --- a/mmocr/datasets/builder.py +++ b/mmocr/datasets/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import Registry, build_from_cfg LOADERS = Registry('loader') diff --git a/mmocr/datasets/icdar_dataset.py b/mmocr/datasets/icdar_dataset.py index bdca0638..71647475 100644 --- a/mmocr/datasets/icdar_dataset.py +++ b/mmocr/datasets/icdar_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmdet.datasets.api_wrappers import COCO from mmdet.datasets.builder import DATASETS diff --git a/mmocr/datasets/kie_dataset.py b/mmocr/datasets/kie_dataset.py index 566e4eaa..6741e5ad 100644 --- a/mmocr/datasets/kie_dataset.py +++ b/mmocr/datasets/kie_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import warnings from os import path as osp diff --git a/mmocr/datasets/ner_dataset.py b/mmocr/datasets/ner_dataset.py index 7e0731fa..923942c3 100644 --- a/mmocr/datasets/ner_dataset.py +++ b/mmocr/datasets/ner_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.datasets.builder import DATASETS from mmocr.core.evaluation.ner_metric import eval_ner_f1 diff --git a/mmocr/datasets/ocr_dataset.py b/mmocr/datasets/ocr_dataset.py index 984fdf58..b24d15d6 100644 --- a/mmocr/datasets/ocr_dataset.py +++ b/mmocr/datasets/ocr_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.datasets.builder import DATASETS from mmocr.core.evaluation.ocr_metric import eval_ocr_metric diff --git a/mmocr/datasets/ocr_seg_dataset.py b/mmocr/datasets/ocr_seg_dataset.py index 2f985dc6..cd4b727d 100644 --- a/mmocr/datasets/ocr_seg_dataset.py +++ b/mmocr/datasets/ocr_seg_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.datasets.builder import DATASETS import mmocr.utils as utils diff --git a/mmocr/datasets/pipelines/__init__.py b/mmocr/datasets/pipelines/__init__.py index 76ce286d..715fb740 100644 --- a/mmocr/datasets/pipelines/__init__.py +++ b/mmocr/datasets/pipelines/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .box_utils import sort_vertex, sort_vertex8 from .custom_format_bundle import CustomFormatBundle from .dbnet_transforms import EastRandomCrop, ImgAug diff --git a/mmocr/datasets/pipelines/box_utils.py b/mmocr/datasets/pipelines/box_utils.py index 3ebc230e..5b5c27d8 100644 --- a/mmocr/datasets/pipelines/box_utils.py +++ b/mmocr/datasets/pipelines/box_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import mmocr.utils as utils diff --git a/mmocr/datasets/pipelines/crop.py b/mmocr/datasets/pipelines/crop.py index 90127578..5fea3b62 100644 --- a/mmocr/datasets/pipelines/crop.py +++ b/mmocr/datasets/pipelines/crop.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np from shapely.geometry import LineString, Point @@ -83,7 +84,11 @@ def warp_img(src_img, return dst_img -def crop_img(src_img, box, long_edge_pad_ratio=0.4, short_edge_pad_ratio=0.2): +def crop_img(src_img, + box, + long_edge_pad_ratio=0.4, + short_edge_pad_ratio=0.2, + debug=False): """Crop text region with their bounding box. Args: diff --git a/mmocr/datasets/pipelines/custom_format_bundle.py b/mmocr/datasets/pipelines/custom_format_bundle.py index dcfca2f8..fc63fa8d 100644 --- a/mmocr/datasets/pipelines/custom_format_bundle.py +++ b/mmocr/datasets/pipelines/custom_format_bundle.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmcv.parallel import DataContainer as DC from mmdet.datasets.builder import PIPELINES diff --git a/mmocr/datasets/pipelines/dbnet_transforms.py b/mmocr/datasets/pipelines/dbnet_transforms.py index 82ede73f..8494cdd6 100644 --- a/mmocr/datasets/pipelines/dbnet_transforms.py +++ b/mmocr/datasets/pipelines/dbnet_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import imgaug import imgaug.augmenters as iaa import mmcv diff --git a/mmocr/datasets/pipelines/kie_transforms.py b/mmocr/datasets/pipelines/kie_transforms.py index 96cfbd4a..2cdff10c 100644 --- a/mmocr/datasets/pipelines/kie_transforms.py +++ b/mmocr/datasets/pipelines/kie_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmcv import rescale_size from mmcv.parallel import DataContainer as DC diff --git a/mmocr/datasets/pipelines/loading.py b/mmocr/datasets/pipelines/loading.py index 05768a1d..3694fc63 100644 --- a/mmocr/datasets/pipelines/loading.py +++ b/mmocr/datasets/pipelines/loading.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import numpy as np from mmdet.core import BitmapMasks, PolygonMasks diff --git a/mmocr/datasets/pipelines/ner_transforms.py b/mmocr/datasets/pipelines/ner_transforms.py index 32f06b7a..b26fe74b 100644 --- a/mmocr/datasets/pipelines/ner_transforms.py +++ b/mmocr/datasets/pipelines/ner_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmdet.datasets.builder import PIPELINES diff --git a/mmocr/datasets/pipelines/ocr_seg_targets.py b/mmocr/datasets/pipelines/ocr_seg_targets.py index 10fa394c..871baed9 100644 --- a/mmocr/datasets/pipelines/ocr_seg_targets.py +++ b/mmocr/datasets/pipelines/ocr_seg_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np from mmdet.core import BitmapMasks diff --git a/mmocr/datasets/pipelines/ocr_transforms.py b/mmocr/datasets/pipelines/ocr_transforms.py index de3d2648..1bf0490b 100644 --- a/mmocr/datasets/pipelines/ocr_transforms.py +++ b/mmocr/datasets/pipelines/ocr_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import mmcv diff --git a/mmocr/datasets/pipelines/test_time_aug.py b/mmocr/datasets/pipelines/test_time_aug.py index 79dca554..773ea14b 100644 --- a/mmocr/datasets/pipelines/test_time_aug.py +++ b/mmocr/datasets/pipelines/test_time_aug.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import numpy as np from mmdet.datasets.builder import PIPELINES diff --git a/mmocr/datasets/pipelines/textdet_targets/__init__.py b/mmocr/datasets/pipelines/textdet_targets/__init__.py index 9cbd494b..2662739a 100644 --- a/mmocr/datasets/pipelines/textdet_targets/__init__.py +++ b/mmocr/datasets/pipelines/textdet_targets/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_textdet_targets import BaseTextDetTargets from .dbnet_targets import DBNetTargets from .drrg_targets import DRRGTargets diff --git a/mmocr/datasets/pipelines/textdet_targets/base_textdet_targets.py b/mmocr/datasets/pipelines/textdet_targets/base_textdet_targets.py index 183743f8..ebb70694 100644 --- a/mmocr/datasets/pipelines/textdet_targets/base_textdet_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/base_textdet_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import sys import cv2 diff --git a/mmocr/datasets/pipelines/textdet_targets/dbnet_targets.py b/mmocr/datasets/pipelines/textdet_targets/dbnet_targets.py index bfe4bf37..d0198692 100644 --- a/mmocr/datasets/pipelines/textdet_targets/dbnet_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/dbnet_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np import pyclipper diff --git a/mmocr/datasets/pipelines/textdet_targets/drrg_targets.py b/mmocr/datasets/pipelines/textdet_targets/drrg_targets.py index c4255ee7..fdf3a494 100644 --- a/mmocr/datasets/pipelines/textdet_targets/drrg_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/drrg_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np from lanms import merge_quadrangle_n9 as la_nms diff --git a/mmocr/datasets/pipelines/textdet_targets/fcenet_targets.py b/mmocr/datasets/pipelines/textdet_targets/fcenet_targets.py index fc9e8677..2d667b58 100644 --- a/mmocr/datasets/pipelines/textdet_targets/fcenet_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/fcenet_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np from mmdet.datasets.builder import PIPELINES diff --git a/mmocr/datasets/pipelines/textdet_targets/panet_targets.py b/mmocr/datasets/pipelines/textdet_targets/panet_targets.py index cd54c004..92449cdb 100644 --- a/mmocr/datasets/pipelines/textdet_targets/panet_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/panet_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.core import BitmapMasks from mmdet.datasets.builder import PIPELINES diff --git a/mmocr/datasets/pipelines/textdet_targets/psenet_targets.py b/mmocr/datasets/pipelines/textdet_targets/psenet_targets.py index 35e831f3..0bdc77fa 100644 --- a/mmocr/datasets/pipelines/textdet_targets/psenet_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/psenet_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.datasets.builder import PIPELINES from . import PANetTargets diff --git a/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py b/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py index 7b43e979..c37d204f 100644 --- a/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py +++ b/mmocr/datasets/pipelines/textdet_targets/textsnake_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np from mmdet.core import BitmapMasks diff --git a/mmocr/datasets/pipelines/transforms.py b/mmocr/datasets/pipelines/transforms.py index 06c43bc4..3366385f 100644 --- a/mmocr/datasets/pipelines/transforms.py +++ b/mmocr/datasets/pipelines/transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import cv2 diff --git a/mmocr/datasets/text_det_dataset.py b/mmocr/datasets/text_det_dataset.py index d428a373..c150b60d 100644 --- a/mmocr/datasets/text_det_dataset.py +++ b/mmocr/datasets/text_det_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmdet.datasets.builder import DATASETS diff --git a/mmocr/datasets/uniform_concat_dataset.py b/mmocr/datasets/uniform_concat_dataset.py index d365aff5..5e1aa42d 100644 --- a/mmocr/datasets/uniform_concat_dataset.py +++ b/mmocr/datasets/uniform_concat_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from mmdet.datasets import DATASETS, ConcatDataset, build_dataset diff --git a/mmocr/datasets/utils/__init__.py b/mmocr/datasets/utils/__init__.py index f014de7e..9501c868 100644 --- a/mmocr/datasets/utils/__init__.py +++ b/mmocr/datasets/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .loader import HardDiskLoader, LmdbLoader from .parser import LineJsonParser, LineStrParser diff --git a/mmocr/datasets/utils/loader.py b/mmocr/datasets/utils/loader.py index e9c65f0d..44a8ae69 100644 --- a/mmocr/datasets/utils/loader.py +++ b/mmocr/datasets/utils/loader.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from mmocr.datasets.builder import LOADERS, build_parser diff --git a/mmocr/datasets/utils/parser.py b/mmocr/datasets/utils/parser.py index b99e1d65..498c6609 100644 --- a/mmocr/datasets/utils/parser.py +++ b/mmocr/datasets/utils/parser.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json from mmocr.datasets.builder import PARSERS diff --git a/mmocr/models/__init__.py b/mmocr/models/__init__.py index 5047b7b4..d4975e27 100644 --- a/mmocr/models/__init__.py +++ b/mmocr/models/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import (BACKBONES, DETECTORS, HEADS, LOSSES, NECKS, build_backbone, build_detector, build_loss) diff --git a/mmocr/models/builder.py b/mmocr/models/builder.py index cd606a8a..7f638c35 100644 --- a/mmocr/models/builder.py +++ b/mmocr/models/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import Registry, build_from_cfg RECOGNIZERS = Registry('recognizer') diff --git a/mmocr/models/common/__init__.py b/mmocr/models/common/__init__.py index 8e414e1e..e8a7f671 100644 --- a/mmocr/models/common/__init__.py +++ b/mmocr/models/common/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from . import backbones, losses from .backbones import * # NOQA diff --git a/mmocr/models/common/backbones/__init__.py b/mmocr/models/common/backbones/__init__.py index de67ca96..3c384ba3 100644 --- a/mmocr/models/common/backbones/__init__.py +++ b/mmocr/models/common/backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .unet import UNet __all__ = ['UNet'] diff --git a/mmocr/models/common/backbones/unet.py b/mmocr/models/common/backbones/unet.py index 5efec3ad..215518e5 100644 --- a/mmocr/models/common/backbones/unet.py +++ b/mmocr/models/common/backbones/unet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torch.utils.checkpoint as cp diff --git a/mmocr/models/common/losses/__init__.py b/mmocr/models/common/losses/__init__.py index cbba1432..67151b69 100644 --- a/mmocr/models/common/losses/__init__.py +++ b/mmocr/models/common/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .dice_loss import DiceLoss from .focal_loss import FocalLoss diff --git a/mmocr/models/common/losses/dice_loss.py b/mmocr/models/common/losses/dice_loss.py index 81b917f4..da019509 100644 --- a/mmocr/models/common/losses/dice_loss.py +++ b/mmocr/models/common/losses/dice_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmdet.models.builder import LOSSES diff --git a/mmocr/models/common/losses/focal_loss.py b/mmocr/models/common/losses/focal_loss.py index b67a1c46..1a42ab01 100644 --- a/mmocr/models/common/losses/focal_loss.py +++ b/mmocr/models/common/losses/focal_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torch.nn.functional as F diff --git a/mmocr/models/kie/__init__.py b/mmocr/models/kie/__init__.py index fddc25d9..9fd5d2de 100644 --- a/mmocr/models/kie/__init__.py +++ b/mmocr/models/kie/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from . import extractors, heads, losses from .extractors import * # NOQA diff --git a/mmocr/models/kie/extractors/__init__.py b/mmocr/models/kie/extractors/__init__.py index f58541e6..914d0f69 100644 --- a/mmocr/models/kie/extractors/__init__.py +++ b/mmocr/models/kie/extractors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .sdmgr import SDMGR __all__ = ['SDMGR'] diff --git a/mmocr/models/kie/extractors/sdmgr.py b/mmocr/models/kie/extractors/sdmgr.py index d008508d..016b11fe 100644 --- a/mmocr/models/kie/extractors/sdmgr.py +++ b/mmocr/models/kie/extractors/sdmgr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import mmcv diff --git a/mmocr/models/kie/heads/__init__.py b/mmocr/models/kie/heads/__init__.py index 00a11469..1c08ed6f 100644 --- a/mmocr/models/kie/heads/__init__.py +++ b/mmocr/models/kie/heads/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .sdmgr_head import SDMGRHead __all__ = ['SDMGRHead'] diff --git a/mmocr/models/kie/heads/sdmgr_head.py b/mmocr/models/kie/heads/sdmgr_head.py index 7b62d6bb..012467e6 100644 --- a/mmocr/models/kie/heads/sdmgr_head.py +++ b/mmocr/models/kie/heads/sdmgr_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.runner import BaseModule from mmdet.models.builder import HEADS, build_loss diff --git a/mmocr/models/kie/losses/__init__.py b/mmocr/models/kie/losses/__init__.py index 96b4afde..4a72f8ca 100644 --- a/mmocr/models/kie/losses/__init__.py +++ b/mmocr/models/kie/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .sdmgr_loss import SDMGRLoss __all__ = ['SDMGRLoss'] diff --git a/mmocr/models/kie/losses/sdmgr_loss.py b/mmocr/models/kie/losses/sdmgr_loss.py index 7ccc47d5..9625bc8d 100644 --- a/mmocr/models/kie/losses/sdmgr_loss.py +++ b/mmocr/models/kie/losses/sdmgr_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmdet.models.builder import LOSSES from mmdet.models.losses import accuracy diff --git a/mmocr/models/ner/__init__.py b/mmocr/models/ner/__init__.py index 248c7eca..1f8245c9 100644 --- a/mmocr/models/ner/__init__.py +++ b/mmocr/models/ner/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from . import classifiers, convertors, decoders, encoders, losses from .classifiers import * # NOQA diff --git a/mmocr/models/ner/classifiers/__init__.py b/mmocr/models/ner/classifiers/__init__.py index 13189569..63891874 100644 --- a/mmocr/models/ner/classifiers/__init__.py +++ b/mmocr/models/ner/classifiers/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .ner_classifier import NerClassifier __all__ = ['NerClassifier'] diff --git a/mmocr/models/ner/classifiers/ner_classifier.py b/mmocr/models/ner/classifiers/ner_classifier.py index 44b372be..161b68b6 100644 --- a/mmocr/models/ner/classifiers/ner_classifier.py +++ b/mmocr/models/ner/classifiers/ner_classifier.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS, build_loss from mmocr.models.builder import build_convertor, build_decoder, build_encoder diff --git a/mmocr/models/ner/convertors/__init__.py b/mmocr/models/ner/convertors/__init__.py index 9186e46d..4d4e15c3 100644 --- a/mmocr/models/ner/convertors/__init__.py +++ b/mmocr/models/ner/convertors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .ner_convertor import NerConvertor __all__ = ['NerConvertor'] diff --git a/mmocr/models/ner/convertors/ner_convertor.py b/mmocr/models/ner/convertors/ner_convertor.py index 27db595d..ca7288bc 100644 --- a/mmocr/models/ner/convertors/ner_convertor.py +++ b/mmocr/models/ner/convertors/ner_convertor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmocr.models.builder import CONVERTORS diff --git a/mmocr/models/ner/decoders/__init__.py b/mmocr/models/ner/decoders/__init__.py index c3fd96b7..737e98fa 100644 --- a/mmocr/models/ner/decoders/__init__.py +++ b/mmocr/models/ner/decoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .fc_decoder import FCDecoder __all__ = ['FCDecoder'] diff --git a/mmocr/models/ner/decoders/fc_decoder.py b/mmocr/models/ner/decoders/fc_decoder.py index 47adf8b7..b88302f1 100644 --- a/mmocr/models/ner/decoders/fc_decoder.py +++ b/mmocr/models/ner/decoders/fc_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch.nn as nn import torch.nn.functional as F diff --git a/mmocr/models/ner/encoders/__init__.py b/mmocr/models/ner/encoders/__init__.py index 0110ca0c..4d7629bd 100755 --- a/mmocr/models/ner/encoders/__init__.py +++ b/mmocr/models/ner/encoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .bert_encoder import BertEncoder __all__ = ['BertEncoder'] diff --git a/mmocr/models/ner/encoders/bert_encoder.py b/mmocr/models/ner/encoders/bert_encoder.py index fc5c6acb..24c60aae 100644 --- a/mmocr/models/ner/encoders/bert_encoder.py +++ b/mmocr/models/ner/encoders/bert_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import BaseModule from mmocr.models.builder import ENCODERS diff --git a/mmocr/models/ner/losses/__init__.py b/mmocr/models/ner/losses/__init__.py index e35c095e..44cb725b 100644 --- a/mmocr/models/ner/losses/__init__.py +++ b/mmocr/models/ner/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .masked_cross_entropy_loss import MaskedCrossEntropyLoss from .masked_focal_loss import MaskedFocalLoss diff --git a/mmocr/models/ner/losses/masked_cross_entropy_loss.py b/mmocr/models/ner/losses/masked_cross_entropy_loss.py index 62821477..9f30c2be 100644 --- a/mmocr/models/ner/losses/masked_cross_entropy_loss.py +++ b/mmocr/models/ner/losses/masked_cross_entropy_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import LOSSES from torch import nn from torch.nn import CrossEntropyLoss diff --git a/mmocr/models/ner/losses/masked_focal_loss.py b/mmocr/models/ner/losses/masked_focal_loss.py index 4b2ee170..e1c4309c 100644 --- a/mmocr/models/ner/losses/masked_focal_loss.py +++ b/mmocr/models/ner/losses/masked_focal_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import LOSSES from torch import nn diff --git a/mmocr/models/ner/utils/__init__.py b/mmocr/models/ner/utils/__init__.py index 4d4740b2..076239cd 100644 --- a/mmocr/models/ner/utils/__init__.py +++ b/mmocr/models/ner/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .activations import GeluNew from .bert import BertModel diff --git a/mmocr/models/textdet/__init__.py b/mmocr/models/textdet/__init__.py index b7ff7a24..b644d103 100644 --- a/mmocr/models/textdet/__init__.py +++ b/mmocr/models/textdet/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from . import dense_heads, detectors, losses, necks, postprocess from .dense_heads import * # NOQA diff --git a/mmocr/models/textdet/dense_heads/__init__.py b/mmocr/models/textdet/dense_heads/__init__.py index d4403527..785cb55f 100644 --- a/mmocr/models/textdet/dense_heads/__init__.py +++ b/mmocr/models/textdet/dense_heads/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .db_head import DBHead from .drrg_head import DRRGHead from .fce_head import FCEHead diff --git a/mmocr/models/textdet/dense_heads/db_head.py b/mmocr/models/textdet/dense_heads/db_head.py index 7ea59250..49fa2030 100644 --- a/mmocr/models/textdet/dense_heads/db_head.py +++ b/mmocr/models/textdet/dense_heads/db_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import BaseModule, Sequential diff --git a/mmocr/models/textdet/dense_heads/drrg_head.py b/mmocr/models/textdet/dense_heads/drrg_head.py index 00e37219..041604ad 100644 --- a/mmocr/models/textdet/dense_heads/drrg_head.py +++ b/mmocr/models/textdet/dense_heads/drrg_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/mmocr/models/textdet/dense_heads/fce_head.py b/mmocr/models/textdet/dense_heads/fce_head.py index 9d956b5a..d4edd18f 100644 --- a/mmocr/models/textdet/dense_heads/fce_head.py +++ b/mmocr/models/textdet/dense_heads/fce_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import BaseModule from mmdet.core import multi_apply diff --git a/mmocr/models/textdet/dense_heads/head_mixin.py b/mmocr/models/textdet/dense_heads/head_mixin.py index 6ede6221..9361d521 100644 --- a/mmocr/models/textdet/dense_heads/head_mixin.py +++ b/mmocr/models/textdet/dense_heads/head_mixin.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmdet.models.builder import HEADS diff --git a/mmocr/models/textdet/dense_heads/pan_head.py b/mmocr/models/textdet/dense_heads/pan_head.py index e7882271..466c8173 100644 --- a/mmocr/models/textdet/dense_heads/pan_head.py +++ b/mmocr/models/textdet/dense_heads/pan_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/mmocr/models/textdet/dense_heads/pse_head.py b/mmocr/models/textdet/dense_heads/pse_head.py index adb33583..e22ba736 100644 --- a/mmocr/models/textdet/dense_heads/pse_head.py +++ b/mmocr/models/textdet/dense_heads/pse_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import HEADS from . import PANHead diff --git a/mmocr/models/textdet/dense_heads/textsnake_head.py b/mmocr/models/textdet/dense_heads/textsnake_head.py index 873c027e..27c6e502 100644 --- a/mmocr/models/textdet/dense_heads/textsnake_head.py +++ b/mmocr/models/textdet/dense_heads/textsnake_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import BaseModule from mmdet.models.builder import HEADS, build_loss diff --git a/mmocr/models/textdet/detectors/__init__.py b/mmocr/models/textdet/detectors/__init__.py index 7919a8a5..290beee9 100644 --- a/mmocr/models/textdet/detectors/__init__.py +++ b/mmocr/models/textdet/detectors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .dbnet import DBNet from .drrg import DRRG from .fcenet import FCENet diff --git a/mmocr/models/textdet/detectors/dbnet.py b/mmocr/models/textdet/detectors/dbnet.py index 7ed2132a..e4931998 100644 --- a/mmocr/models/textdet/detectors/dbnet.py +++ b/mmocr/models/textdet/detectors/dbnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from mmocr.models.textdet.detectors.single_stage_text_detector import \ diff --git a/mmocr/models/textdet/detectors/drrg.py b/mmocr/models/textdet/detectors/drrg.py index 93c51369..2d4bdc61 100644 --- a/mmocr/models/textdet/detectors/drrg.py +++ b/mmocr/models/textdet/detectors/drrg.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from mmocr.models.textdet.detectors.single_stage_text_detector import \ diff --git a/mmocr/models/textdet/detectors/fcenet.py b/mmocr/models/textdet/detectors/fcenet.py index 1a2ed440..0d8bcf59 100644 --- a/mmocr/models/textdet/detectors/fcenet.py +++ b/mmocr/models/textdet/detectors/fcenet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from .single_stage_text_detector import SingleStageTextDetector diff --git a/mmocr/models/textdet/detectors/ocr_mask_rcnn.py b/mmocr/models/textdet/detectors/ocr_mask_rcnn.py index afaafc15..2ce02000 100644 --- a/mmocr/models/textdet/detectors/ocr_mask_rcnn.py +++ b/mmocr/models/textdet/detectors/ocr_mask_rcnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from mmdet.models.detectors import MaskRCNN diff --git a/mmocr/models/textdet/detectors/panet.py b/mmocr/models/textdet/detectors/panet.py index 643b51c9..20667a9d 100644 --- a/mmocr/models/textdet/detectors/panet.py +++ b/mmocr/models/textdet/detectors/panet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from mmocr.models.textdet.detectors.single_stage_text_detector import \ diff --git a/mmocr/models/textdet/detectors/psenet.py b/mmocr/models/textdet/detectors/psenet.py index 9d65d812..45dfed6f 100644 --- a/mmocr/models/textdet/detectors/psenet.py +++ b/mmocr/models/textdet/detectors/psenet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from mmocr.models.textdet.detectors.single_stage_text_detector import \ diff --git a/mmocr/models/textdet/detectors/single_stage_text_detector.py b/mmocr/models/textdet/detectors/single_stage_text_detector.py index 82be5877..2e55b0ff 100644 --- a/mmocr/models/textdet/detectors/single_stage_text_detector.py +++ b/mmocr/models/textdet/detectors/single_stage_text_detector.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmdet.models.builder import DETECTORS from mmdet.models.detectors import SingleStageDetector diff --git a/mmocr/models/textdet/detectors/text_detector_mixin.py b/mmocr/models/textdet/detectors/text_detector_mixin.py index 615c8ad2..a2a56b63 100644 --- a/mmocr/models/textdet/detectors/text_detector_mixin.py +++ b/mmocr/models/textdet/detectors/text_detector_mixin.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import mmcv diff --git a/mmocr/models/textdet/detectors/textsnake.py b/mmocr/models/textdet/detectors/textsnake.py index 38062135..693af0f9 100644 --- a/mmocr/models/textdet/detectors/textsnake.py +++ b/mmocr/models/textdet/detectors/textsnake.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from . import SingleStageTextDetector, TextDetectorMixin diff --git a/mmocr/models/textdet/losses/__init__.py b/mmocr/models/textdet/losses/__init__.py index 0049a7e9..a4f247b6 100644 --- a/mmocr/models/textdet/losses/__init__.py +++ b/mmocr/models/textdet/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .db_loss import DBLoss from .drrg_loss import DRRGLoss from .fce_loss import FCELoss diff --git a/mmocr/models/textdet/losses/db_loss.py b/mmocr/models/textdet/losses/db_loss.py index abfa377f..acfcbc7f 100644 --- a/mmocr/models/textdet/losses/db_loss.py +++ b/mmocr/models/textdet/losses/db_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmdet.models.builder import LOSSES diff --git a/mmocr/models/textdet/losses/drrg_loss.py b/mmocr/models/textdet/losses/drrg_loss.py index fcaf9a9c..9915f8a4 100644 --- a/mmocr/models/textdet/losses/drrg_loss.py +++ b/mmocr/models/textdet/losses/drrg_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmdet.core import BitmapMasks diff --git a/mmocr/models/textdet/losses/fce_loss.py b/mmocr/models/textdet/losses/fce_loss.py index 1632e98e..81f21f1b 100644 --- a/mmocr/models/textdet/losses/fce_loss.py +++ b/mmocr/models/textdet/losses/fce_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn.functional as F diff --git a/mmocr/models/textdet/losses/pan_loss.py b/mmocr/models/textdet/losses/pan_loss.py index c0751f48..9d34db4b 100644 --- a/mmocr/models/textdet/losses/pan_loss.py +++ b/mmocr/models/textdet/losses/pan_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import itertools import warnings diff --git a/mmocr/models/textdet/losses/pse_loss.py b/mmocr/models/textdet/losses/pse_loss.py index 739bc59d..b8cbc040 100644 --- a/mmocr/models/textdet/losses/pse_loss.py +++ b/mmocr/models/textdet/losses/pse_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.core import BitmapMasks from mmdet.models.builder import LOSSES diff --git a/mmocr/models/textdet/losses/textsnake_loss.py b/mmocr/models/textdet/losses/textsnake_loss.py index fe151b8a..00477430 100644 --- a/mmocr/models/textdet/losses/textsnake_loss.py +++ b/mmocr/models/textdet/losses/textsnake_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmdet.core import BitmapMasks diff --git a/mmocr/models/textdet/modules/__init__.py b/mmocr/models/textdet/modules/__init__.py index cbd0a9be..1a863d0f 100644 --- a/mmocr/models/textdet/modules/__init__.py +++ b/mmocr/models/textdet/modules/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .gcn import GCN from .local_graph import LocalGraphs from .proposal_local_graph import ProposalLocalGraphs diff --git a/mmocr/models/textdet/modules/gcn.py b/mmocr/models/textdet/modules/gcn.py index 77225b68..092d6463 100644 --- a/mmocr/models/textdet/modules/gcn.py +++ b/mmocr/models/textdet/modules/gcn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torch.nn.functional as F diff --git a/mmocr/models/textdet/modules/local_graph.py b/mmocr/models/textdet/modules/local_graph.py index 903de9d5..86158203 100644 --- a/mmocr/models/textdet/modules/local_graph.py +++ b/mmocr/models/textdet/modules/local_graph.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch from mmcv.ops import RoIAlignRotated diff --git a/mmocr/models/textdet/modules/proposal_local_graph.py b/mmocr/models/textdet/modules/proposal_local_graph.py index 3bf6913d..f98cf4e0 100644 --- a/mmocr/models/textdet/modules/proposal_local_graph.py +++ b/mmocr/models/textdet/modules/proposal_local_graph.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np import torch diff --git a/mmocr/models/textdet/modules/utils.py b/mmocr/models/textdet/modules/utils.py index c299a44d..48e2eff1 100644 --- a/mmocr/models/textdet/modules/utils.py +++ b/mmocr/models/textdet/modules/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np diff --git a/mmocr/models/textdet/necks/__init__.py b/mmocr/models/textdet/necks/__init__.py index 17d96d82..0b21bf19 100644 --- a/mmocr/models/textdet/necks/__init__.py +++ b/mmocr/models/textdet/necks/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .fpem_ffm import FPEM_FFM from .fpn_cat import FPNC from .fpn_unet import FPN_UNet diff --git a/mmocr/models/textdet/necks/fpem_ffm.py b/mmocr/models/textdet/necks/fpem_ffm.py index 999eaf95..e65e2e27 100644 --- a/mmocr/models/textdet/necks/fpem_ffm.py +++ b/mmocr/models/textdet/necks/fpem_ffm.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn.functional as F from mmcv.runner import BaseModule, ModuleList from mmdet.models.builder import NECKS diff --git a/mmocr/models/textdet/necks/fpn_cat.py b/mmocr/models/textdet/necks/fpn_cat.py index 1a762e9d..b3eb4944 100644 --- a/mmocr/models/textdet/necks/fpn_cat.py +++ b/mmocr/models/textdet/necks/fpn_cat.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmcv.cnn import ConvModule diff --git a/mmocr/models/textdet/necks/fpn_unet.py b/mmocr/models/textdet/necks/fpn_unet.py index ff92d822..9b5d3c40 100644 --- a/mmocr/models/textdet/necks/fpn_unet.py +++ b/mmocr/models/textdet/necks/fpn_unet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F # from mmcv.cnn import xavier_init diff --git a/mmocr/models/textdet/necks/fpnf.py b/mmocr/models/textdet/necks/fpnf.py index 870fd87e..110c5509 100644 --- a/mmocr/models/textdet/necks/fpnf.py +++ b/mmocr/models/textdet/necks/fpnf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmcv.cnn import ConvModule diff --git a/mmocr/models/textdet/postprocess/__init__.py b/mmocr/models/textdet/postprocess/__init__.py index acc72530..2f86a7ce 100644 --- a/mmocr/models/textdet/postprocess/__init__.py +++ b/mmocr/models/textdet/postprocess/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .wrapper import decode __all__ = ['decode'] diff --git a/mmocr/models/textdet/postprocess/wrapper.py b/mmocr/models/textdet/postprocess/wrapper.py index ff4c7ae8..289aa7a7 100644 --- a/mmocr/models/textdet/postprocess/wrapper.py +++ b/mmocr/models/textdet/postprocess/wrapper.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import functools import operator diff --git a/mmocr/models/textrecog/__init__.py b/mmocr/models/textrecog/__init__.py index a5a2729c..be5afd5d 100644 --- a/mmocr/models/textrecog/__init__.py +++ b/mmocr/models/textrecog/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from . import (backbones, convertors, decoders, encoders, heads, losses, necks, preprocessor, recognizer) diff --git a/mmocr/models/textrecog/backbones/__init__.py b/mmocr/models/textrecog/backbones/__init__.py index 51517090..57a69ffd 100644 --- a/mmocr/models/textrecog/backbones/__init__.py +++ b/mmocr/models/textrecog/backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .nrtr_modality_transformer import NRTRModalityTransform from .resnet31_ocr import ResNet31OCR from .very_deep_vgg import VeryDeepVgg diff --git a/mmocr/models/textrecog/backbones/nrtr_modality_transformer.py b/mmocr/models/textrecog/backbones/nrtr_modality_transformer.py index 9645824e..5f92fb41 100644 --- a/mmocr/models/textrecog/backbones/nrtr_modality_transformer.py +++ b/mmocr/models/textrecog/backbones/nrtr_modality_transformer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import BaseModule from mmdet.models.builder import BACKBONES diff --git a/mmocr/models/textrecog/backbones/resnet31_ocr.py b/mmocr/models/textrecog/backbones/resnet31_ocr.py index 0b0fbd6f..3a97d240 100644 --- a/mmocr/models/textrecog/backbones/resnet31_ocr.py +++ b/mmocr/models/textrecog/backbones/resnet31_ocr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import BaseModule, Sequential from mmdet.models.builder import BACKBONES diff --git a/mmocr/models/textrecog/backbones/very_deep_vgg.py b/mmocr/models/textrecog/backbones/very_deep_vgg.py index 8cceb4d7..faf4a58d 100644 --- a/mmocr/models/textrecog/backbones/very_deep_vgg.py +++ b/mmocr/models/textrecog/backbones/very_deep_vgg.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import BaseModule, Sequential from mmdet.models.builder import BACKBONES diff --git a/mmocr/models/textrecog/convertors/__init__.py b/mmocr/models/textrecog/convertors/__init__.py index 60fc6300..345e8933 100644 --- a/mmocr/models/textrecog/convertors/__init__.py +++ b/mmocr/models/textrecog/convertors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .attn import AttnConvertor from .base import BaseConvertor from .ctc import CTCConvertor diff --git a/mmocr/models/textrecog/convertors/attn.py b/mmocr/models/textrecog/convertors/attn.py index a80282e8..e90f841e 100644 --- a/mmocr/models/textrecog/convertors/attn.py +++ b/mmocr/models/textrecog/convertors/attn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import mmocr.utils as utils diff --git a/mmocr/models/textrecog/convertors/base.py b/mmocr/models/textrecog/convertors/base.py index e54a232a..976299d9 100644 --- a/mmocr/models/textrecog/convertors/base.py +++ b/mmocr/models/textrecog/convertors/base.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmocr.models.builder import CONVERTORS from mmocr.utils import list_from_file diff --git a/mmocr/models/textrecog/convertors/ctc.py b/mmocr/models/textrecog/convertors/ctc.py index c14fc23f..ec4d037d 100644 --- a/mmocr/models/textrecog/convertors/ctc.py +++ b/mmocr/models/textrecog/convertors/ctc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/convertors/seg.py b/mmocr/models/textrecog/convertors/seg.py index 97f11c3d..5bc115d1 100644 --- a/mmocr/models/textrecog/convertors/seg.py +++ b/mmocr/models/textrecog/convertors/seg.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np import torch diff --git a/mmocr/models/textrecog/decoders/__init__.py b/mmocr/models/textrecog/decoders/__init__.py index 8b374733..7b6fad8c 100755 --- a/mmocr/models/textrecog/decoders/__init__.py +++ b/mmocr/models/textrecog/decoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_decoder import BaseDecoder from .crnn_decoder import CRNNDecoder from .position_attention_decoder import PositionAttentionDecoder diff --git a/mmocr/models/textrecog/decoders/base_decoder.py b/mmocr/models/textrecog/decoders/base_decoder.py index c55c40df..09e2db88 100644 --- a/mmocr/models/textrecog/decoders/base_decoder.py +++ b/mmocr/models/textrecog/decoders/base_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import BaseModule from mmocr.models.builder import DECODERS diff --git a/mmocr/models/textrecog/decoders/crnn_decoder.py b/mmocr/models/textrecog/decoders/crnn_decoder.py index 97a7ab27..9e324597 100644 --- a/mmocr/models/textrecog/decoders/crnn_decoder.py +++ b/mmocr/models/textrecog/decoders/crnn_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import Sequential diff --git a/mmocr/models/textrecog/decoders/position_attention_decoder.py b/mmocr/models/textrecog/decoders/position_attention_decoder.py index 1fd65252..1c74baae 100644 --- a/mmocr/models/textrecog/decoders/position_attention_decoder.py +++ b/mmocr/models/textrecog/decoders/position_attention_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/decoders/robust_scanner_decoder.py b/mmocr/models/textrecog/decoders/robust_scanner_decoder.py index 527b6fc3..4d734ec9 100644 --- a/mmocr/models/textrecog/decoders/robust_scanner_decoder.py +++ b/mmocr/models/textrecog/decoders/robust_scanner_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torch.nn.functional as F diff --git a/mmocr/models/textrecog/decoders/sar_decoder.py b/mmocr/models/textrecog/decoders/sar_decoder.py index 61c905c7..5bd440bf 100755 --- a/mmocr/models/textrecog/decoders/sar_decoder.py +++ b/mmocr/models/textrecog/decoders/sar_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/decoders/sar_decoder_with_bs.py b/mmocr/models/textrecog/decoders/sar_decoder_with_bs.py index 30168bf2..d00e385d 100755 --- a/mmocr/models/textrecog/decoders/sar_decoder_with_bs.py +++ b/mmocr/models/textrecog/decoders/sar_decoder_with_bs.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from queue import PriorityQueue import torch diff --git a/mmocr/models/textrecog/decoders/sequence_attention_decoder.py b/mmocr/models/textrecog/decoders/sequence_attention_decoder.py index cf53df40..b961b15e 100644 --- a/mmocr/models/textrecog/decoders/sequence_attention_decoder.py +++ b/mmocr/models/textrecog/decoders/sequence_attention_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/decoders/transformer_decoder.py b/mmocr/models/textrecog/decoders/transformer_decoder.py index d0085997..9ae9a564 100644 --- a/mmocr/models/textrecog/decoders/transformer_decoder.py +++ b/mmocr/models/textrecog/decoders/transformer_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/encoders/__init__.py b/mmocr/models/textrecog/encoders/__init__.py index e0d9394a..70d690c3 100755 --- a/mmocr/models/textrecog/encoders/__init__.py +++ b/mmocr/models/textrecog/encoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_encoder import BaseEncoder from .channel_reduction_encoder import ChannelReductionEncoder from .sar_encoder import SAREncoder diff --git a/mmocr/models/textrecog/encoders/base_encoder.py b/mmocr/models/textrecog/encoders/base_encoder.py index 29e2a5dc..726c78a8 100644 --- a/mmocr/models/textrecog/encoders/base_encoder.py +++ b/mmocr/models/textrecog/encoders/base_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import BaseModule from mmocr.models.builder import ENCODERS diff --git a/mmocr/models/textrecog/encoders/channel_reduction_encoder.py b/mmocr/models/textrecog/encoders/channel_reduction_encoder.py index 81d84403..789a6778 100644 --- a/mmocr/models/textrecog/encoders/channel_reduction_encoder.py +++ b/mmocr/models/textrecog/encoders/channel_reduction_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmocr.models.builder import ENCODERS diff --git a/mmocr/models/textrecog/encoders/sar_encoder.py b/mmocr/models/textrecog/encoders/sar_encoder.py index e43ce982..de2e97ef 100644 --- a/mmocr/models/textrecog/encoders/sar_encoder.py +++ b/mmocr/models/textrecog/encoders/sar_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/encoders/transformer_encoder.py b/mmocr/models/textrecog/encoders/transformer_encoder.py index e2bbd61a..dd799e39 100644 --- a/mmocr/models/textrecog/encoders/transformer_encoder.py +++ b/mmocr/models/textrecog/encoders/transformer_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch.nn as nn diff --git a/mmocr/models/textrecog/heads/__init__.py b/mmocr/models/textrecog/heads/__init__.py index 761bb9a9..03e27606 100755 --- a/mmocr/models/textrecog/heads/__init__.py +++ b/mmocr/models/textrecog/heads/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .seg_head import SegHead __all__ = ['SegHead'] diff --git a/mmocr/models/textrecog/heads/seg_head.py b/mmocr/models/textrecog/heads/seg_head.py index e9ec3b6b..6cb3d94f 100644 --- a/mmocr/models/textrecog/heads/seg_head.py +++ b/mmocr/models/textrecog/heads/seg_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn.functional as F from mmcv.cnn import ConvModule from mmcv.runner import BaseModule diff --git a/mmocr/models/textrecog/layers/__init__.py b/mmocr/models/textrecog/layers/__init__.py index 7d85a865..f7b72727 100755 --- a/mmocr/models/textrecog/layers/__init__.py +++ b/mmocr/models/textrecog/layers/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .conv_layer import BasicBlock, Bottleneck from .dot_product_attention_layer import DotProductAttentionLayer from .lstm_layer import BidirectionalLSTM diff --git a/mmocr/models/textrecog/layers/conv_layer.py b/mmocr/models/textrecog/layers/conv_layer.py index d0ce32a3..22922479 100644 --- a/mmocr/models/textrecog/layers/conv_layer.py +++ b/mmocr/models/textrecog/layers/conv_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn diff --git a/mmocr/models/textrecog/layers/dot_product_attention_layer.py b/mmocr/models/textrecog/layers/dot_product_attention_layer.py index efa55a8c..6d9cdb65 100644 --- a/mmocr/models/textrecog/layers/dot_product_attention_layer.py +++ b/mmocr/models/textrecog/layers/dot_product_attention_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torch.nn.functional as F diff --git a/mmocr/models/textrecog/layers/lstm_layer.py b/mmocr/models/textrecog/layers/lstm_layer.py index e4017d02..16d3c1a4 100644 --- a/mmocr/models/textrecog/layers/lstm_layer.py +++ b/mmocr/models/textrecog/layers/lstm_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn diff --git a/mmocr/models/textrecog/layers/position_aware_layer.py b/mmocr/models/textrecog/layers/position_aware_layer.py index cf8cf27d..2c994e37 100644 --- a/mmocr/models/textrecog/layers/position_aware_layer.py +++ b/mmocr/models/textrecog/layers/position_aware_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn diff --git a/mmocr/models/textrecog/layers/robust_scanner_fusion_layer.py b/mmocr/models/textrecog/layers/robust_scanner_fusion_layer.py index 1929bb56..af256874 100644 --- a/mmocr/models/textrecog/layers/robust_scanner_fusion_layer.py +++ b/mmocr/models/textrecog/layers/robust_scanner_fusion_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import BaseModule diff --git a/mmocr/models/textrecog/layers/transformer_layer.py b/mmocr/models/textrecog/layers/transformer_layer.py index 8377827b..fc1ce39b 100644 --- a/mmocr/models/textrecog/layers/transformer_layer.py +++ b/mmocr/models/textrecog/layers/transformer_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """This code is from https://github.com/jadore801120/attention-is-all-you-need- pytorch.""" import numpy as np diff --git a/mmocr/models/textrecog/losses/__init__.py b/mmocr/models/textrecog/losses/__init__.py index 226aa006..1b758bc7 100755 --- a/mmocr/models/textrecog/losses/__init__.py +++ b/mmocr/models/textrecog/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .ce_loss import CELoss, SARLoss, TFLoss from .ctc_loss import CTCLoss from .seg_loss import SegLoss diff --git a/mmocr/models/textrecog/losses/ce_loss.py b/mmocr/models/textrecog/losses/ce_loss.py index 0c7dc8fc..1dddefa4 100644 --- a/mmocr/models/textrecog/losses/ce_loss.py +++ b/mmocr/models/textrecog/losses/ce_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmdet.models.builder import LOSSES diff --git a/mmocr/models/textrecog/losses/ctc_loss.py b/mmocr/models/textrecog/losses/ctc_loss.py index d2747180..6b88ffaa 100644 --- a/mmocr/models/textrecog/losses/ctc_loss.py +++ b/mmocr/models/textrecog/losses/ctc_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmocr/models/textrecog/losses/seg_loss.py b/mmocr/models/textrecog/losses/seg_loss.py index 31c7996f..9dbe561d 100644 --- a/mmocr/models/textrecog/losses/seg_loss.py +++ b/mmocr/models/textrecog/losses/seg_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torch.nn.functional as F diff --git a/mmocr/models/textrecog/necks/__init__.py b/mmocr/models/textrecog/necks/__init__.py index 71ceadc1..81a57144 100755 --- a/mmocr/models/textrecog/necks/__init__.py +++ b/mmocr/models/textrecog/necks/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .fpn_ocr import FPNOCR __all__ = ['FPNOCR'] diff --git a/mmocr/models/textrecog/necks/fpn_ocr.py b/mmocr/models/textrecog/necks/fpn_ocr.py index 716d9750..90237570 100644 --- a/mmocr/models/textrecog/necks/fpn_ocr.py +++ b/mmocr/models/textrecog/necks/fpn_ocr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmcv.cnn import ConvModule diff --git a/mmocr/models/textrecog/preprocessor/__init__.py b/mmocr/models/textrecog/preprocessor/__init__.py index f2acaff3..57ea828a 100644 --- a/mmocr/models/textrecog/preprocessor/__init__.py +++ b/mmocr/models/textrecog/preprocessor/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_preprocessor import BasePreprocessor from .tps_preprocessor import TPSPreprocessor diff --git a/mmocr/models/textrecog/preprocessor/base_preprocessor.py b/mmocr/models/textrecog/preprocessor/base_preprocessor.py index f08447d7..ddd4a8f7 100644 --- a/mmocr/models/textrecog/preprocessor/base_preprocessor.py +++ b/mmocr/models/textrecog/preprocessor/base_preprocessor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import BaseModule from mmocr.models.builder import PREPROCESSOR diff --git a/mmocr/models/textrecog/recognizer/__init__.py b/mmocr/models/textrecog/recognizer/__init__.py index 91af5666..bf70bf19 100644 --- a/mmocr/models/textrecog/recognizer/__init__.py +++ b/mmocr/models/textrecog/recognizer/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base import BaseRecognizer from .crnn import CRNNNet from .encode_decode_recognizer import EncodeDecodeRecognizer diff --git a/mmocr/models/textrecog/recognizer/base.py b/mmocr/models/textrecog/recognizer/base.py index ff4d2c61..66ae90f6 100644 --- a/mmocr/models/textrecog/recognizer/base.py +++ b/mmocr/models/textrecog/recognizer/base.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings from abc import ABCMeta, abstractmethod from collections import OrderedDict diff --git a/mmocr/models/textrecog/recognizer/crnn.py b/mmocr/models/textrecog/recognizer/crnn.py index 56f410c1..e2df61f9 100644 --- a/mmocr/models/textrecog/recognizer/crnn.py +++ b/mmocr/models/textrecog/recognizer/crnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from .encode_decode_recognizer import EncodeDecodeRecognizer diff --git a/mmocr/models/textrecog/recognizer/encode_decode_recognizer.py b/mmocr/models/textrecog/recognizer/encode_decode_recognizer.py index a0e5f407..0d05c046 100644 --- a/mmocr/models/textrecog/recognizer/encode_decode_recognizer.py +++ b/mmocr/models/textrecog/recognizer/encode_decode_recognizer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch diff --git a/mmocr/models/textrecog/recognizer/nrtr.py b/mmocr/models/textrecog/recognizer/nrtr.py index ddc4ea5a..63278f01 100644 --- a/mmocr/models/textrecog/recognizer/nrtr.py +++ b/mmocr/models/textrecog/recognizer/nrtr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from .encode_decode_recognizer import EncodeDecodeRecognizer diff --git a/mmocr/models/textrecog/recognizer/robust_scanner.py b/mmocr/models/textrecog/recognizer/robust_scanner.py index 924d3a67..3bb8faae 100644 --- a/mmocr/models/textrecog/recognizer/robust_scanner.py +++ b/mmocr/models/textrecog/recognizer/robust_scanner.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from .encode_decode_recognizer import EncodeDecodeRecognizer diff --git a/mmocr/models/textrecog/recognizer/sar.py b/mmocr/models/textrecog/recognizer/sar.py index 07f31819..22541674 100644 --- a/mmocr/models/textrecog/recognizer/sar.py +++ b/mmocr/models/textrecog/recognizer/sar.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmdet.models.builder import DETECTORS from .encode_decode_recognizer import EncodeDecodeRecognizer diff --git a/mmocr/models/textrecog/recognizer/seg_recognizer.py b/mmocr/models/textrecog/recognizer/seg_recognizer.py index ae48dd0a..16912996 100644 --- a/mmocr/models/textrecog/recognizer/seg_recognizer.py +++ b/mmocr/models/textrecog/recognizer/seg_recognizer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings from mmdet.models.builder import (DETECTORS, build_backbone, build_head, diff --git a/mmocr/utils/__init__.py b/mmocr/utils/__init__.py index ef2290f8..e725aade 100644 --- a/mmocr/utils/__init__.py +++ b/mmocr/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import Registry, build_from_cfg from .box_util import is_on_same_line, stitch_boxes_into_lines diff --git a/mmocr/utils/box_util.py b/mmocr/utils/box_util.py index 7e1fb892..0c2591dd 100644 --- a/mmocr/utils/box_util.py +++ b/mmocr/utils/box_util.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np diff --git a/mmocr/utils/check_argument.py b/mmocr/utils/check_argument.py index 92955df2..86963890 100644 --- a/mmocr/utils/check_argument.py +++ b/mmocr/utils/check_argument.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np diff --git a/mmocr/utils/collect_env.py b/mmocr/utils/collect_env.py index 10eef970..a8cb3c40 100644 --- a/mmocr/utils/collect_env.py +++ b/mmocr/utils/collect_env.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import collect_env as collect_base_env from mmcv.utils import get_git_hash diff --git a/mmocr/utils/data_convert_util.py b/mmocr/utils/data_convert_util.py index d50a7971..77580fc7 100644 --- a/mmocr/utils/data_convert_util.py +++ b/mmocr/utils/data_convert_util.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv diff --git a/mmocr/utils/fileio.py b/mmocr/utils/fileio.py index 2157a1d4..2e455daf 100644 --- a/mmocr/utils/fileio.py +++ b/mmocr/utils/fileio.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import mmcv diff --git a/mmocr/utils/img_util.py b/mmocr/utils/img_util.py index 949f961c..0804cfa0 100644 --- a/mmocr/utils/img_util.py +++ b/mmocr/utils/img_util.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import mmcv diff --git a/mmocr/utils/lmdb_util.py b/mmocr/utils/lmdb_util.py index 24edae96..72276cb2 100644 --- a/mmocr/utils/lmdb_util.py +++ b/mmocr/utils/lmdb_util.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import shutil import sys import time diff --git a/mmocr/utils/logger.py b/mmocr/utils/logger.py index 221bb9a9..294837fa 100644 --- a/mmocr/utils/logger.py +++ b/mmocr/utils/logger.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging from mmcv.utils import get_logger diff --git a/mmocr/utils/model.py b/mmocr/utils/model.py index 66978fa3..701a0311 100644 --- a/mmocr/utils/model.py +++ b/mmocr/utils/model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch diff --git a/mmocr/utils/ocr.py b/mmocr/utils/ocr.py index 7df9ea16..477a1520 100644 --- a/mmocr/utils/ocr.py +++ b/mmocr/utils/ocr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os import warnings diff --git a/mmocr/utils/string_util.py b/mmocr/utils/string_util.py index 8c7787e6..5a8946ee 100644 --- a/mmocr/utils/string_util.py +++ b/mmocr/utils/string_util.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. class StringStrip: """Removing the leading and/or the trailing characters based on the string argument passed. diff --git a/tests/test_apis/test_model_inference.py b/tests/test_apis/test_model_inference.py index 1b722de5..6420fa8e 100644 --- a/tests/test_apis/test_model_inference.py +++ b/tests/test_apis/test_model_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import pytest diff --git a/tests/test_core/test_deploy_utils.py b/tests/test_core/test_deploy_utils.py index e7303c8c..ec889db7 100644 --- a/tests/test_core/test_deploy_utils.py +++ b/tests/test_core/test_deploy_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile from functools import partial diff --git a/tests/test_core/test_end2end_vis.py b/tests/test_core/test_end2end_vis.py index e24d974f..2e7a6812 100644 --- a/tests/test_core/test_end2end_vis.py +++ b/tests/test_core/test_end2end_vis.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmocr.core import det_recog_show_result diff --git a/tests/test_dataset/test_base_dataset.py b/tests/test_dataset/test_base_dataset.py index 43260d37..b11aea00 100644 --- a/tests/test_dataset/test_base_dataset.py +++ b/tests/test_dataset/test_base_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/tests/test_dataset/test_crop.py b/tests/test_dataset/test_crop.py index 84842c2d..f1806198 100644 --- a/tests/test_dataset/test_crop.py +++ b/tests/test_dataset/test_crop.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math from itertools import chain, permutations diff --git a/tests/test_dataset/test_dbnet_transforms.py b/tests/test_dataset/test_dbnet_transforms.py index a8f18e38..71c1e1c9 100644 --- a/tests/test_dataset/test_dbnet_transforms.py +++ b/tests/test_dataset/test_dbnet_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import mmocr.datasets.pipelines.dbnet_transforms as transforms diff --git a/tests/test_dataset/test_detect_dataset.py b/tests/test_dataset/test_detect_dataset.py index 83480c45..b2015ba3 100644 --- a/tests/test_dataset/test_detect_dataset.py +++ b/tests/test_dataset/test_detect_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json import os.path as osp import tempfile diff --git a/tests/test_dataset/test_icdar_dataset.py b/tests/test_dataset/test_icdar_dataset.py index 7f6a1e7b..26a3307f 100644 --- a/tests/test_dataset/test_icdar_dataset.py +++ b/tests/test_dataset/test_icdar_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/tests/test_dataset/test_kie_dataset.py b/tests/test_dataset/test_kie_dataset.py index 4c2c5881..2291f355 100644 --- a/tests/test_dataset/test_kie_dataset.py +++ b/tests/test_dataset/test_kie_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json import math import os.path as osp diff --git a/tests/test_dataset/test_loader.py b/tests/test_dataset/test_loader.py index 41227b1b..19ce5ec9 100644 --- a/tests/test_dataset/test_loader.py +++ b/tests/test_dataset/test_loader.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json import os.path as osp import tempfile diff --git a/tests/test_dataset/test_loading.py b/tests/test_dataset/test_loading.py index 4b45de06..d4305f6d 100644 --- a/tests/test_dataset/test_loading.py +++ b/tests/test_dataset/test_loading.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmocr.datasets.pipelines import LoadImageFromNdarray, LoadTextAnnotations diff --git a/tests/test_dataset/test_ner_dataset.py b/tests/test_dataset/test_ner_dataset.py index f79ba7b9..ddc1174a 100644 --- a/tests/test_dataset/test_ner_dataset.py +++ b/tests/test_dataset/test_ner_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json import os.path as osp import tempfile diff --git a/tests/test_dataset/test_ocr_dataset.py b/tests/test_dataset/test_ocr_dataset.py index 1787db88..5d8d5dd3 100644 --- a/tests/test_dataset/test_ocr_dataset.py +++ b/tests/test_dataset/test_ocr_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import os.path as osp import tempfile diff --git a/tests/test_dataset/test_ocr_seg_dataset.py b/tests/test_dataset/test_ocr_seg_dataset.py index 0ecfcfdf..f7678123 100644 --- a/tests/test_dataset/test_ocr_seg_dataset.py +++ b/tests/test_dataset/test_ocr_seg_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json import math import os.path as osp diff --git a/tests/test_dataset/test_ocr_seg_target.py b/tests/test_dataset/test_ocr_seg_target.py index 45b85352..54f78bf0 100644 --- a/tests/test_dataset/test_ocr_seg_target.py +++ b/tests/test_dataset/test_ocr_seg_target.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/tests/test_dataset/test_ocr_transforms.py b/tests/test_dataset/test_ocr_transforms.py index 15522b25..612cea12 100644 --- a/tests/test_dataset/test_ocr_transforms.py +++ b/tests/test_dataset/test_ocr_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import unittest.mock as mock diff --git a/tests/test_dataset/test_parser.py b/tests/test_dataset/test_parser.py index e255a0c4..e20f3fbe 100644 --- a/tests/test_dataset/test_parser.py +++ b/tests/test_dataset/test_parser.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import json import pytest diff --git a/tests/test_dataset/test_test_time_aug.py b/tests/test_dataset/test_test_time_aug.py index 22bf80c6..5d68ac42 100644 --- a/tests/test_dataset/test_test_time_aug.py +++ b/tests/test_dataset/test_test_time_aug.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest diff --git a/tests/test_dataset/test_textdet_targets.py b/tests/test_dataset/test_textdet_targets.py index 16860997..f9c97bfe 100644 --- a/tests/test_dataset/test_textdet_targets.py +++ b/tests/test_dataset/test_textdet_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from unittest import mock import numpy as np diff --git a/tests/test_dataset/test_transforms.py b/tests/test_dataset/test_transforms.py index 26ecef0b..0008ac4b 100644 --- a/tests/test_dataset/test_transforms.py +++ b/tests/test_dataset/test_transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import unittest.mock as mock import numpy as np diff --git a/tests/test_dataset/test_uniform_concat_dataset.py b/tests/test_dataset/test_uniform_concat_dataset.py index c5d12d99..994ef394 100644 --- a/tests/test_dataset/test_uniform_concat_dataset.py +++ b/tests/test_dataset/test_uniform_concat_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmocr.datasets import UniformConcatDataset from mmocr.utils import list_from_file diff --git a/tests/test_metrics/test_eval_utils.py b/tests/test_metrics/test_eval_utils.py index a5ca2a35..50fbdd4d 100644 --- a/tests/test_metrics/test_eval_utils.py +++ b/tests/test_metrics/test_eval_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Tests the utils of evaluation.""" import numpy as np import pytest diff --git a/tests/test_metrics/test_hmean_detect.py b/tests/test_metrics/test_hmean_detect.py index 99c6f074..18bcda1e 100644 --- a/tests/test_metrics/test_hmean_detect.py +++ b/tests/test_metrics/test_hmean_detect.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile import numpy as np diff --git a/tests/test_metrics/test_hmean_ic13.py b/tests/test_metrics/test_hmean_ic13.py index c5b3e69c..ac02b38e 100644 --- a/tests/test_metrics/test_hmean_ic13.py +++ b/tests/test_metrics/test_hmean_ic13.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Test hmean_ic13.""" import math diff --git a/tests/test_metrics/test_hmean_iou.py b/tests/test_metrics/test_hmean_iou.py index 8be4ca50..6aa5eaa9 100644 --- a/tests/test_metrics/test_hmean_iou.py +++ b/tests/test_metrics/test_hmean_iou.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Test hmean_iou.""" import pytest diff --git a/tests/test_models/test_detector.py b/tests/test_models/test_detector.py index 3f538ffa..91522254 100644 --- a/tests/test_models/test_detector.py +++ b/tests/test_models/test_detector.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """pytest tests/test_detector.py.""" import copy import tempfile diff --git a/tests/test_models/test_kie_config.py b/tests/test_models/test_kie_config.py index c134e93a..b2b1f351 100644 --- a/tests/test_models/test_kie_config.py +++ b/tests/test_models/test_kie_config.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from os.path import dirname, exists, join diff --git a/tests/test_models/test_label_convertor/test_attn_label_convertor.py b/tests/test_models/test_label_convertor/test_attn_label_convertor.py index 00eaeacc..16049ba9 100644 --- a/tests/test_models/test_label_convertor/test_attn_label_convertor.py +++ b/tests/test_models/test_label_convertor/test_attn_label_convertor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/tests/test_models/test_label_convertor/test_ctc_label_convertor.py b/tests/test_models/test_label_convertor/test_ctc_label_convertor.py index 07c9cbf0..df677e68 100644 --- a/tests/test_models/test_label_convertor/test_ctc_label_convertor.py +++ b/tests/test_models/test_label_convertor/test_ctc_label_convertor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/tests/test_models/test_loss.py b/tests/test_models/test_loss.py index 5235b673..edef1202 100644 --- a/tests/test_models/test_loss.py +++ b/tests/test_models/test_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch from mmdet.core import BitmapMasks diff --git a/tests/test_models/test_modules.py b/tests/test_models/test_modules.py index 30a56661..9e19ea3b 100644 --- a/tests/test_models/test_modules.py +++ b/tests/test_models/test_modules.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch diff --git a/tests/test_models/test_ner_model.py b/tests/test_models/test_ner_model.py index d7a429ba..1fa68c9f 100644 --- a/tests/test_models/test_ner_model.py +++ b/tests/test_models/test_ner_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os.path as osp import tempfile diff --git a/tests/test_models/test_ocr_backbone.py b/tests/test_models/test_ocr_backbone.py index f49a334d..780bfa6a 100644 --- a/tests/test_models/test_ocr_backbone.py +++ b/tests/test_models/test_ocr_backbone.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_ocr_decoder.py b/tests/test_models/test_ocr_decoder.py index 709f86ed..5d41a2fc 100644 --- a/tests/test_models/test_ocr_decoder.py +++ b/tests/test_models/test_ocr_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import pytest diff --git a/tests/test_models/test_ocr_encoder.py b/tests/test_models/test_ocr_encoder.py index ac185eeb..ef16dac6 100644 --- a/tests/test_models/test_ocr_encoder.py +++ b/tests/test_models/test_ocr_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_ocr_head.py b/tests/test_models/test_ocr_head.py index 7df0f77b..761bd129 100644 --- a/tests/test_models/test_ocr_head.py +++ b/tests/test_models/test_ocr_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_ocr_layer.py b/tests/test_models/test_ocr_layer.py index f6e3d718..78cd60c2 100644 --- a/tests/test_models/test_ocr_layer.py +++ b/tests/test_models/test_ocr_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmocr.models.textrecog.layers import (BasicBlock, Bottleneck, diff --git a/tests/test_models/test_ocr_loss.py b/tests/test_models/test_ocr_loss.py index 51f84fb2..0bce289d 100644 --- a/tests/test_models/test_ocr_loss.py +++ b/tests/test_models/test_ocr_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_ocr_neck.py b/tests/test_models/test_ocr_neck.py index 28009311..3454eab3 100644 --- a/tests/test_models/test_ocr_neck.py +++ b/tests/test_models/test_ocr_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmocr.models.textrecog.necks import FPNOCR diff --git a/tests/test_models/test_ocr_preprocessor.py b/tests/test_models/test_ocr_preprocessor.py index 47bd985f..2a694e33 100644 --- a/tests/test_models/test_ocr_preprocessor.py +++ b/tests/test_models/test_ocr_preprocessor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_panhead.py b/tests/test_models/test_panhead.py index 5afbd6d1..a6b7bbb1 100644 --- a/tests/test_models/test_panhead.py +++ b/tests/test_models/test_panhead.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest diff --git a/tests/test_models/test_recog_config.py b/tests/test_models/test_recog_config.py index 19469bb5..2eb0a2db 100644 --- a/tests/test_models/test_recog_config.py +++ b/tests/test_models/test_recog_config.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from os.path import dirname, exists, join diff --git a/tests/test_models/test_recognizer.py b/tests/test_models/test_recognizer.py index 3270ad4d..3813e736 100644 --- a/tests/test_models/test_recognizer.py +++ b/tests/test_models/test_recognizer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile from functools import partial diff --git a/tests/test_models/test_targets.py b/tests/test_models/test_targets.py index d270a53a..6030a256 100644 --- a/tests/test_models/test_targets.py +++ b/tests/test_models/test_targets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmocr.datasets.pipelines.textdet_targets.dbnet_targets import DBNetTargets diff --git a/tests/test_models/test_textdet_head.py b/tests/test_models/test_textdet_head.py index 0db1b642..5cb81bca 100644 --- a/tests/test_models/test_textdet_head.py +++ b/tests/test_models/test_textdet_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch diff --git a/tests/test_models/test_textdet_neck.py b/tests/test_models/test_textdet_neck.py index d41ac420..7bee9d7e 100644 --- a/tests/test_models/test_textdet_neck.py +++ b/tests/test_models/test_textdet_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_tools/test_data_converter.py b/tests/test_tools/test_data_converter.py index 3677c91b..76ff0047 100644 --- a/tests/test_tools/test_data_converter.py +++ b/tests/test_tools/test_data_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Test orientation check and ignore method.""" import shutil diff --git a/tests/test_utils/test_box.py b/tests/test_utils/test_box.py index 49953726..6ba52cd6 100644 --- a/tests/test_utils/test_box.py +++ b/tests/test_utils/test_box.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmocr.utils import is_on_same_line, stitch_boxes_into_lines diff --git a/tests/test_utils/test_check_argument.py b/tests/test_utils/test_check_argument.py index fd247353..7cbbbb54 100644 --- a/tests/test_utils/test_check_argument.py +++ b/tests/test_utils/test_check_argument.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import mmocr.utils as utils diff --git a/tests/test_utils/test_mask/test_mask_utils.py b/tests/test_utils/test_mask/test_mask_utils.py index a1f2b0dd..12319bbb 100644 --- a/tests/test_utils/test_mask/test_mask_utils.py +++ b/tests/test_utils/test_mask/test_mask_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Test text mask_utils.""" import tempfile from unittest import mock diff --git a/tests/test_utils/test_model.py b/tests/test_utils/test_model.py index 39d6df13..fd5044c4 100644 --- a/tests/test_utils/test_model.py +++ b/tests/test_utils/test_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch from mmcv.cnn.bricks import ConvModule diff --git a/tests/test_utils/test_ocr.py b/tests/test_utils/test_ocr.py index b38fbf9e..71fe74d2 100644 --- a/tests/test_utils/test_ocr.py +++ b/tests/test_utils/test_ocr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import io import json import os diff --git a/tests/test_utils/test_string_util.py b/tests/test_utils/test_string_util.py index ab6f3868..c0eb4678 100644 --- a/tests/test_utils/test_string_util.py +++ b/tests/test_utils/test_string_util.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest from mmocr.utils import StringStrip diff --git a/tests/test_utils/test_text/test_text_utils.py b/tests/test_utils/test_text/test_text_utils.py index e0925c27..aa3b2d24 100644 --- a/tests/test_utils/test_text/test_text_utils.py +++ b/tests/test_utils/test_text/test_text_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Test text label visualize.""" import os.path as osp import random diff --git a/tests/test_utils/test_textio.py b/tests/test_utils/test_textio.py index 2b504dfd..c017dec4 100644 --- a/tests/test_utils/test_textio.py +++ b/tests/test_utils/test_textio.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile from mmocr.utils import list_from_file, list_to_file diff --git a/tests/test_utils/test_version_utils.py b/tests/test_utils/test_version_utils.py index 99e37635..ad43344d 100644 --- a/tests/test_utils/test_version_utils.py +++ b/tests/test_utils/test_version_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmocr import digit_version diff --git a/tests/test_utils/test_wrapper.py b/tests/test_utils/test_wrapper.py index d92ec175..a890cdab 100644 --- a/tests/test_utils/test_wrapper.py +++ b/tests/test_utils/test_wrapper.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest import torch diff --git a/tools/benchmark_processing.py b/tools/benchmark_processing.py index 8bd2c779..13b215ef 100755 --- a/tools/benchmark_processing.py +++ b/tools/benchmark_processing.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. """This file is for benchmark data loading process. It can also be used to refresh the memcached cache. The command line to run this file is: diff --git a/tools/data/textdet/coco_to_line_dict.py b/tools/data/textdet/coco_to_line_dict.py index ed81d403..b8d0583e 100644 --- a/tools/data/textdet/coco_to_line_dict.py +++ b/tools/data/textdet/coco_to_line_dict.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import json diff --git a/tools/data/textdet/ctw1500_converter.py b/tools/data/textdet/ctw1500_converter.py index 06941deb..cfe5ce7b 100644 --- a/tools/data/textdet/ctw1500_converter.py +++ b/tools/data/textdet/ctw1500_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import glob import os.path as osp diff --git a/tools/data/textdet/icdar_converter.py b/tools/data/textdet/icdar_converter.py index 28eaac0a..e478f8c6 100644 --- a/tools/data/textdet/icdar_converter.py +++ b/tools/data/textdet/icdar_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import glob import os.path as osp diff --git a/tools/data/textdet/synthtext_converter.py b/tools/data/textdet/synthtext_converter.py index 2b6cb912..9c796402 100644 --- a/tools/data/textdet/synthtext_converter.py +++ b/tools/data/textdet/synthtext_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import json import os.path as osp diff --git a/tools/data/textdet/textocr_converter.py b/tools/data/textdet/textocr_converter.py index b95f480f..50b6a62a 100644 --- a/tools/data/textdet/textocr_converter.py +++ b/tools/data/textdet/textocr_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import math import os.path as osp diff --git a/tools/data/textdet/totaltext_converter.py b/tools/data/textdet/totaltext_converter.py index 865a7bc7..3a49e864 100644 --- a/tools/data/textdet/totaltext_converter.py +++ b/tools/data/textdet/totaltext_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import glob import os diff --git a/tools/data/textrecog/seg_synthtext_converter.py b/tools/data/textrecog/seg_synthtext_converter.py index 14ec35f4..2d3e1928 100644 --- a/tools/data/textrecog/seg_synthtext_converter.py +++ b/tools/data/textrecog/seg_synthtext_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import json import os.path as osp diff --git a/tools/data/textrecog/svt_converter.py b/tools/data/textrecog/svt_converter.py index 35e4d7f0..0ecb3e73 100644 --- a/tools/data/textrecog/svt_converter.py +++ b/tools/data/textrecog/svt_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/data/textrecog/synthtext_converter.py b/tools/data/textrecog/synthtext_converter.py index d9f8a909..aa004926 100644 --- a/tools/data/textrecog/synthtext_converter.py +++ b/tools/data/textrecog/synthtext_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os from functools import partial diff --git a/tools/data/textrecog/textocr_converter.py b/tools/data/textrecog/textocr_converter.py index 42996acb..2c161788 100644 --- a/tools/data/textrecog/textocr_converter.py +++ b/tools/data/textrecog/textocr_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import math import os diff --git a/tools/data/textrecog/totaltext_converter.py b/tools/data/textrecog/totaltext_converter.py index 090966f8..3bf0ff39 100644 --- a/tools/data/textrecog/totaltext_converter.py +++ b/tools/data/textrecog/totaltext_converter.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import glob import os diff --git a/tools/data/utils/txt2lmdb.py b/tools/data/utils/txt2lmdb.py index e97bfda0..2b83dc26 100644 --- a/tools/data/utils/txt2lmdb.py +++ b/tools/data/utils/txt2lmdb.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse from mmocr.utils import lmdb_converter diff --git a/tools/deployment/deploy_test.py b/tools/deployment/deploy_test.py index ca387f6d..33580663 100644 --- a/tools/deployment/deploy_test.py +++ b/tools/deployment/deploy_test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse from mmcv import Config diff --git a/tools/deployment/onnx2tensorrt.py b/tools/deployment/onnx2tensorrt.py index 3a842595..42495494 100644 --- a/tools/deployment/onnx2tensorrt.py +++ b/tools/deployment/onnx2tensorrt.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/deployment/pytorch2onnx.py b/tools/deployment/pytorch2onnx.py index f0c6fc8c..abbb87b2 100644 --- a/tools/deployment/pytorch2onnx.py +++ b/tools/deployment/pytorch2onnx.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from argparse import ArgumentParser from functools import partial diff --git a/tools/det_test_imgs.py b/tools/det_test_imgs.py index 008075d3..bdb3fa60 100755 --- a/tools/det_test_imgs.py +++ b/tools/det_test_imgs.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from argparse import ArgumentParser diff --git a/tools/kie_test_imgs.py b/tools/kie_test_imgs.py index 0b83eeb4..d6a4f553 100755 --- a/tools/kie_test_imgs.py +++ b/tools/kie_test_imgs.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import argparse import ast import os diff --git a/tools/publish_model.py b/tools/publish_model.py index 525ccade..93b704f5 100755 --- a/tools/publish_model.py +++ b/tools/publish_model.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import argparse import subprocess diff --git a/tools/recog_test_imgs.py b/tools/recog_test_imgs.py index 51a85a46..657ffdcf 100755 --- a/tools/recog_test_imgs.py +++ b/tools/recog_test_imgs.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import shutil import time diff --git a/tools/test.py b/tools/test.py index 66084256..c5fde264 100755 --- a/tools/test.py +++ b/tools/test.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import warnings diff --git a/tools/train.py b/tools/train.py index dde875fc..bb84e3ea 100755 --- a/tools/train.py +++ b/tools/train.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import argparse import copy import os