[Enhancement] Add copyright info (#439)

* add copyright info

demo/batch_image_demo.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from argparse import ArgumentParser
 from pathlib import Path
 

demo/image_demo.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from argparse import ArgumentParser
 
 import mmcv

demo/ner_demo.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from argparse import ArgumentParser
 
 from mmdet.apis import init_detector

demo/ocr_image_demo.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from argparse import ArgumentParser
 
 import mmcv

demo/webcam_demo.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import argparse
 
 import cv2

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

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

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

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

mmocr/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import warnings
 
 import mmcv

mmocr/apis/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .inference import model_inference
 from .train import train_detector
 

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

mmocr/apis/train.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import warnings
 
 import torch

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,

mmocr/core/deployment/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .deploy_utils import (ONNXRuntimeDetector, ONNXRuntimeRecognizer,
                            TensorRTDetector, TensorRTRecognizer)
 

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

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

mmocr/core/evaluation/hmean.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from operator import itemgetter
 
 import mmcv

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

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

mmocr/core/evaluation/kie_metric.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import torch
 
 

mmocr/core/evaluation/ner_metric.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from collections import Counter
 
 

mmocr/core/evaluation/ocr_metric.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import re
 from difflib import SequenceMatcher
 

mmocr/core/evaluation/utils.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import numpy as np
 import Polygon as plg
 

mmocr/core/mask.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import cv2
 import numpy as np
 

mmocr/core/visualize.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import math
 import os
 import shutil

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

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

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')

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

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

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

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

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

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

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

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:

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

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

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

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

mmocr/datasets/pipelines/ner_transforms.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import torch
 from mmdet.datasets.builder import PIPELINES
 

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

mmocr/datasets/pipelines/ocr_transforms.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import math
 
 import mmcv

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

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

mmocr/datasets/pipelines/textdet_targets/base_textdet_targets.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import sys
 
 import cv2

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

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

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

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
 

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

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

mmocr/datasets/pipelines/transforms.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import math
 
 import cv2

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
 

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

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
 

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

mmocr/datasets/utils/parser.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import json
 
 from mmocr.datasets.builder import PARSERS

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)
 

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')

mmocr/models/common/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from . import backbones, losses
 
 from .backbones import *  # NOQA

mmocr/models/common/backbones/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .unet import UNet
 
 __all__ = ['UNet']

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

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
 

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

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

mmocr/models/kie/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from . import extractors, heads, losses
 
 from .extractors import *  # NOQA

mmocr/models/kie/extractors/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .sdmgr import SDMGR
 
 __all__ = ['SDMGR']

mmocr/models/kie/extractors/sdmgr.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 import warnings
 
 import mmcv

mmocr/models/kie/heads/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .sdmgr_head import SDMGRHead
 
 __all__ = ['SDMGRHead']

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

mmocr/models/kie/losses/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .sdmgr_loss import SDMGRLoss
 
 __all__ = ['SDMGRLoss']

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

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

mmocr/models/ner/classifiers/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .ner_classifier import NerClassifier
 
 __all__ = ['NerClassifier']

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

mmocr/models/ner/convertors/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .ner_convertor import NerConvertor
 
 __all__ = ['NerConvertor']

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

mmocr/models/ner/decoders/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .fc_decoder import FCDecoder
 
 __all__ = ['FCDecoder']

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

mmocr/models/ner/encoders/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .bert_encoder import BertEncoder
 
 __all__ = ['BertEncoder']

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

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
 

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

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
 

mmocr/models/ner/utils/__init__.py

@@ -1,3 +1,4 @@
+# Copyright (c) OpenMMLab. All rights reserved.
 from .activations import GeluNew
 from .bert import BertModel
 

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

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

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

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

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

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
 

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

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

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

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

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 \

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 \

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