From 441be4e435127868a0c72a4e0e6b87662a4c415b Mon Sep 17 00:00:00 2001 From: Junjun2016 Date: Tue, 17 Aug 2021 14:16:55 +0800 Subject: [PATCH] [Dcos] Add header for files (#796) * Add header for files * Delete header in config files --- .dev/gather_models.py | 1 + .dev/md2yml.py | 1 + .dev/upload_modelzoo.py | 1 + demo/image_demo.py | 1 + docs/conf.py | 1 + docs/stat.py | 1 + docs_zh-CN/conf.py | 1 + docs_zh-CN/stat.py | 1 + mmseg/__init__.py | 1 + mmseg/apis/__init__.py | 1 + mmseg/apis/inference.py | 1 + mmseg/apis/test.py | 1 + mmseg/apis/train.py | 1 + mmseg/core/__init__.py | 1 + mmseg/core/evaluation/__init__.py | 1 + mmseg/core/evaluation/class_names.py | 1 + mmseg/core/evaluation/eval_hooks.py | 1 + mmseg/core/evaluation/metrics.py | 1 + mmseg/core/seg/__init__.py | 1 + mmseg/core/seg/builder.py | 1 + mmseg/core/seg/sampler/__init__.py | 1 + mmseg/core/seg/sampler/base_pixel_sampler.py | 1 + mmseg/core/seg/sampler/ohem_pixel_sampler.py | 1 + mmseg/core/utils/__init__.py | 1 + mmseg/core/utils/misc.py | 1 + mmseg/datasets/__init__.py | 1 + mmseg/datasets/ade.py | 1 + mmseg/datasets/builder.py | 1 + mmseg/datasets/chase_db1.py | 1 + mmseg/datasets/cityscapes.py | 1 + mmseg/datasets/custom.py | 1 + mmseg/datasets/dataset_wrappers.py | 1 + mmseg/datasets/drive.py | 1 + mmseg/datasets/hrf.py | 1 + mmseg/datasets/pascal_context.py | 1 + mmseg/datasets/pipelines/__init__.py | 1 + mmseg/datasets/pipelines/compose.py | 1 + mmseg/datasets/pipelines/formating.py | 1 + mmseg/datasets/pipelines/loading.py | 1 + mmseg/datasets/pipelines/test_time_aug.py | 1 + mmseg/datasets/pipelines/transforms.py | 1 + mmseg/datasets/stare.py | 1 + mmseg/datasets/voc.py | 1 + mmseg/models/__init__.py | 1 + mmseg/models/backbones/__init__.py | 1 + mmseg/models/backbones/cgnet.py | 1 + mmseg/models/backbones/fast_scnn.py | 1 + mmseg/models/backbones/hrnet.py | 1 + mmseg/models/backbones/mit.py | 1 + mmseg/models/backbones/mobilenet_v2.py | 1 + mmseg/models/backbones/mobilenet_v3.py | 1 + mmseg/models/backbones/resnest.py | 1 + mmseg/models/backbones/resnet.py | 1 + mmseg/models/backbones/resnext.py | 1 + mmseg/models/backbones/swin.py | 1 + mmseg/models/backbones/unet.py | 1 + mmseg/models/backbones/vit.py | 1 + mmseg/models/builder.py | 1 + mmseg/models/decode_heads/__init__.py | 1 + mmseg/models/decode_heads/ann_head.py | 1 + mmseg/models/decode_heads/apc_head.py | 1 + mmseg/models/decode_heads/aspp_head.py | 1 + mmseg/models/decode_heads/cascade_decode_head.py | 1 + mmseg/models/decode_heads/cc_head.py | 1 + mmseg/models/decode_heads/da_head.py | 1 + mmseg/models/decode_heads/decode_head.py | 1 + mmseg/models/decode_heads/dm_head.py | 1 + mmseg/models/decode_heads/dnl_head.py | 1 + mmseg/models/decode_heads/ema_head.py | 1 + mmseg/models/decode_heads/enc_head.py | 1 + mmseg/models/decode_heads/fcn_head.py | 1 + mmseg/models/decode_heads/fpn_head.py | 1 + mmseg/models/decode_heads/gc_head.py | 1 + mmseg/models/decode_heads/lraspp_head.py | 1 + mmseg/models/decode_heads/nl_head.py | 1 + mmseg/models/decode_heads/ocr_head.py | 1 + mmseg/models/decode_heads/point_head.py | 1 + mmseg/models/decode_heads/psa_head.py | 1 + mmseg/models/decode_heads/psp_head.py | 1 + mmseg/models/decode_heads/segformer_head.py | 1 + mmseg/models/decode_heads/sep_aspp_head.py | 1 + mmseg/models/decode_heads/sep_fcn_head.py | 1 + mmseg/models/decode_heads/setr_mla_head.py | 1 + mmseg/models/decode_heads/setr_up_head.py | 1 + mmseg/models/decode_heads/uper_head.py | 1 + mmseg/models/losses/__init__.py | 1 + mmseg/models/losses/accuracy.py | 1 + mmseg/models/losses/cross_entropy_loss.py | 1 + mmseg/models/losses/dice_loss.py | 1 + mmseg/models/losses/lovasz_loss.py | 1 + mmseg/models/losses/utils.py | 1 + mmseg/models/necks/__init__.py | 1 + mmseg/models/necks/fpn.py | 1 + mmseg/models/necks/mla_neck.py | 1 + mmseg/models/necks/multilevel_neck.py | 1 + mmseg/models/segmentors/__init__.py | 1 + mmseg/models/segmentors/base.py | 1 + mmseg/models/segmentors/cascade_encoder_decoder.py | 1 + mmseg/models/segmentors/encoder_decoder.py | 1 + mmseg/models/utils/__init__.py | 1 + mmseg/models/utils/ckpt_convert.py | 1 + mmseg/models/utils/embed.py | 1 + mmseg/models/utils/inverted_residual.py | 1 + mmseg/models/utils/make_divisible.py | 1 + mmseg/models/utils/res_layer.py | 1 + mmseg/models/utils/se_layer.py | 1 + mmseg/models/utils/self_attention_block.py | 1 + mmseg/models/utils/shape_convert.py | 1 + mmseg/models/utils/up_conv_block.py | 1 + mmseg/ops/__init__.py | 1 + mmseg/ops/encoding.py | 1 + mmseg/ops/wrappers.py | 1 + mmseg/utils/__init__.py | 1 + mmseg/utils/collect_env.py | 1 + mmseg/utils/logger.py | 1 + setup.py | 3 ++- tests/__init__.py | 1 + tests/test_config.py | 1 + tests/test_data/test_dataset.py | 1 + tests/test_data/test_dataset_builder.py | 1 + tests/test_data/test_loading.py | 1 + tests/test_data/test_transform.py | 1 + tests/test_data/test_tta.py | 1 + tests/test_digit_version.py | 1 + tests/test_eval_hook.py | 1 + tests/test_inference.py | 1 + tests/test_metrics.py | 1 + tests/test_models/__init__.py | 1 + tests/test_models/test_backbones/__init__.py | 1 + tests/test_models/test_backbones/test_blocks.py | 1 + tests/test_models/test_backbones/test_cgnet.py | 1 + tests/test_models/test_backbones/test_fast_scnn.py | 1 + tests/test_models/test_backbones/test_hrnet.py | 1 + tests/test_models/test_backbones/test_mit.py | 1 + tests/test_models/test_backbones/test_mobilenet_v3.py | 1 + tests/test_models/test_backbones/test_resnest.py | 1 + tests/test_models/test_backbones/test_resnet.py | 1 + tests/test_models/test_backbones/test_resnext.py | 1 + tests/test_models/test_backbones/test_swin.py | 1 + tests/test_models/test_backbones/test_unet.py | 1 + tests/test_models/test_backbones/test_vit.py | 1 + tests/test_models/test_backbones/utils.py | 1 + tests/test_models/test_forward.py | 1 + tests/test_models/test_heads/__init__.py | 1 + tests/test_models/test_heads/test_ann_head.py | 1 + tests/test_models/test_heads/test_apc_head.py | 1 + tests/test_models/test_heads/test_aspp_head.py | 1 + tests/test_models/test_heads/test_cc_head.py | 1 + tests/test_models/test_heads/test_da_head.py | 1 + tests/test_models/test_heads/test_decode_head.py | 1 + tests/test_models/test_heads/test_dm_head.py | 1 + tests/test_models/test_heads/test_dnl_head.py | 1 + tests/test_models/test_heads/test_ema_head.py | 1 + tests/test_models/test_heads/test_enc_head.py | 1 + tests/test_models/test_heads/test_fcn_head.py | 1 + tests/test_models/test_heads/test_gc_head.py | 1 + tests/test_models/test_heads/test_lraspp_head.py | 1 + tests/test_models/test_heads/test_nl_head.py | 1 + tests/test_models/test_heads/test_ocr_head.py | 1 + tests/test_models/test_heads/test_point_head.py | 1 + tests/test_models/test_heads/test_psa_head.py | 1 + tests/test_models/test_heads/test_psp_head.py | 1 + tests/test_models/test_heads/test_segformer_head.py | 1 + tests/test_models/test_heads/test_setr_mla_head.py | 1 + tests/test_models/test_heads/test_setr_up_head.py | 1 + tests/test_models/test_heads/test_uper_head.py | 1 + tests/test_models/test_heads/utils.py | 1 + tests/test_models/test_losses/__init__.py | 1 + tests/test_models/test_losses/test_ce_loss.py | 1 + tests/test_models/test_losses/test_dice_loss.py | 1 + tests/test_models/test_losses/test_lovasz_loss.py | 1 + tests/test_models/test_losses/test_utils.py | 1 + tests/test_models/test_necks/__init__.py | 1 + tests/test_models/test_necks/test_fpn.py | 1 + tests/test_models/test_necks/test_mla_neck.py | 1 + tests/test_models/test_necks/test_multilevel_neck.py | 1 + tests/test_models/test_segmentors/__init__.py | 1 + .../test_segmentors/test_cascade_encoder_decoder.py | 1 + tests/test_models/test_segmentors/test_encoder_decoder.py | 1 + tests/test_models/test_segmentors/utils.py | 1 + tests/test_sampler.py | 1 + tools/analyze_logs.py | 1 + tools/benchmark.py | 1 + tools/convert_datasets/chase_db1.py | 1 + tools/convert_datasets/cityscapes.py | 1 + tools/convert_datasets/drive.py | 1 + tools/convert_datasets/hrf.py | 1 + tools/convert_datasets/pascal_context.py | 1 + tools/convert_datasets/stare.py | 1 + tools/convert_datasets/voc_aug.py | 1 + tools/deploy_test.py | 1 + tools/get_flops.py | 1 + tools/mmseg2torchserve.py | 1 + tools/mmseg_handler.py | 1 + tools/model_converters/mit_convert.py | 1 + tools/onnx2tensorrt.py | 1 + tools/print_config.py | 1 + tools/publish_model.py | 1 + tools/pytorch2onnx.py | 1 + tools/pytorch2torchscript.py | 1 + tools/test.py | 1 + tools/train.py | 1 + 202 files changed, 203 insertions(+), 1 deletion(-) diff --git a/.dev/gather_models.py b/.dev/gather_models.py index 1899195..0db26a5 100644 --- a/.dev/gather_models.py +++ b/.dev/gather_models.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import glob import json diff --git a/.dev/md2yml.py b/.dev/md2yml.py index 36c82ff..3f118c1 100755 --- a/.dev/md2yml.py +++ b/.dev/md2yml.py @@ -1,5 +1,6 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. # This tool is used to update model-index.yml which is required by MIM, and # will be automatically called as a pre-commit hook. The updating will be # triggered if any change of model information (.md files in configs/) has been diff --git a/.dev/upload_modelzoo.py b/.dev/upload_modelzoo.py index bd78bc4..303c80d 100644 --- a/.dev/upload_modelzoo.py +++ b/.dev/upload_modelzoo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/demo/image_demo.py b/demo/image_demo.py index 2698e42..05e1a79 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 from mmseg.apis import inference_segmentor, init_segmentor, show_result_pyplot diff --git a/docs/conf.py b/docs/conf.py index 758b5ff..aaea424 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/stat.py b/docs/stat.py index 941296d..eef00af 100755 --- a/docs/stat.py +++ b/docs/stat.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import functools as func import glob import os.path as osp diff --git a/docs_zh-CN/conf.py b/docs_zh-CN/conf.py index 72c8c52..ed5eb52 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/stat.py b/docs_zh-CN/stat.py index dc7c90f..955bee8 100755 --- a/docs_zh-CN/stat.py +++ b/docs_zh-CN/stat.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. import functools as func import glob import os.path as osp diff --git a/mmseg/__init__.py b/mmseg/__init__.py index 317622c..08c8102 100644 --- a/mmseg/__init__.py +++ b/mmseg/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import mmcv diff --git a/mmseg/apis/__init__.py b/mmseg/apis/__init__.py index 170724b..ba5ab77 100644 --- a/mmseg/apis/__init__.py +++ b/mmseg/apis/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .inference import inference_segmentor, init_segmentor, show_result_pyplot from .test import multi_gpu_test, single_gpu_test from .train import get_root_logger, set_random_seed, train_segmentor diff --git a/mmseg/apis/inference.py b/mmseg/apis/inference.py index bf875cb..9069438 100644 --- a/mmseg/apis/inference.py +++ b/mmseg/apis/inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import matplotlib.pyplot as plt import mmcv import torch diff --git a/mmseg/apis/test.py b/mmseg/apis/test.py index 0034159..fb0bb93 100644 --- a/mmseg/apis/test.py +++ b/mmseg/apis/test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/mmseg/apis/train.py b/mmseg/apis/train.py index e1e771b..fe85e91 100644 --- a/mmseg/apis/train.py +++ b/mmseg/apis/train.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import random import warnings diff --git a/mmseg/core/__init__.py b/mmseg/core/__init__.py index 9656055..4022786 100644 --- a/mmseg/core/__init__.py +++ b/mmseg/core/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .evaluation import * # noqa: F401, F403 from .seg import * # noqa: F401, F403 from .utils import * # noqa: F401, F403 diff --git a/mmseg/core/evaluation/__init__.py b/mmseg/core/evaluation/__init__.py index f7cc4b2..237cf24 100644 --- a/mmseg/core/evaluation/__init__.py +++ b/mmseg/core/evaluation/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .class_names import get_classes, get_palette from .eval_hooks import DistEvalHook, EvalHook from .metrics import eval_metrics, mean_dice, mean_fscore, mean_iou diff --git a/mmseg/core/evaluation/class_names.py b/mmseg/core/evaluation/class_names.py index 0d8e66d..4527fba 100644 --- a/mmseg/core/evaluation/class_names.py +++ b/mmseg/core/evaluation/class_names.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv diff --git a/mmseg/core/evaluation/eval_hooks.py b/mmseg/core/evaluation/eval_hooks.py index 928f2ba..a2f08d7 100644 --- a/mmseg/core/evaluation/eval_hooks.py +++ b/mmseg/core/evaluation/eval_hooks.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import torch.distributed as dist diff --git a/mmseg/core/evaluation/metrics.py b/mmseg/core/evaluation/metrics.py index a216afe..3c5f63f 100644 --- a/mmseg/core/evaluation/metrics.py +++ b/mmseg/core/evaluation/metrics.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections import OrderedDict import mmcv diff --git a/mmseg/core/seg/__init__.py b/mmseg/core/seg/__init__.py index 93bc129..5206b96 100644 --- a/mmseg/core/seg/__init__.py +++ b/mmseg/core/seg/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .builder import build_pixel_sampler from .sampler import BasePixelSampler, OHEMPixelSampler diff --git a/mmseg/core/seg/builder.py b/mmseg/core/seg/builder.py index f5a117c..1cecd34 100644 --- a/mmseg/core/seg/builder.py +++ b/mmseg/core/seg/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import Registry, build_from_cfg PIXEL_SAMPLERS = Registry('pixel sampler') diff --git a/mmseg/core/seg/sampler/__init__.py b/mmseg/core/seg/sampler/__init__.py index 332b242..5a76485 100644 --- a/mmseg/core/seg/sampler/__init__.py +++ b/mmseg/core/seg/sampler/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_pixel_sampler import BasePixelSampler from .ohem_pixel_sampler import OHEMPixelSampler diff --git a/mmseg/core/seg/sampler/base_pixel_sampler.py b/mmseg/core/seg/sampler/base_pixel_sampler.py index b75b156..03672cd 100644 --- a/mmseg/core/seg/sampler/base_pixel_sampler.py +++ b/mmseg/core/seg/sampler/base_pixel_sampler.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from abc import ABCMeta, abstractmethod diff --git a/mmseg/core/seg/sampler/ohem_pixel_sampler.py b/mmseg/core/seg/sampler/ohem_pixel_sampler.py index 88bb10d..bcd481a 100644 --- a/mmseg/core/seg/sampler/ohem_pixel_sampler.py +++ b/mmseg/core/seg/sampler/ohem_pixel_sampler.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F diff --git a/mmseg/core/utils/__init__.py b/mmseg/core/utils/__init__.py index f2678b3..be9de55 100644 --- a/mmseg/core/utils/__init__.py +++ b/mmseg/core/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .misc import add_prefix __all__ = ['add_prefix'] diff --git a/mmseg/core/utils/misc.py b/mmseg/core/utils/misc.py index eb862a8..282bb8d 100644 --- a/mmseg/core/utils/misc.py +++ b/mmseg/core/utils/misc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. def add_prefix(inputs, prefix): """Add prefix for dict. diff --git a/mmseg/datasets/__init__.py b/mmseg/datasets/__init__.py index ebeaef4..bdea832 100644 --- a/mmseg/datasets/__init__.py +++ b/mmseg/datasets/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .ade import ADE20KDataset from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset from .chase_db1 import ChaseDB1Dataset diff --git a/mmseg/datasets/ade.py b/mmseg/datasets/ade.py index 5daf7e3..9af4371 100644 --- a/mmseg/datasets/ade.py +++ b/mmseg/datasets/ade.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/mmseg/datasets/builder.py b/mmseg/datasets/builder.py index 82f6f46..bfb54ef 100644 --- a/mmseg/datasets/builder.py +++ b/mmseg/datasets/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import platform import random diff --git a/mmseg/datasets/chase_db1.py b/mmseg/datasets/chase_db1.py index 8bc29be..7f14b2d 100644 --- a/mmseg/datasets/chase_db1.py +++ b/mmseg/datasets/chase_db1.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .builder import DATASETS diff --git a/mmseg/datasets/cityscapes.py b/mmseg/datasets/cityscapes.py index fa9958a..fd814f9 100644 --- a/mmseg/datasets/cityscapes.py +++ b/mmseg/datasets/cityscapes.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile diff --git a/mmseg/datasets/custom.py b/mmseg/datasets/custom.py index 9c88235..719ca01 100644 --- a/mmseg/datasets/custom.py +++ b/mmseg/datasets/custom.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import os.path as osp from collections import OrderedDict diff --git a/mmseg/datasets/dataset_wrappers.py b/mmseg/datasets/dataset_wrappers.py index d6a5e95..f161f71 100644 --- a/mmseg/datasets/dataset_wrappers.py +++ b/mmseg/datasets/dataset_wrappers.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from torch.utils.data.dataset import ConcatDataset as _ConcatDataset from .builder import DATASETS diff --git a/mmseg/datasets/drive.py b/mmseg/datasets/drive.py index 3cbfda8..6509911 100644 --- a/mmseg/datasets/drive.py +++ b/mmseg/datasets/drive.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .builder import DATASETS diff --git a/mmseg/datasets/hrf.py b/mmseg/datasets/hrf.py index 923203b..e4e10ae 100644 --- a/mmseg/datasets/hrf.py +++ b/mmseg/datasets/hrf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .builder import DATASETS diff --git a/mmseg/datasets/pascal_context.py b/mmseg/datasets/pascal_context.py index 541a63c..1e7a09d 100644 --- a/mmseg/datasets/pascal_context.py +++ b/mmseg/datasets/pascal_context.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .builder import DATASETS diff --git a/mmseg/datasets/pipelines/__init__.py b/mmseg/datasets/pipelines/__init__.py index 8b9046b..660291e 100644 --- a/mmseg/datasets/pipelines/__init__.py +++ b/mmseg/datasets/pipelines/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .compose import Compose from .formating import (Collect, ImageToTensor, ToDataContainer, ToTensor, Transpose, to_tensor) diff --git a/mmseg/datasets/pipelines/compose.py b/mmseg/datasets/pipelines/compose.py index ca48f1c..30280c1 100644 --- a/mmseg/datasets/pipelines/compose.py +++ b/mmseg/datasets/pipelines/compose.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import collections from mmcv.utils import build_from_cfg diff --git a/mmseg/datasets/pipelines/formating.py b/mmseg/datasets/pipelines/formating.py index 34061c1..45824fc 100644 --- a/mmseg/datasets/pipelines/formating.py +++ b/mmseg/datasets/pipelines/formating.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections.abc import Sequence import mmcv diff --git a/mmseg/datasets/pipelines/loading.py b/mmseg/datasets/pipelines/loading.py index fdfc496..e1c82bd 100644 --- a/mmseg/datasets/pipelines/loading.py +++ b/mmseg/datasets/pipelines/loading.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/mmseg/datasets/pipelines/test_time_aug.py b/mmseg/datasets/pipelines/test_time_aug.py index 473a12b..5c17cbb 100644 --- a/mmseg/datasets/pipelines/test_time_aug.py +++ b/mmseg/datasets/pipelines/test_time_aug.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import mmcv diff --git a/mmseg/datasets/pipelines/transforms.py b/mmseg/datasets/pipelines/transforms.py index c5e94a0..f2a642c 100644 --- a/mmseg/datasets/pipelines/transforms.py +++ b/mmseg/datasets/pipelines/transforms.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import numpy as np from mmcv.utils import deprecated_api_warning, is_tuple_of diff --git a/mmseg/datasets/stare.py b/mmseg/datasets/stare.py index cbd14e0..a24d1d9 100644 --- a/mmseg/datasets/stare.py +++ b/mmseg/datasets/stare.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .builder import DATASETS diff --git a/mmseg/datasets/voc.py b/mmseg/datasets/voc.py index a885520..3cec9e3 100644 --- a/mmseg/datasets/voc.py +++ b/mmseg/datasets/voc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .builder import DATASETS diff --git a/mmseg/models/__init__.py b/mmseg/models/__init__.py index 3cf93f8..87d8108 100644 --- a/mmseg/models/__init__.py +++ b/mmseg/models/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .backbones import * # noqa: F401,F403 from .builder import (BACKBONES, HEADS, LOSSES, SEGMENTORS, build_backbone, build_head, build_loss, build_segmentor) diff --git a/mmseg/models/backbones/__init__.py b/mmseg/models/backbones/__init__.py index b8c17b2..75ef2c3 100644 --- a/mmseg/models/backbones/__init__.py +++ b/mmseg/models/backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .cgnet import CGNet from .fast_scnn import FastSCNN from .hrnet import HRNet diff --git a/mmseg/models/backbones/cgnet.py b/mmseg/models/backbones/cgnet.py index 32bdbc4..67c0671 100644 --- a/mmseg/models/backbones/cgnet.py +++ b/mmseg/models/backbones/cgnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch diff --git a/mmseg/models/backbones/fast_scnn.py b/mmseg/models/backbones/fast_scnn.py index 84289da..95a4344 100644 --- a/mmseg/models/backbones/fast_scnn.py +++ b/mmseg/models/backbones/fast_scnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, DepthwiseSeparableConvModule diff --git a/mmseg/models/backbones/hrnet.py b/mmseg/models/backbones/hrnet.py index 0f064cf..c8ec506 100644 --- a/mmseg/models/backbones/hrnet.py +++ b/mmseg/models/backbones/hrnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch.nn as nn diff --git a/mmseg/models/backbones/mit.py b/mmseg/models/backbones/mit.py index 9d41ea5..90abfe5 100644 --- a/mmseg/models/backbones/mit.py +++ b/mmseg/models/backbones/mit.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import warnings diff --git a/mmseg/models/backbones/mobilenet_v2.py b/mmseg/models/backbones/mobilenet_v2.py index 46d57fb..988e29c 100644 --- a/mmseg/models/backbones/mobilenet_v2.py +++ b/mmseg/models/backbones/mobilenet_v2.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch.nn as nn diff --git a/mmseg/models/backbones/mobilenet_v3.py b/mmseg/models/backbones/mobilenet_v3.py index ae0b45d..dd3d6eb 100644 --- a/mmseg/models/backbones/mobilenet_v3.py +++ b/mmseg/models/backbones/mobilenet_v3.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import mmcv diff --git a/mmseg/models/backbones/resnest.py b/mmseg/models/backbones/resnest.py index 8931dec..f47adb5 100644 --- a/mmseg/models/backbones/resnest.py +++ b/mmseg/models/backbones/resnest.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmseg/models/backbones/resnet.py b/mmseg/models/backbones/resnet.py index f7238f0..f9a1ceb 100644 --- a/mmseg/models/backbones/resnet.py +++ b/mmseg/models/backbones/resnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch.nn as nn diff --git a/mmseg/models/backbones/resnext.py b/mmseg/models/backbones/resnext.py index fa8149c..450b77b 100644 --- a/mmseg/models/backbones/resnext.py +++ b/mmseg/models/backbones/resnext.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math from mmcv.cnn import build_conv_layer, build_norm_layer diff --git a/mmseg/models/backbones/swin.py b/mmseg/models/backbones/swin.py index 68a989b..c75bf5f 100644 --- a/mmseg/models/backbones/swin.py +++ b/mmseg/models/backbones/swin.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings from copy import deepcopy diff --git a/mmseg/models/backbones/unet.py b/mmseg/models/backbones/unet.py index 705dd2b..680c79e 100644 --- a/mmseg/models/backbones/unet.py +++ b/mmseg/models/backbones/unet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch.nn as nn diff --git a/mmseg/models/backbones/vit.py b/mmseg/models/backbones/vit.py index e4f1839..5bee596 100644 --- a/mmseg/models/backbones/vit.py +++ b/mmseg/models/backbones/vit.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import warnings diff --git a/mmseg/models/builder.py b/mmseg/models/builder.py index 05d0606..5e18e4e 100644 --- a/mmseg/models/builder.py +++ b/mmseg/models/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings from mmcv.cnn import MODELS as MMCV_MODELS diff --git a/mmseg/models/decode_heads/__init__.py b/mmseg/models/decode_heads/__init__.py index 5b64125..b0daf0e 100644 --- a/mmseg/models/decode_heads/__init__.py +++ b/mmseg/models/decode_heads/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .ann_head import ANNHead from .apc_head import APCHead from .aspp_head import ASPPHead diff --git a/mmseg/models/decode_heads/ann_head.py b/mmseg/models/decode_heads/ann_head.py index 396c54e..c8d882e 100644 --- a/mmseg/models/decode_heads/ann_head.py +++ b/mmseg/models/decode_heads/ann_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/apc_head.py b/mmseg/models/decode_heads/apc_head.py index 2118232..3198fd1 100644 --- a/mmseg/models/decode_heads/apc_head.py +++ b/mmseg/models/decode_heads/apc_head.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/mmseg/models/decode_heads/aspp_head.py b/mmseg/models/decode_heads/aspp_head.py index 6332ab1..1fbd1bc 100644 --- a/mmseg/models/decode_heads/aspp_head.py +++ b/mmseg/models/decode_heads/aspp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/cascade_decode_head.py b/mmseg/models/decode_heads/cascade_decode_head.py index d02122c..f7c3da0 100644 --- a/mmseg/models/decode_heads/cascade_decode_head.py +++ b/mmseg/models/decode_heads/cascade_decode_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from abc import ABCMeta, abstractmethod from .decode_head import BaseDecodeHead diff --git a/mmseg/models/decode_heads/cc_head.py b/mmseg/models/decode_heads/cc_head.py index 95c2706..ed19eb4 100644 --- a/mmseg/models/decode_heads/cc_head.py +++ b/mmseg/models/decode_heads/cc_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from ..builder import HEADS diff --git a/mmseg/models/decode_heads/da_head.py b/mmseg/models/decode_heads/da_head.py index 8ee0e08..77fd663 100644 --- a/mmseg/models/decode_heads/da_head.py +++ b/mmseg/models/decode_heads/da_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F from mmcv.cnn import ConvModule, Scale diff --git a/mmseg/models/decode_heads/decode_head.py b/mmseg/models/decode_heads/decode_head.py index 54d517f..b38701a 100644 --- a/mmseg/models/decode_heads/decode_head.py +++ b/mmseg/models/decode_heads/decode_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from abc import ABCMeta, abstractmethod import torch diff --git a/mmseg/models/decode_heads/dm_head.py b/mmseg/models/decode_heads/dm_head.py index 3161b06..ffaa870 100644 --- a/mmseg/models/decode_heads/dm_head.py +++ b/mmseg/models/decode_heads/dm_head.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/mmseg/models/decode_heads/dnl_head.py b/mmseg/models/decode_heads/dnl_head.py index 52a662c..ab53d9a 100644 --- a/mmseg/models/decode_heads/dnl_head.py +++ b/mmseg/models/decode_heads/dnl_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.cnn import NonLocal2d from torch import nn diff --git a/mmseg/models/decode_heads/ema_head.py b/mmseg/models/decode_heads/ema_head.py index 619d757..f6de167 100644 --- a/mmseg/models/decode_heads/ema_head.py +++ b/mmseg/models/decode_heads/ema_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmseg/models/decode_heads/enc_head.py b/mmseg/models/decode_heads/enc_head.py index 0c11994..648c890 100644 --- a/mmseg/models/decode_heads/enc_head.py +++ b/mmseg/models/decode_heads/enc_head.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/mmseg/models/decode_heads/fcn_head.py b/mmseg/models/decode_heads/fcn_head.py index 4ea3742..3c8de51 100644 --- a/mmseg/models/decode_heads/fcn_head.py +++ b/mmseg/models/decode_heads/fcn_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/fpn_head.py b/mmseg/models/decode_heads/fpn_head.py index 1e5bfd6..e41f324 100644 --- a/mmseg/models/decode_heads/fpn_head.py +++ b/mmseg/models/decode_heads/fpn_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/gc_head.py b/mmseg/models/decode_heads/gc_head.py index 93f60ad..eed5074 100644 --- a/mmseg/models/decode_heads/gc_head.py +++ b/mmseg/models/decode_heads/gc_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.cnn import ContextBlock diff --git a/mmseg/models/decode_heads/lraspp_head.py b/mmseg/models/decode_heads/lraspp_head.py index 32a093c..c10ff0d 100644 --- a/mmseg/models/decode_heads/lraspp_head.py +++ b/mmseg/models/decode_heads/lraspp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv import is_tuple_of diff --git a/mmseg/models/decode_heads/nl_head.py b/mmseg/models/decode_heads/nl_head.py index 3165875..637517e 100644 --- a/mmseg/models/decode_heads/nl_head.py +++ b/mmseg/models/decode_heads/nl_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.cnn import NonLocal2d diff --git a/mmseg/models/decode_heads/ocr_head.py b/mmseg/models/decode_heads/ocr_head.py index e180e10..09eadfb 100644 --- a/mmseg/models/decode_heads/ocr_head.py +++ b/mmseg/models/decode_heads/ocr_head.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/mmseg/models/decode_heads/point_head.py b/mmseg/models/decode_heads/point_head.py index f2d9fcc..4bc388c 100644 --- a/mmseg/models/decode_heads/point_head.py +++ b/mmseg/models/decode_heads/point_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. # Modified from https://github.com/facebookresearch/detectron2/tree/master/projects/PointRend/point_head/point_head.py # noqa import torch diff --git a/mmseg/models/decode_heads/psa_head.py b/mmseg/models/decode_heads/psa_head.py index 8d915e5..df7593c 100644 --- a/mmseg/models/decode_heads/psa_head.py +++ b/mmseg/models/decode_heads/psa_head.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/mmseg/models/decode_heads/psp_head.py b/mmseg/models/decode_heads/psp_head.py index 4416199..a27ae4b 100644 --- a/mmseg/models/decode_heads/psp_head.py +++ b/mmseg/models/decode_heads/psp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/segformer_head.py b/mmseg/models/decode_heads/segformer_head.py index 9ae1ff6..2e75d50 100644 --- a/mmseg/models/decode_heads/segformer_head.py +++ b/mmseg/models/decode_heads/segformer_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/sep_aspp_head.py b/mmseg/models/decode_heads/sep_aspp_head.py index 50bd52b..4e894e2 100644 --- a/mmseg/models/decode_heads/sep_aspp_head.py +++ b/mmseg/models/decode_heads/sep_aspp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, DepthwiseSeparableConvModule diff --git a/mmseg/models/decode_heads/sep_fcn_head.py b/mmseg/models/decode_heads/sep_fcn_head.py index 39844c9..5e22a66 100644 --- a/mmseg/models/decode_heads/sep_fcn_head.py +++ b/mmseg/models/decode_heads/sep_fcn_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.cnn import DepthwiseSeparableConvModule from ..builder import HEADS diff --git a/mmseg/models/decode_heads/setr_mla_head.py b/mmseg/models/decode_heads/setr_mla_head.py index c4e22bf..6bb94ae 100644 --- a/mmseg/models/decode_heads/setr_mla_head.py +++ b/mmseg/models/decode_heads/setr_mla_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/decode_heads/setr_up_head.py b/mmseg/models/decode_heads/setr_up_head.py index a2595ad..87e7ea7 100644 --- a/mmseg/models/decode_heads/setr_up_head.py +++ b/mmseg/models/decode_heads/setr_up_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule, build_norm_layer diff --git a/mmseg/models/decode_heads/uper_head.py b/mmseg/models/decode_heads/uper_head.py index bb617f6..4a50316 100644 --- a/mmseg/models/decode_heads/uper_head.py +++ b/mmseg/models/decode_heads/uper_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/losses/__init__.py b/mmseg/models/losses/__init__.py index beca720..e85d8e0 100644 --- a/mmseg/models/losses/__init__.py +++ b/mmseg/models/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .accuracy import Accuracy, accuracy from .cross_entropy_loss import (CrossEntropyLoss, binary_cross_entropy, cross_entropy, mask_cross_entropy) diff --git a/mmseg/models/losses/accuracy.py b/mmseg/models/losses/accuracy.py index c0fd2e7..f2cd16b 100644 --- a/mmseg/models/losses/accuracy.py +++ b/mmseg/models/losses/accuracy.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn diff --git a/mmseg/models/losses/cross_entropy_loss.py b/mmseg/models/losses/cross_entropy_loss.py index 42c0790..9a7ccea 100644 --- a/mmseg/models/losses/cross_entropy_loss.py +++ b/mmseg/models/losses/cross_entropy_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/mmseg/models/losses/dice_loss.py b/mmseg/models/losses/dice_loss.py index 27a77b9..0b07e97 100644 --- a/mmseg/models/losses/dice_loss.py +++ b/mmseg/models/losses/dice_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Modified from https://github.com/LikeLy-Journey/SegmenTron/blob/master/ segmentron/solver/loss.py (Apache-2.0 License)""" import torch diff --git a/mmseg/models/losses/lovasz_loss.py b/mmseg/models/losses/lovasz_loss.py index e8df6e8..275c4c5 100644 --- a/mmseg/models/losses/lovasz_loss.py +++ b/mmseg/models/losses/lovasz_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Modified from https://github.com/bermanmaxim/LovaszSoftmax/blob/master/pytor ch/lovasz_losses.py Lovasz-Softmax and Jaccard hinge loss in PyTorch Maxim Berman 2018 ESAT-PSI KU Leuven (MIT License)""" diff --git a/mmseg/models/losses/utils.py b/mmseg/models/losses/utils.py index ab58766..c57e4b1 100644 --- a/mmseg/models/losses/utils.py +++ b/mmseg/models/losses/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import functools import mmcv diff --git a/mmseg/models/necks/__init__.py b/mmseg/models/necks/__init__.py index 3d6a4c0..c496853 100644 --- a/mmseg/models/necks/__init__.py +++ b/mmseg/models/necks/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .fpn import FPN from .mla_neck import MLANeck from .multilevel_neck import MultiLevelNeck diff --git a/mmseg/models/necks/fpn.py b/mmseg/models/necks/fpn.py index 5e1bd21..8461a75 100644 --- a/mmseg/models/necks/fpn.py +++ b/mmseg/models/necks/fpn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn import torch.nn.functional as F from mmcv.cnn import ConvModule diff --git a/mmseg/models/necks/mla_neck.py b/mmseg/models/necks/mla_neck.py index 010c43d..5fc3b98 100644 --- a/mmseg/models/necks/mla_neck.py +++ b/mmseg/models/necks/mla_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule, build_norm_layer diff --git a/mmseg/models/necks/multilevel_neck.py b/mmseg/models/necks/multilevel_neck.py index 9f63893..cbf4b01 100644 --- a/mmseg/models/necks/multilevel_neck.py +++ b/mmseg/models/necks/multilevel_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule, xavier_init diff --git a/mmseg/models/segmentors/__init__.py b/mmseg/models/segmentors/__init__.py index dca2f09..387c858 100644 --- a/mmseg/models/segmentors/__init__.py +++ b/mmseg/models/segmentors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base import BaseSegmentor from .cascade_encoder_decoder import CascadeEncoderDecoder from .encoder_decoder import EncoderDecoder diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py index 0ace142..906c6fe 100644 --- a/mmseg/models/segmentors/base.py +++ b/mmseg/models/segmentors/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/mmseg/models/segmentors/cascade_encoder_decoder.py b/mmseg/models/segmentors/cascade_encoder_decoder.py index fb5a9ae..7f9f900 100644 --- a/mmseg/models/segmentors/cascade_encoder_decoder.py +++ b/mmseg/models/segmentors/cascade_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from torch import nn from mmseg.core import add_prefix diff --git a/mmseg/models/segmentors/encoder_decoder.py b/mmseg/models/segmentors/encoder_decoder.py index 04de3f4..72467b4 100644 --- a/mmseg/models/segmentors/encoder_decoder.py +++ b/mmseg/models/segmentors/encoder_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/mmseg/models/utils/__init__.py b/mmseg/models/utils/__init__.py index 6ef12bb..817ab9c 100644 --- a/mmseg/models/utils/__init__.py +++ b/mmseg/models/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .ckpt_convert import swin_convert, vit_convert from .embed import PatchEmbed from .inverted_residual import InvertedResidual, InvertedResidualV3 diff --git a/mmseg/models/utils/ckpt_convert.py b/mmseg/models/utils/ckpt_convert.py index 0b1b277..fd46320 100644 --- a/mmseg/models/utils/ckpt_convert.py +++ b/mmseg/models/utils/ckpt_convert.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections import OrderedDict diff --git a/mmseg/models/utils/embed.py b/mmseg/models/utils/embed.py index 73d8ed1..c0cf143 100644 --- a/mmseg/models/utils/embed.py +++ b/mmseg/models/utils/embed.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn.functional as F from mmcv.cnn import build_conv_layer, build_norm_layer from mmcv.runner.base_module import BaseModule diff --git a/mmseg/models/utils/inverted_residual.py b/mmseg/models/utils/inverted_residual.py index 5a209a5..c9cda76 100644 --- a/mmseg/models/utils/inverted_residual.py +++ b/mmseg/models/utils/inverted_residual.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.cnn import ConvModule from torch import nn from torch.utils import checkpoint as cp diff --git a/mmseg/models/utils/make_divisible.py b/mmseg/models/utils/make_divisible.py index 75ad756..ed42c2e 100644 --- a/mmseg/models/utils/make_divisible.py +++ b/mmseg/models/utils/make_divisible.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. def make_divisible(value, divisor, min_value=None, min_ratio=0.9): """Make divisible function. diff --git a/mmseg/models/utils/res_layer.py b/mmseg/models/utils/res_layer.py index 9c474ed..190a0c5 100644 --- a/mmseg/models/utils/res_layer.py +++ b/mmseg/models/utils/res_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.cnn import build_conv_layer, build_norm_layer from mmcv.runner import Sequential from torch import nn as nn diff --git a/mmseg/models/utils/se_layer.py b/mmseg/models/utils/se_layer.py index e083404..16f52aa 100644 --- a/mmseg/models/utils/se_layer.py +++ b/mmseg/models/utils/se_layer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmseg/models/utils/self_attention_block.py b/mmseg/models/utils/self_attention_block.py index 372fad2..c945fa7 100644 --- a/mmseg/models/utils/self_attention_block.py +++ b/mmseg/models/utils/self_attention_block.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.cnn import ConvModule, constant_init from torch import nn as nn diff --git a/mmseg/models/utils/shape_convert.py b/mmseg/models/utils/shape_convert.py index 7444160..34c8648 100644 --- a/mmseg/models/utils/shape_convert.py +++ b/mmseg/models/utils/shape_convert.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. def nlc_to_nchw(x, hw_shape): """Convert [N, L, C] shape tensor to [N, C, H, W] shape tensor. diff --git a/mmseg/models/utils/up_conv_block.py b/mmseg/models/utils/up_conv_block.py index 6566b74..d8396d9 100644 --- a/mmseg/models/utils/up_conv_block.py +++ b/mmseg/models/utils/up_conv_block.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, build_upsample_layer diff --git a/mmseg/ops/__init__.py b/mmseg/ops/__init__.py index bec51c7..bc075cd 100644 --- a/mmseg/ops/__init__.py +++ b/mmseg/ops/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .encoding import Encoding from .wrappers import Upsample, resize diff --git a/mmseg/ops/encoding.py b/mmseg/ops/encoding.py index 7eb3629..f397cc5 100644 --- a/mmseg/ops/encoding.py +++ b/mmseg/ops/encoding.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from torch import nn from torch.nn import functional as F diff --git a/mmseg/ops/wrappers.py b/mmseg/ops/wrappers.py index 0ed9a0c..ce67e4b 100644 --- a/mmseg/ops/wrappers.py +++ b/mmseg/ops/wrappers.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch.nn as nn diff --git a/mmseg/utils/__init__.py b/mmseg/utils/__init__.py index ac489e2..3f15580 100644 --- a/mmseg/utils/__init__.py +++ b/mmseg/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .collect_env import collect_env from .logger import get_root_logger diff --git a/mmseg/utils/collect_env.py b/mmseg/utils/collect_env.py index 8293a05..3379ecb 100644 --- a/mmseg/utils/collect_env.py +++ b/mmseg/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/mmseg/utils/logger.py b/mmseg/utils/logger.py index 05d2f13..0cb3c78 100644 --- a/mmseg/utils/logger.py +++ b/mmseg/utils/logger.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging from mmcv.utils import get_logger diff --git a/setup.py b/setup.py index 92c0950..bc85294 100755 --- a/setup.py +++ b/setup.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import os.path as osp import shutil @@ -35,9 +36,9 @@ def parse_requirements(fname='requirements.txt', with_version=True): CommandLine: python -c "import setup; print(setup.parse_requirements())" """ + import re import sys from os.path import exists - import re require_fpath = fname def parse_line(line): diff --git a/tests/__init__.py b/tests/__init__.py index e69de29..ef101fe 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_config.py b/tests/test_config.py index b991fbf..e6cec40 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import glob import os from os.path import dirname, exists, isdir, join, relpath diff --git a/tests/test_data/test_dataset.py b/tests/test_data/test_dataset.py index 57a33da..7ef59f2 100644 --- a/tests/test_data/test_dataset.py +++ b/tests/test_data/test_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from unittest.mock import MagicMock, patch diff --git a/tests/test_data/test_dataset_builder.py b/tests/test_data/test_dataset_builder.py index c6827e4..c945fe5 100644 --- a/tests/test_data/test_dataset_builder.py +++ b/tests/test_data/test_dataset_builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import os.path as osp diff --git a/tests/test_data/test_loading.py b/tests/test_data/test_loading.py index e8aa5d3..fdda93e 100644 --- a/tests/test_data/test_loading.py +++ b/tests/test_data/test_loading.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_data/test_transform.py b/tests/test_data/test_transform.py index 33ed4ec..3862e75 100644 --- a/tests/test_data/test_transform.py +++ b/tests/test_data/test_transform.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os.path as osp diff --git a/tests/test_data/test_tta.py b/tests/test_data/test_tta.py index cc8c71e..d61af27 100644 --- a/tests/test_data/test_tta.py +++ b/tests/test_data/test_tta.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/tests/test_digit_version.py b/tests/test_digit_version.py index 4d66490..45daf09 100644 --- a/tests/test_digit_version.py +++ b/tests/test_digit_version.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmseg import digit_version diff --git a/tests/test_eval_hook.py b/tests/test_eval_hook.py index 394051b..54d2a43 100644 --- a/tests/test_eval_hook.py +++ b/tests/test_eval_hook.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging import tempfile from unittest.mock import MagicMock, patch diff --git a/tests/test_inference.py b/tests/test_inference.py index 046e036..f71a7ea 100644 --- a/tests/test_inference.py +++ b/tests/test_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/tests/test_metrics.py b/tests/test_metrics.py index 4030505..a8db802 100644 --- a/tests/test_metrics.py +++ b/tests/test_metrics.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmseg.core.evaluation import (eval_metrics, mean_dice, mean_fscore, diff --git a/tests/test_models/__init__.py b/tests/test_models/__init__.py index e69de29..ef101fe 100644 --- a/tests/test_models/__init__.py +++ b/tests/test_models/__init__.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_models/test_backbones/__init__.py b/tests/test_models/test_backbones/__init__.py index 78a93a5..8b673fa 100644 --- a/tests/test_models/test_backbones/__init__.py +++ b/tests/test_models/test_backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .utils import all_zeros, check_norm_state, is_block, is_norm __all__ = ['is_norm', 'is_block', 'all_zeros', 'check_norm_state'] diff --git a/tests/test_models/test_backbones/test_blocks.py b/tests/test_models/test_backbones/test_blocks.py index f459fbb..ad3ad2d 100644 --- a/tests/test_models/test_backbones/test_blocks.py +++ b/tests/test_models/test_backbones/test_blocks.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import pytest import torch diff --git a/tests/test_models/test_backbones/test_cgnet.py b/tests/test_models/test_backbones/test_cgnet.py index dfc4e9a..f938525 100644 --- a/tests/test_models/test_backbones/test_cgnet.py +++ b/tests/test_models/test_backbones/test_cgnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_fast_scnn.py b/tests/test_models/test_backbones/test_fast_scnn.py index f4a5809..e639046 100644 --- a/tests/test_models/test_backbones/test_fast_scnn.py +++ b/tests/test_models/test_backbones/test_fast_scnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_hrnet.py b/tests/test_models/test_backbones/test_hrnet.py index 81611a0..aa42c85 100644 --- a/tests/test_models/test_backbones/test_hrnet.py +++ b/tests/test_models/test_backbones/test_hrnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils.parrots_wrapper import _BatchNorm from mmseg.models.backbones import HRNet diff --git a/tests/test_models/test_backbones/test_mit.py b/tests/test_models/test_backbones/test_mit.py index bf6cca1..86d98bf 100644 --- a/tests/test_models/test_backbones/test_mit.py +++ b/tests/test_models/test_backbones/test_mit.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_mobilenet_v3.py b/tests/test_models/test_backbones/test_mobilenet_v3.py index 1ebeac4..a238035 100644 --- a/tests/test_models/test_backbones/test_mobilenet_v3.py +++ b/tests/test_models/test_backbones/test_mobilenet_v3.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_resnest.py b/tests/test_models/test_backbones/test_resnest.py index 78d97de..3013f34 100644 --- a/tests/test_models/test_backbones/test_resnest.py +++ b/tests/test_models/test_backbones/test_resnest.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_resnet.py b/tests/test_models/test_backbones/test_resnet.py index e0947db..2efb498 100644 --- a/tests/test_models/test_backbones/test_resnet.py +++ b/tests/test_models/test_backbones/test_resnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch from mmcv.ops import DeformConv2dPack diff --git a/tests/test_models/test_backbones/test_resnext.py b/tests/test_models/test_backbones/test_resnext.py index 2ba5f8e..2aecaf0 100644 --- a/tests/test_models/test_backbones/test_resnext.py +++ b/tests/test_models/test_backbones/test_resnext.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_swin.py b/tests/test_models/test_backbones/test_swin.py index 42e3086..d82a4ba 100644 --- a/tests/test_models/test_backbones/test_swin.py +++ b/tests/test_models/test_backbones/test_swin.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_unet.py b/tests/test_models/test_backbones/test_unet.py index 52f2123..3a035c8 100644 --- a/tests/test_models/test_backbones/test_unet.py +++ b/tests/test_models/test_backbones/test_unet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch from mmcv.cnn import ConvModule diff --git a/tests/test_models/test_backbones/test_vit.py b/tests/test_models/test_backbones/test_vit.py index 16d6aba..c9afe07 100644 --- a/tests/test_models/test_backbones/test_vit.py +++ b/tests/test_models/test_backbones/test_vit.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/utils.py b/tests/test_models/test_backbones/utils.py index d50b772..54b6404 100644 --- a/tests/test_models/test_backbones/utils.py +++ b/tests/test_models/test_backbones/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from torch.nn.modules import GroupNorm from torch.nn.modules.batchnorm import _BatchNorm diff --git a/tests/test_models/test_forward.py b/tests/test_models/test_forward.py index ea9d70b..5aa3a2f 100644 --- a/tests/test_models/test_forward.py +++ b/tests/test_models/test_forward.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """pytest tests/test_forward.py.""" import copy from os.path import dirname, exists, join diff --git a/tests/test_models/test_heads/__init__.py b/tests/test_models/test_heads/__init__.py index e69de29..ef101fe 100644 --- a/tests/test_models/test_heads/__init__.py +++ b/tests/test_models/test_heads/__init__.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_models/test_heads/test_ann_head.py b/tests/test_models/test_heads/test_ann_head.py index 61556c0..22caf03 100644 --- a/tests/test_models/test_heads/test_ann_head.py +++ b/tests/test_models/test_heads/test_ann_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import ANNHead diff --git a/tests/test_models/test_heads/test_apc_head.py b/tests/test_models/test_heads/test_apc_head.py index 37f1a55..a79d66f 100644 --- a/tests/test_models/test_heads/test_apc_head.py +++ b/tests/test_models/test_heads/test_apc_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_aspp_head.py b/tests/test_models/test_heads/test_aspp_head.py index bd4ce56..203fef0 100644 --- a/tests/test_models/test_heads/test_aspp_head.py +++ b/tests/test_models/test_heads/test_aspp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_cc_head.py b/tests/test_models/test_heads/test_cc_head.py index 12a19bf..ff284ef 100644 --- a/tests/test_models/test_heads/test_cc_head.py +++ b/tests/test_models/test_heads/test_cc_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_da_head.py b/tests/test_models/test_heads/test_da_head.py index 20f3a21..7bc46aa 100644 --- a/tests/test_models/test_heads/test_da_head.py +++ b/tests/test_models/test_heads/test_da_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import DAHead diff --git a/tests/test_models/test_heads/test_decode_head.py b/tests/test_models/test_heads/test_decode_head.py index 97262b9..421043d 100644 --- a/tests/test_models/test_heads/test_decode_head.py +++ b/tests/test_models/test_heads/test_decode_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from unittest.mock import patch import pytest diff --git a/tests/test_models/test_heads/test_dm_head.py b/tests/test_models/test_heads/test_dm_head.py index e85127b..f85d547 100644 --- a/tests/test_models/test_heads/test_dm_head.py +++ b/tests/test_models/test_heads/test_dm_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_dnl_head.py b/tests/test_models/test_heads/test_dnl_head.py index b3e98aa..1724201 100644 --- a/tests/test_models/test_heads/test_dnl_head.py +++ b/tests/test_models/test_heads/test_dnl_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import DNLHead diff --git a/tests/test_models/test_heads/test_ema_head.py b/tests/test_models/test_heads/test_ema_head.py index 4214b0c..8947e0d 100644 --- a/tests/test_models/test_heads/test_ema_head.py +++ b/tests/test_models/test_heads/test_ema_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import EMAHead diff --git a/tests/test_models/test_heads/test_enc_head.py b/tests/test_models/test_heads/test_enc_head.py index 3a29330..db5383d 100644 --- a/tests/test_models/test_heads/test_enc_head.py +++ b/tests/test_models/test_heads/test_enc_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import EncHead diff --git a/tests/test_models/test_heads/test_fcn_head.py b/tests/test_models/test_heads/test_fcn_head.py index 24ae086..3783fe3 100644 --- a/tests/test_models/test_heads/test_fcn_head.py +++ b/tests/test_models/test_heads/test_fcn_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch from mmcv.cnn import ConvModule, DepthwiseSeparableConvModule diff --git a/tests/test_models/test_heads/test_gc_head.py b/tests/test_models/test_heads/test_gc_head.py index 5201730..4540222 100644 --- a/tests/test_models/test_heads/test_gc_head.py +++ b/tests/test_models/test_heads/test_gc_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import GCHead diff --git a/tests/test_models/test_heads/test_lraspp_head.py b/tests/test_models/test_heads/test_lraspp_head.py index 5031936..c83377f 100644 --- a/tests/test_models/test_heads/test_lraspp_head.py +++ b/tests/test_models/test_heads/test_lraspp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_nl_head.py b/tests/test_models/test_heads/test_nl_head.py index 6f4bede..04b173f 100644 --- a/tests/test_models/test_heads/test_nl_head.py +++ b/tests/test_models/test_heads/test_nl_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import NLHead diff --git a/tests/test_models/test_heads/test_ocr_head.py b/tests/test_models/test_heads/test_ocr_head.py index bc2af75..c6551f8 100644 --- a/tests/test_models/test_heads/test_ocr_head.py +++ b/tests/test_models/test_heads/test_ocr_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models.decode_heads import FCNHead, OCRHead diff --git a/tests/test_models/test_heads/test_point_head.py b/tests/test_models/test_heads/test_point_head.py index b54b979..6c5ea65 100644 --- a/tests/test_models/test_heads/test_point_head.py +++ b/tests/test_models/test_heads/test_point_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.utils import ConfigDict diff --git a/tests/test_models/test_heads/test_psa_head.py b/tests/test_models/test_heads/test_psa_head.py index d8f38b6..21450b5 100644 --- a/tests/test_models/test_heads/test_psa_head.py +++ b/tests/test_models/test_heads/test_psa_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_psp_head.py b/tests/test_models/test_heads/test_psp_head.py index 38b39d7..f4a8781 100644 --- a/tests/test_models/test_heads/test_psp_head.py +++ b/tests/test_models/test_heads/test_psp_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_segformer_head.py b/tests/test_models/test_heads/test_segformer_head.py index aa8dedb..73afaba 100644 --- a/tests/test_models/test_heads/test_segformer_head.py +++ b/tests/test_models/test_heads/test_segformer_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_setr_mla_head.py b/tests/test_models/test_heads/test_setr_mla_head.py index d43aab0..07992d0 100644 --- a/tests/test_models/test_heads/test_setr_mla_head.py +++ b/tests/test_models/test_heads/test_setr_mla_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_setr_up_head.py b/tests/test_models/test_heads/test_setr_up_head.py index ad6ca56..d552e17 100644 --- a/tests/test_models/test_heads/test_setr_up_head.py +++ b/tests/test_models/test_heads/test_setr_up_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/test_uper_head.py b/tests/test_models/test_heads/test_uper_head.py index 2c66db8..961b01b 100644 --- a/tests/test_models/test_heads/test_uper_head.py +++ b/tests/test_models/test_heads/test_uper_head.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_heads/utils.py b/tests/test_models/test_heads/utils.py index 1407f0a..675241c 100644 --- a/tests/test_models/test_heads/utils.py +++ b/tests/test_models/test_heads/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.cnn import ConvModule from mmcv.utils.parrots_wrapper import SyncBatchNorm diff --git a/tests/test_models/test_losses/__init__.py b/tests/test_models/test_losses/__init__.py index e69de29..ef101fe 100644 --- a/tests/test_models/test_losses/__init__.py +++ b/tests/test_models/test_losses/__init__.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_models/test_losses/test_ce_loss.py b/tests/test_models/test_losses/test_ce_loss.py index 9619b60..73217ec 100644 --- a/tests/test_models/test_losses/test_ce_loss.py +++ b/tests/test_models/test_losses/test_ce_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_losses/test_dice_loss.py b/tests/test_models/test_losses/test_dice_loss.py index 01ded6f..05d1b1e 100644 --- a/tests/test_models/test_losses/test_dice_loss.py +++ b/tests/test_models/test_losses/test_dice_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch diff --git a/tests/test_models/test_losses/test_lovasz_loss.py b/tests/test_models/test_losses/test_lovasz_loss.py index 6fac430..e2dee81 100644 --- a/tests/test_models/test_losses/test_lovasz_loss.py +++ b/tests/test_models/test_losses/test_lovasz_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_losses/test_utils.py b/tests/test_models/test_losses/test_utils.py index a5251e4..1d94387 100644 --- a/tests/test_models/test_losses/test_utils.py +++ b/tests/test_models/test_losses/test_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest import torch diff --git a/tests/test_models/test_necks/__init__.py b/tests/test_models/test_necks/__init__.py index e69de29..ef101fe 100644 --- a/tests/test_models/test_necks/__init__.py +++ b/tests/test_models/test_necks/__init__.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_models/test_necks/test_fpn.py b/tests/test_models/test_necks/test_fpn.py index 8fc9684..f7b8e41 100644 --- a/tests/test_models/test_necks/test_fpn.py +++ b/tests/test_models/test_necks/test_fpn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models import FPN diff --git a/tests/test_models/test_necks/test_mla_neck.py b/tests/test_models/test_necks/test_mla_neck.py index 75f0401..a20c132 100644 --- a/tests/test_models/test_necks/test_mla_neck.py +++ b/tests/test_models/test_necks/test_mla_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models import MLANeck diff --git a/tests/test_models/test_necks/test_multilevel_neck.py b/tests/test_models/test_necks/test_multilevel_neck.py index c5a567d..641a212 100644 --- a/tests/test_models/test_necks/test_multilevel_neck.py +++ b/tests/test_models/test_necks/test_multilevel_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmseg.models import MultiLevelNeck diff --git a/tests/test_models/test_segmentors/__init__.py b/tests/test_models/test_segmentors/__init__.py index e69de29..ef101fe 100644 --- a/tests/test_models/test_segmentors/__init__.py +++ b/tests/test_models/test_segmentors/__init__.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_models/test_segmentors/test_cascade_encoder_decoder.py b/tests/test_models/test_segmentors/test_cascade_encoder_decoder.py index 142e81f..07ad5c3 100644 --- a/tests/test_models/test_segmentors/test_cascade_encoder_decoder.py +++ b/tests/test_models/test_segmentors/test_cascade_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv import ConfigDict from mmseg.models import build_segmentor diff --git a/tests/test_models/test_segmentors/test_encoder_decoder.py b/tests/test_models/test_segmentors/test_encoder_decoder.py index f40c4ea..4ed1437 100644 --- a/tests/test_models/test_segmentors/test_encoder_decoder.py +++ b/tests/test_models/test_segmentors/test_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv import ConfigDict from mmseg.models import build_segmentor diff --git a/tests/test_models/test_segmentors/utils.py b/tests/test_models/test_segmentors/utils.py index cfe9a17..0f51a4b 100644 --- a/tests/test_models/test_segmentors/utils.py +++ b/tests/test_models/test_segmentors/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch from torch import nn diff --git a/tests/test_sampler.py b/tests/test_sampler.py index 3c79c16..8e613a5 100644 --- a/tests/test_sampler.py +++ b/tests/test_sampler.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tools/analyze_logs.py b/tools/analyze_logs.py index fb017ef..8c62a34 100644 --- a/tools/analyze_logs.py +++ b/tools/analyze_logs.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. """Modified from https://github.com/open- mmlab/mmdetection/blob/master/tools/analysis_tools/analyze_logs.py.""" import argparse diff --git a/tools/benchmark.py b/tools/benchmark.py index 0a61793..d72980e 100644 --- a/tools/benchmark.py +++ b/tools/benchmark.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import time diff --git a/tools/convert_datasets/chase_db1.py b/tools/convert_datasets/chase_db1.py index 56bb210..580e6e7 100644 --- a/tools/convert_datasets/chase_db1.py +++ b/tools/convert_datasets/chase_db1.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/convert_datasets/cityscapes.py b/tools/convert_datasets/cityscapes.py index 99d05b4..17b6168 100644 --- a/tools/convert_datasets/cityscapes.py +++ b/tools/convert_datasets/cityscapes.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os.path as osp diff --git a/tools/convert_datasets/drive.py b/tools/convert_datasets/drive.py index 891f06f..f547579 100644 --- a/tools/convert_datasets/drive.py +++ b/tools/convert_datasets/drive.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/convert_datasets/hrf.py b/tools/convert_datasets/hrf.py index bdeb6e7..5e016e3 100644 --- a/tools/convert_datasets/hrf.py +++ b/tools/convert_datasets/hrf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/convert_datasets/pascal_context.py b/tools/convert_datasets/pascal_context.py index dc49ab7..03b79d5 100644 --- a/tools/convert_datasets/pascal_context.py +++ b/tools/convert_datasets/pascal_context.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os.path as osp from functools import partial diff --git a/tools/convert_datasets/stare.py b/tools/convert_datasets/stare.py index 6238d62..29b78c0 100644 --- a/tools/convert_datasets/stare.py +++ b/tools/convert_datasets/stare.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import gzip import os diff --git a/tools/convert_datasets/voc_aug.py b/tools/convert_datasets/voc_aug.py index 9427463..1d42c27 100644 --- a/tools/convert_datasets/voc_aug.py +++ b/tools/convert_datasets/voc_aug.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os.path as osp from functools import partial diff --git a/tools/deploy_test.py b/tools/deploy_test.py index 56fd61c..6e709b8 100644 --- a/tools/deploy_test.py +++ b/tools/deploy_test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/get_flops.py b/tools/get_flops.py index bc98c52..83dea0a 100644 --- a/tools/get_flops.py +++ b/tools/get_flops.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse from mmcv import Config diff --git a/tools/mmseg2torchserve.py b/tools/mmseg2torchserve.py index 373f5ca..9063634 100644 --- a/tools/mmseg2torchserve.py +++ b/tools/mmseg2torchserve.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from argparse import ArgumentParser, Namespace from pathlib import Path from tempfile import TemporaryDirectory diff --git a/tools/mmseg_handler.py b/tools/mmseg_handler.py index b0cb248..7fabd46 100644 --- a/tools/mmseg_handler.py +++ b/tools/mmseg_handler.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import base64 import io import os diff --git a/tools/model_converters/mit_convert.py b/tools/model_converters/mit_convert.py index c914c4e..125345c 100644 --- a/tools/model_converters/mit_convert.py +++ b/tools/model_converters/mit_convert.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse from collections import OrderedDict diff --git a/tools/onnx2tensorrt.py b/tools/onnx2tensorrt.py index 203ae82..1cda222 100644 --- a/tools/onnx2tensorrt.py +++ b/tools/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/print_config.py b/tools/print_config.py index 88984e4..fb978c9 100644 --- a/tools/print_config.py +++ b/tools/print_config.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse from mmcv import Config, DictAction diff --git a/tools/publish_model.py b/tools/publish_model.py index a049f17..e266057 100644 --- a/tools/publish_model.py +++ b/tools/publish_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import subprocess diff --git a/tools/pytorch2onnx.py b/tools/pytorch2onnx.py index 17f1093..1751a7b 100644 --- a/tools/pytorch2onnx.py +++ b/tools/pytorch2onnx.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse from functools import partial diff --git a/tools/pytorch2torchscript.py b/tools/pytorch2torchscript.py index 206c4bb..fad6fd1 100644 --- a/tools/pytorch2torchscript.py +++ b/tools/pytorch2torchscript.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import mmcv diff --git a/tools/test.py b/tools/test.py index ab2bd60..87bd365 100644 --- a/tools/test.py +++ b/tools/test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os diff --git a/tools/train.py b/tools/train.py index 2d11df3..490b3ff 100644 --- a/tools/train.py +++ b/tools/train.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import copy import os