From 23213c839ff2d1907a80d6ea29f13c32a24bb8ef Mon Sep 17 00:00:00 2001 From: "Kelvin C.K. Chan" Date: Thu, 19 Aug 2021 14:01:56 +0800 Subject: [PATCH] [Docs] Add license header (#496) * Add license header * Add license header * tiny Signed-off-by: lizz Co-authored-by: lizz --- demo/generation_demo.py | 1 + demo/inpainting_demo.py | 1 + demo/matting_demo.py | 1 + demo/restoration_demo.py | 1 + demo/restoration_video_demo.py | 1 + docs/conf.py | 1 + docs/stat.py | 1 + docs_zh-CN/conf.py | 1 + docs_zh-CN/stat.py | 1 + mmedit/__init__.py | 1 + mmedit/apis/__init__.py | 1 + mmedit/apis/generation_inference.py | 1 + mmedit/apis/inpainting_inference.py | 1 + mmedit/apis/matting_inference.py | 1 + mmedit/apis/restoration_inference.py | 1 + mmedit/apis/restoration_video_inference.py | 1 + mmedit/apis/test.py | 1 + mmedit/apis/train.py | 1 + mmedit/core/__init__.py | 1 + mmedit/core/distributed_wrapper.py | 1 + mmedit/core/evaluation/__init__.py | 1 + mmedit/core/evaluation/eval_hooks.py | 1 + mmedit/core/evaluation/metric_utils.py | 1 + mmedit/core/evaluation/metrics.py | 1 + mmedit/core/export/__init__.py | 1 + mmedit/core/export/wrappers.py | 1 + mmedit/core/hooks/__init__.py | 1 + mmedit/core/hooks/visualization.py | 1 + mmedit/core/mask.py | 1 + mmedit/core/misc.py | 1 + mmedit/core/optimizer/__init__.py | 1 + mmedit/core/optimizer/builder.py | 1 + mmedit/core/scheduler/__init__.py | 1 + mmedit/core/scheduler/lr_updater.py | 1 + mmedit/datasets/__init__.py | 1 + mmedit/datasets/base_dataset.py | 1 + mmedit/datasets/base_generation_dataset.py | 1 + mmedit/datasets/base_matting_dataset.py | 1 + mmedit/datasets/base_sr_dataset.py | 1 + mmedit/datasets/builder.py | 1 + mmedit/datasets/comp1k_dataset.py | 1 + mmedit/datasets/dataset_wrappers.py | 1 + mmedit/datasets/generation_paired_dataset.py | 1 + mmedit/datasets/generation_unpaired_dataset.py | 1 + mmedit/datasets/img_inpainting_dataset.py | 1 + mmedit/datasets/pipelines/__init__.py | 1 + mmedit/datasets/pipelines/augmentation.py | 1 + mmedit/datasets/pipelines/compose.py | 1 + mmedit/datasets/pipelines/crop.py | 1 + mmedit/datasets/pipelines/formating.py | 1 + mmedit/datasets/pipelines/generate_assistant.py | 1 + mmedit/datasets/pipelines/loading.py | 1 + mmedit/datasets/pipelines/matting_aug.py | 1 + mmedit/datasets/pipelines/normalization.py | 1 + mmedit/datasets/pipelines/random_down_sampling.py | 1 + mmedit/datasets/pipelines/utils.py | 1 + mmedit/datasets/registry.py | 1 + mmedit/datasets/samplers/__init__.py | 1 + mmedit/datasets/samplers/distributed_sampler.py | 1 + mmedit/datasets/sr_annotation_dataset.py | 1 + mmedit/datasets/sr_facial_landmark_dataset.py | 1 + mmedit/datasets/sr_folder_dataset.py | 1 + mmedit/datasets/sr_folder_gt_dataset.py | 1 + mmedit/datasets/sr_folder_multiple_gt_dataset.py | 1 + mmedit/datasets/sr_folder_ref_dataset.py | 1 + mmedit/datasets/sr_folder_video_dataset.py | 1 + mmedit/datasets/sr_lmdb_dataset.py | 1 + mmedit/datasets/sr_reds_dataset.py | 1 + mmedit/datasets/sr_reds_multiple_gt_dataset.py | 1 + mmedit/datasets/sr_test_multiple_gt_dataset.py | 1 + mmedit/datasets/sr_vid4_dataset.py | 1 + mmedit/datasets/sr_vimeo90k_dataset.py | 1 + mmedit/datasets/sr_vimeo90k_multiple_gt_dataset.py | 1 + mmedit/models/__init__.py | 1 + mmedit/models/backbones/__init__.py | 1 + mmedit/models/backbones/encoder_decoders/__init__.py | 1 + mmedit/models/backbones/encoder_decoders/decoders/__init__.py | 1 + .../backbones/encoder_decoders/decoders/deepfill_decoder.py | 1 + mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py | 1 + mmedit/models/backbones/encoder_decoders/decoders/gl_decoder.py | 1 + .../backbones/encoder_decoders/decoders/indexnet_decoder.py | 1 + .../models/backbones/encoder_decoders/decoders/pconv_decoder.py | 1 + .../models/backbones/encoder_decoders/decoders/plain_decoder.py | 1 + mmedit/models/backbones/encoder_decoders/decoders/resnet_dec.py | 1 + mmedit/models/backbones/encoder_decoders/encoders/__init__.py | 1 + .../backbones/encoder_decoders/encoders/deepfill_encoder.py | 1 + mmedit/models/backbones/encoder_decoders/encoders/fba_encoder.py | 1 + mmedit/models/backbones/encoder_decoders/encoders/gl_encoder.py | 1 + .../backbones/encoder_decoders/encoders/indexnet_encoder.py | 1 + .../models/backbones/encoder_decoders/encoders/pconv_encoder.py | 1 + mmedit/models/backbones/encoder_decoders/encoders/resnet.py | 1 + mmedit/models/backbones/encoder_decoders/encoders/resnet_enc.py | 1 + mmedit/models/backbones/encoder_decoders/encoders/vgg.py | 1 + mmedit/models/backbones/encoder_decoders/gl_encoder_decoder.py | 1 + mmedit/models/backbones/encoder_decoders/necks/__init__.py | 1 + .../encoder_decoders/necks/contextual_attention_neck.py | 1 + mmedit/models/backbones/encoder_decoders/necks/gl_dilation.py | 1 + .../models/backbones/encoder_decoders/pconv_encoder_decoder.py | 1 + .../models/backbones/encoder_decoders/simple_encoder_decoder.py | 1 + .../backbones/encoder_decoders/two_stage_encoder_decoder.py | 1 + mmedit/models/backbones/generation_backbones/__init__.py | 1 + mmedit/models/backbones/generation_backbones/resnet_generator.py | 1 + mmedit/models/backbones/generation_backbones/unet_generator.py | 1 + mmedit/models/backbones/sr_backbones/__init__.py | 1 + mmedit/models/backbones/sr_backbones/basicvsr_net.py | 1 + mmedit/models/backbones/sr_backbones/basicvsr_pp.py | 1 + mmedit/models/backbones/sr_backbones/dic_net.py | 1 + mmedit/models/backbones/sr_backbones/duf.py | 1 + mmedit/models/backbones/sr_backbones/edsr.py | 1 + mmedit/models/backbones/sr_backbones/edvr_net.py | 1 + mmedit/models/backbones/sr_backbones/glean_styleganv2.py | 1 + mmedit/models/backbones/sr_backbones/iconvsr.py | 1 + mmedit/models/backbones/sr_backbones/rrdb_net.py | 1 + mmedit/models/backbones/sr_backbones/sr_resnet.py | 1 + mmedit/models/backbones/sr_backbones/srcnn.py | 1 + mmedit/models/backbones/sr_backbones/tdan_net.py | 1 + mmedit/models/backbones/sr_backbones/tof.py | 1 + mmedit/models/base.py | 1 + mmedit/models/builder.py | 1 + mmedit/models/common/__init__.py | 1 + mmedit/models/common/aspp.py | 1 + mmedit/models/common/contextual_attention.py | 1 + mmedit/models/common/conv.py | 1 + mmedit/models/common/flow_warp.py | 1 + mmedit/models/common/gated_conv_module.py | 1 + mmedit/models/common/gca_module.py | 1 + mmedit/models/common/generation_model_utils.py | 1 + mmedit/models/common/img_normalize.py | 1 + mmedit/models/common/linear_module.py | 1 + mmedit/models/common/mask_conv_module.py | 1 + mmedit/models/common/model_utils.py | 1 + mmedit/models/common/partial_conv.py | 1 + mmedit/models/common/separable_conv_module.py | 1 + mmedit/models/common/sr_backbone_utils.py | 1 + mmedit/models/common/upsample.py | 1 + mmedit/models/components/__init__.py | 1 + mmedit/models/components/discriminators/__init__.py | 1 + mmedit/models/components/discriminators/deepfill_disc.py | 1 + mmedit/models/components/discriminators/gl_disc.py | 1 + mmedit/models/components/discriminators/light_cnn.py | 1 + mmedit/models/components/discriminators/modified_vgg.py | 1 + mmedit/models/components/discriminators/multi_layer_disc.py | 1 + mmedit/models/components/discriminators/patch_disc.py | 1 + mmedit/models/components/discriminators/ttsr_disc.py | 1 + mmedit/models/components/refiners/__init__.py | 1 + mmedit/models/components/refiners/deepfill_refiner.py | 1 + mmedit/models/components/refiners/mlp_refiner.py | 1 + mmedit/models/components/refiners/plain_refiner.py | 1 + mmedit/models/components/stylegan2/__init__.py | 1 + mmedit/models/components/stylegan2/common.py | 1 + mmedit/models/components/stylegan2/generator_discriminator.py | 1 + mmedit/models/components/stylegan2/modules.py | 1 + mmedit/models/extractors/__init__.py | 1 + mmedit/models/extractors/feedback_hour_glass.py | 1 + mmedit/models/extractors/lte.py | 1 + mmedit/models/inpaintors/__init__.py | 1 + mmedit/models/inpaintors/deepfillv1.py | 1 + mmedit/models/inpaintors/gl_inpaintor.py | 1 + mmedit/models/inpaintors/one_stage.py | 1 + mmedit/models/inpaintors/pconv_inpaintor.py | 1 + mmedit/models/inpaintors/two_stage.py | 1 + mmedit/models/losses/__init__.py | 1 + mmedit/models/losses/composition_loss.py | 1 + mmedit/models/losses/feature_loss.py | 1 + mmedit/models/losses/gan_loss.py | 1 + mmedit/models/losses/gradient_loss.py | 1 + mmedit/models/losses/perceptual_loss.py | 1 + mmedit/models/losses/pixelwise_loss.py | 1 + mmedit/models/losses/utils.py | 1 + mmedit/models/mattors/__init__.py | 1 + mmedit/models/mattors/base_mattor.py | 1 + mmedit/models/mattors/dim.py | 1 + mmedit/models/mattors/gca.py | 1 + mmedit/models/mattors/indexnet.py | 1 + mmedit/models/mattors/utils.py | 1 + mmedit/models/registry.py | 1 + mmedit/models/restorers/__init__.py | 1 + mmedit/models/restorers/basic_restorer.py | 1 + mmedit/models/restorers/basicvsr.py | 1 + mmedit/models/restorers/dic.py | 1 + mmedit/models/restorers/edvr.py | 1 + mmedit/models/restorers/esrgan.py | 1 + mmedit/models/restorers/glean.py | 1 + mmedit/models/restorers/liif.py | 1 + mmedit/models/restorers/srgan.py | 1 + mmedit/models/restorers/tdan.py | 1 + mmedit/models/restorers/ttsr.py | 1 + mmedit/models/synthesizers/__init__.py | 1 + mmedit/models/synthesizers/cycle_gan.py | 1 + mmedit/models/synthesizers/pix2pix.py | 1 + mmedit/models/transformers/__init__.py | 1 + mmedit/models/transformers/search_transformer.py | 1 + mmedit/utils/__init__.py | 1 + mmedit/utils/collect_env.py | 1 + mmedit/utils/logger.py | 1 + tests/data/inpaintor_config/gl_test.py | 1 + tests/data/inpaintor_config/one_stage_gl.py | 1 + tests/data/inpaintor_config/pconv_inpaintor_test.py | 1 + tests/test_data/test_datasets/test_generation_datasets.py | 1 + tests/test_data/test_datasets/test_matting_datasets.py | 1 + tests/test_data/test_datasets/test_repeat_dataset.py | 1 + tests/test_data/test_datasets/test_sr_dataset.py | 1 + tests/test_data/test_pipelines/test_augmentation.py | 1 + tests/test_data/test_pipelines/test_compose.py | 1 + tests/test_data/test_pipelines/test_crop.py | 1 + tests/test_data/test_pipelines/test_formating.py | 1 + tests/test_data/test_pipelines/test_generate_assistant.py | 1 + tests/test_data/test_pipelines/test_loading.py | 1 + tests/test_data/test_pipelines/test_normalization.py | 1 + tests/test_data/test_pipelines/test_pipeline_utils.py | 1 + tests/test_data/test_pipelines/test_random_down_sampling.py | 1 + tests/test_data/test_pipelines/test_trimap.py | 1 + tests/test_inference.py | 1 + tests/test_metrics/test_metrics.py | 1 + .../test_backbones/test_encoder_decoders/test_decoders.py | 1 + .../test_encoder_decoders/test_deepfill_decoder.py | 1 + .../test_backbones/test_encoder_decoders/test_deepfill_encdec.py | 1 + .../test_encoder_decoders/test_deepfill_encoder.py | 1 + .../test_backbones/test_encoder_decoders/test_encoder_decoder.py | 1 + .../test_backbones/test_encoder_decoders/test_encoders.py | 1 + .../test_backbones/test_encoder_decoders/test_gl_model.py | 1 + .../test_backbones/test_encoder_decoders/test_pconv_encdec.py | 1 + .../test_backbones/test_generation_backbones/test_generators.py | 1 + .../test_backbones/test_sr_backbones/test_basicvsr_net.py | 1 + .../test_backbones/test_sr_backbones/test_basicvsr_plusplus.py | 1 + .../test_models/test_backbones/test_sr_backbones/test_dic_net.py | 1 + tests/test_models/test_backbones/test_sr_backbones/test_duf.py | 1 + .../test_backbones/test_sr_backbones/test_edvr_net.py | 1 + .../test_backbones/test_sr_backbones/test_glean_net.py | 1 + .../test_models/test_backbones/test_sr_backbones/test_iconvsr.py | 1 + .../test_backbones/test_sr_backbones/test_liif_net.py | 1 + tests/test_models/test_backbones/test_sr_backbones/test_rdn.py | 1 + .../test_backbones/test_sr_backbones/test_sr_backbones.py | 1 + .../test_backbones/test_sr_backbones/test_tdan_net.py | 1 + tests/test_models/test_backbones/test_sr_backbones/test_tof.py | 1 + tests/test_models/test_base_model.py | 1 + tests/test_models/test_common/test_common_module.py | 1 + tests/test_models/test_common/test_flow_warp.py | 1 + tests/test_models/test_common/test_img_normalize.py | 1 + tests/test_models/test_common/test_model_utils.py | 1 + .../test_components/test_discriminators/test_deepfill_disc.py | 1 + .../test_components/test_discriminators/test_discriminators.py | 1 + .../test_components/test_discriminators/test_light_cnn.py | 1 + .../test_components/test_discriminators/test_multi_layer_disc.py | 1 + .../test_components/test_refiners/test_deepfill_refiner.py | 1 + .../test_components/test_refiners/test_matting_refiners.py | 1 + .../test_components/test_refiners/test_mlp_refiner.py | 1 + tests/test_models/test_components/test_stylegan2.py | 1 + tests/test_models/test_extractors/test_feedback_hour_glass.py | 1 + tests/test_models/test_extractors/test_lte.py | 1 + tests/test_models/test_inpaintors/test_deepfill_inpaintor.py | 1 + tests/test_models/test_inpaintors/test_gl_inpaintor.py | 1 + tests/test_models/test_inpaintors/test_one_stage_inpaintor.py | 1 + tests/test_models/test_inpaintors/test_pconv_inpaintor.py | 1 + tests/test_models/test_inpaintors/test_two_stage_inpaintor.py | 1 + tests/test_models/test_losses/test_feature_loss.py | 1 + tests/test_models/test_losses/test_losses.py | 1 + tests/test_models/test_mattors/test_mattors.py | 1 + tests/test_models/test_restorers/test_basic_restorer.py | 1 + tests/test_models/test_restorers/test_basicvsr_model.py | 1 + tests/test_models/test_restorers/test_dic_model.py | 1 + tests/test_models/test_restorers/test_edvr_model.py | 1 + tests/test_models/test_restorers/test_esrgan.py | 1 + tests/test_models/test_restorers/test_glean.py | 1 + tests/test_models/test_restorers/test_liif.py | 1 + tests/test_models/test_restorers/test_srgan.py | 1 + tests/test_models/test_restorers/test_tdan.py | 1 + tests/test_models/test_restorers/test_ttsr.py | 1 + tests/test_models/test_synthesizers/test_cyclegan.py | 1 + tests/test_models/test_transformer/test_search_transformer.py | 1 + tests/test_runtime/test_dataset_builder.py | 1 + tests/test_runtime/test_eval_hook.py | 1 + tests/test_runtime/test_optimizer.py | 1 + tests/test_runtime/test_visual_hook.py | 1 + tests/test_utils/test_mask_generation.py | 1 + tests/test_utils/test_onnx_wraper.py | 1 + tests/test_utils/test_pix2pix.py | 1 + tests/test_utils/test_tensor2img.py | 1 + tools/data/matting/bgm/preprocess_bgm_dataset.py | 1 + tools/data/matting/comp1k/check_extended_fg.py | 1 + tools/data/matting/comp1k/extend_fg.py | 1 + tools/data/matting/comp1k/filter_comp1k_anno.py | 1 + tools/data/matting/comp1k/preprocess_comp1k_dataset.py | 1 + tools/data/super-resolution/div2k/preprocess_div2k_dataset.py | 1 + tools/data/super-resolution/reds/preprocess_reds_dataset.py | 1 + .../super-resolution/vimeo90k/preprocess_vimeo90k_dataset.py | 1 + tools/deploy_test.py | 1 + tools/evaluate_comp1k.py | 1 + tools/get_flops.py | 1 + tools/onnx2tensorrt.py | 1 + tools/publish_model.py | 1 + tools/pytorch2onnx.py | 1 + tools/test.py | 1 + tools/train.py | 1 + 294 files changed, 294 insertions(+) diff --git a/demo/generation_demo.py b/demo/generation_demo.py index 738d2b0b61..4c573f71b9 100644 --- a/demo/generation_demo.py +++ b/demo/generation_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import mmcv diff --git a/demo/inpainting_demo.py b/demo/inpainting_demo.py index 8db2e7a46d..0e0012a4b6 100644 --- a/demo/inpainting_demo.py +++ b/demo/inpainting_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import mmcv diff --git a/demo/matting_demo.py b/demo/matting_demo.py index 5c020de8f8..10030ad75d 100644 --- a/demo/matting_demo.py +++ b/demo/matting_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import mmcv diff --git a/demo/restoration_demo.py b/demo/restoration_demo.py index 9348b6021d..84e39c5030 100644 --- a/demo/restoration_demo.py +++ b/demo/restoration_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os diff --git a/demo/restoration_video_demo.py b/demo/restoration_video_demo.py index 553f06d8a2..af1cc1b1b1 100644 --- a/demo/restoration_video_demo.py +++ b/demo/restoration_video_demo.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import mmcv diff --git a/docs/conf.py b/docs/conf.py index 6a4f235cff..e9ab808ac5 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 d182b610e7..9c40e835eb 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 re diff --git a/docs_zh-CN/conf.py b/docs_zh-CN/conf.py index 1dc501f3b7..1dfd0c73ed 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 b6d6992d96..42755729ef 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 re diff --git a/mmedit/__init__.py b/mmedit/__init__.py index 099c9493c5..cc48d64457 100644 --- a/mmedit/__init__.py +++ b/mmedit/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv from .version import __version__, version_info diff --git a/mmedit/apis/__init__.py b/mmedit/apis/__init__.py index a967763f3e..2752e72935 100644 --- a/mmedit/apis/__init__.py +++ b/mmedit/apis/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .generation_inference import generation_inference from .inpainting_inference import inpainting_inference from .matting_inference import init_model, matting_inference diff --git a/mmedit/apis/generation_inference.py b/mmedit/apis/generation_inference.py index 2e9facde2b..0cd3322ab7 100644 --- a/mmedit/apis/generation_inference.py +++ b/mmedit/apis/generation_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch from mmcv.parallel import collate, scatter diff --git a/mmedit/apis/inpainting_inference.py b/mmedit/apis/inpainting_inference.py index 0c6bfbc7be..3e74a2eb23 100644 --- a/mmedit/apis/inpainting_inference.py +++ b/mmedit/apis/inpainting_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.parallel import collate, scatter diff --git a/mmedit/apis/matting_inference.py b/mmedit/apis/matting_inference.py index e669716d00..b60b200841 100644 --- a/mmedit/apis/matting_inference.py +++ b/mmedit/apis/matting_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import torch from mmcv.parallel import collate, scatter diff --git a/mmedit/apis/restoration_inference.py b/mmedit/apis/restoration_inference.py index a441765357..44c5515ebb 100644 --- a/mmedit/apis/restoration_inference.py +++ b/mmedit/apis/restoration_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.parallel import collate, scatter diff --git a/mmedit/apis/restoration_video_inference.py b/mmedit/apis/restoration_video_inference.py index 7d3132b902..24f0e0aef9 100644 --- a/mmedit/apis/restoration_video_inference.py +++ b/mmedit/apis/restoration_video_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import glob import torch diff --git a/mmedit/apis/test.py b/mmedit/apis/test.py index 1dec48d3bf..4cd68eb076 100644 --- a/mmedit/apis/test.py +++ b/mmedit/apis/test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import pickle import shutil diff --git a/mmedit/apis/train.py b/mmedit/apis/train.py index f7d11d9b40..74f839b49e 100644 --- a/mmedit/apis/train.py +++ b/mmedit/apis/train.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import os.path as osp import random diff --git a/mmedit/core/__init__.py b/mmedit/core/__init__.py index 674c65ab4f..7a391e7edf 100644 --- a/mmedit/core/__init__.py +++ b/mmedit/core/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .evaluation import (DistEvalIterHook, EvalIterHook, L1Evaluation, mse, psnr, reorder_image, sad, ssim) from .hooks import VisualizationHook diff --git a/mmedit/core/distributed_wrapper.py b/mmedit/core/distributed_wrapper.py index cba5a6368b..660f41f3f5 100644 --- a/mmedit/core/distributed_wrapper.py +++ b/mmedit/core/distributed_wrapper.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.parallel import MODULE_WRAPPERS, MMDistributedDataParallel diff --git a/mmedit/core/evaluation/__init__.py b/mmedit/core/evaluation/__init__.py index a43f8fcdaf..00de861edf 100644 --- a/mmedit/core/evaluation/__init__.py +++ b/mmedit/core/evaluation/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .eval_hooks import DistEvalIterHook, EvalIterHook from .metrics import (L1Evaluation, connectivity, gradient_error, mse, niqe, psnr, reorder_image, sad, ssim) diff --git a/mmedit/core/evaluation/eval_hooks.py b/mmedit/core/evaluation/eval_hooks.py index 656776b3da..465731813b 100644 --- a/mmedit/core/evaluation/eval_hooks.py +++ b/mmedit/core/evaluation/eval_hooks.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from mmcv.runner import Hook diff --git a/mmedit/core/evaluation/metric_utils.py b/mmedit/core/evaluation/metric_utils.py index 2735a2e6cf..b5c06acf8e 100644 --- a/mmedit/core/evaluation/metric_utils.py +++ b/mmedit/core/evaluation/metric_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np diff --git a/mmedit/core/evaluation/metrics.py b/mmedit/core/evaluation/metrics.py index 279933628c..4ee4985e87 100644 --- a/mmedit/core/evaluation/metrics.py +++ b/mmedit/core/evaluation/metrics.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import cv2 diff --git a/mmedit/core/export/__init__.py b/mmedit/core/export/__init__.py index 05e413aad7..6cf757d83b 100644 --- a/mmedit/core/export/__init__.py +++ b/mmedit/core/export/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .wrappers import ONNXRuntimeEditing __all__ = ['ONNXRuntimeEditing'] diff --git a/mmedit/core/export/wrappers.py b/mmedit/core/export/wrappers.py index 9b63c7c121..a41bd718ef 100644 --- a/mmedit/core/export/wrappers.py +++ b/mmedit/core/export/wrappers.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import warnings diff --git a/mmedit/core/hooks/__init__.py b/mmedit/core/hooks/__init__.py index 46dd2fd455..9183c5015e 100644 --- a/mmedit/core/hooks/__init__.py +++ b/mmedit/core/hooks/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .visualization import VisualizationHook __all__ = ['VisualizationHook'] diff --git a/mmedit/core/hooks/visualization.py b/mmedit/core/hooks/visualization.py index 66753587d0..63c8ee7d8b 100644 --- a/mmedit/core/hooks/visualization.py +++ b/mmedit/core/hooks/visualization.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/mmedit/core/mask.py b/mmedit/core/mask.py index d0c76aa2e7..f80213c422 100644 --- a/mmedit/core/mask.py +++ b/mmedit/core/mask.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import cv2 diff --git a/mmedit/core/misc.py b/mmedit/core/misc.py index fff21f39c1..c5824068a9 100644 --- a/mmedit/core/misc.py +++ b/mmedit/core/misc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import numpy as np diff --git a/mmedit/core/optimizer/__init__.py b/mmedit/core/optimizer/__init__.py index ea6b36f641..e1c477dd1f 100644 --- a/mmedit/core/optimizer/__init__.py +++ b/mmedit/core/optimizer/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .builder import build_optimizers __all__ = ['build_optimizers'] diff --git a/mmedit/core/optimizer/builder.py b/mmedit/core/optimizer/builder.py index 28d8f65574..2edf94dada 100644 --- a/mmedit/core/optimizer/builder.py +++ b/mmedit/core/optimizer/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import build_optimizer diff --git a/mmedit/core/scheduler/__init__.py b/mmedit/core/scheduler/__init__.py index 94eb73c2e9..77245e7ca4 100644 --- a/mmedit/core/scheduler/__init__.py +++ b/mmedit/core/scheduler/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .lr_updater import LinearLrUpdaterHook __all__ = ['LinearLrUpdaterHook'] diff --git a/mmedit/core/scheduler/lr_updater.py b/mmedit/core/scheduler/lr_updater.py index fb9ad23b30..181824cea1 100644 --- a/mmedit/core/scheduler/lr_updater.py +++ b/mmedit/core/scheduler/lr_updater.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import HOOKS, LrUpdaterHook diff --git a/mmedit/datasets/__init__.py b/mmedit/datasets/__init__.py index b40eb5ff84..924e37c0b8 100644 --- a/mmedit/datasets/__init__.py +++ b/mmedit/datasets/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_dataset import BaseDataset from .base_generation_dataset import BaseGenerationDataset from .base_matting_dataset import BaseMattingDataset diff --git a/mmedit/datasets/base_dataset.py b/mmedit/datasets/base_dataset.py index a1cb4b1322..c8ffea6ebd 100644 --- a/mmedit/datasets/base_dataset.py +++ b/mmedit/datasets/base_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from abc import ABCMeta, abstractmethod diff --git a/mmedit/datasets/base_generation_dataset.py b/mmedit/datasets/base_generation_dataset.py index 4a4fc55010..c58b5af13d 100644 --- a/mmedit/datasets/base_generation_dataset.py +++ b/mmedit/datasets/base_generation_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from pathlib import Path diff --git a/mmedit/datasets/base_matting_dataset.py b/mmedit/datasets/base_matting_dataset.py index 35d838aee3..26a93325f4 100644 --- a/mmedit/datasets/base_matting_dataset.py +++ b/mmedit/datasets/base_matting_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections import defaultdict from .base_dataset import BaseDataset diff --git a/mmedit/datasets/base_sr_dataset.py b/mmedit/datasets/base_sr_dataset.py index 07ff48c8dc..090d14c7db 100644 --- a/mmedit/datasets/base_sr_dataset.py +++ b/mmedit/datasets/base_sr_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os.path as osp from collections import defaultdict diff --git a/mmedit/datasets/builder.py b/mmedit/datasets/builder.py index 243780ef26..f08e3b0572 100644 --- a/mmedit/datasets/builder.py +++ b/mmedit/datasets/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import platform import random diff --git a/mmedit/datasets/comp1k_dataset.py b/mmedit/datasets/comp1k_dataset.py index 320e5cd981..46b66936b4 100644 --- a/mmedit/datasets/comp1k_dataset.py +++ b/mmedit/datasets/comp1k_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/mmedit/datasets/dataset_wrappers.py b/mmedit/datasets/dataset_wrappers.py index 8c189c09f4..3dbca31ea6 100644 --- a/mmedit/datasets/dataset_wrappers.py +++ b/mmedit/datasets/dataset_wrappers.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .registry import DATASETS diff --git a/mmedit/datasets/generation_paired_dataset.py b/mmedit/datasets/generation_paired_dataset.py index 2f3ce262ac..d702a7cce5 100644 --- a/mmedit/datasets/generation_paired_dataset.py +++ b/mmedit/datasets/generation_paired_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_generation_dataset import BaseGenerationDataset diff --git a/mmedit/datasets/generation_unpaired_dataset.py b/mmedit/datasets/generation_unpaired_dataset.py index acf5ce9307..d98010b5e9 100644 --- a/mmedit/datasets/generation_unpaired_dataset.py +++ b/mmedit/datasets/generation_unpaired_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import numpy as np diff --git a/mmedit/datasets/img_inpainting_dataset.py b/mmedit/datasets/img_inpainting_dataset.py index 216e6b5c3e..4fb4562bdc 100644 --- a/mmedit/datasets/img_inpainting_dataset.py +++ b/mmedit/datasets/img_inpainting_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from pathlib import Path from .base_dataset import BaseDataset diff --git a/mmedit/datasets/pipelines/__init__.py b/mmedit/datasets/pipelines/__init__.py index c13aa348d3..74dadbda35 100644 --- a/mmedit/datasets/pipelines/__init__.py +++ b/mmedit/datasets/pipelines/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .augmentation import (BinarizeImage, Flip, GenerateFrameIndices, GenerateFrameIndiceswithPadding, GenerateSegmentIndices, MirrorSequence, Pad, diff --git a/mmedit/datasets/pipelines/augmentation.py b/mmedit/datasets/pipelines/augmentation.py index 91922bfbc9..bfbd498a4f 100644 --- a/mmedit/datasets/pipelines/augmentation.py +++ b/mmedit/datasets/pipelines/augmentation.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import numbers import os.path as osp diff --git a/mmedit/datasets/pipelines/compose.py b/mmedit/datasets/pipelines/compose.py index 9fcf2d1d73..0ffb0fd578 100644 --- a/mmedit/datasets/pipelines/compose.py +++ b/mmedit/datasets/pipelines/compose.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections.abc import Sequence from mmcv.utils import build_from_cfg diff --git a/mmedit/datasets/pipelines/crop.py b/mmedit/datasets/pipelines/crop.py index 1d1e4e00c9..9f79979ebc 100644 --- a/mmedit/datasets/pipelines/crop.py +++ b/mmedit/datasets/pipelines/crop.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import numpy as np from torch.nn.modules.utils import _pair diff --git a/mmedit/datasets/pipelines/formating.py b/mmedit/datasets/pipelines/formating.py index 4c1ce105d1..db4d3c6fcb 100644 --- a/mmedit/datasets/pipelines/formating.py +++ b/mmedit/datasets/pipelines/formating.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections.abc import Sequence import mmcv diff --git a/mmedit/datasets/pipelines/generate_assistant.py b/mmedit/datasets/pipelines/generate_assistant.py index 5de6e78b0f..8d83955341 100644 --- a/mmedit/datasets/pipelines/generate_assistant.py +++ b/mmedit/datasets/pipelines/generate_assistant.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch diff --git a/mmedit/datasets/pipelines/loading.py b/mmedit/datasets/pipelines/loading.py index 2c33282925..8fbefbf23f 100644 --- a/mmedit/datasets/pipelines/loading.py +++ b/mmedit/datasets/pipelines/loading.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from pathlib import Path import mmcv diff --git a/mmedit/datasets/pipelines/matting_aug.py b/mmedit/datasets/pipelines/matting_aug.py index 4092d2f99d..abe01ee425 100644 --- a/mmedit/datasets/pipelines/matting_aug.py +++ b/mmedit/datasets/pipelines/matting_aug.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import random diff --git a/mmedit/datasets/pipelines/normalization.py b/mmedit/datasets/pipelines/normalization.py index 1b41ef9e23..8ff774d7f9 100644 --- a/mmedit/datasets/pipelines/normalization.py +++ b/mmedit/datasets/pipelines/normalization.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import numpy as np diff --git a/mmedit/datasets/pipelines/random_down_sampling.py b/mmedit/datasets/pipelines/random_down_sampling.py index f1187f7f76..bba1ef7bbe 100644 --- a/mmedit/datasets/pipelines/random_down_sampling.py +++ b/mmedit/datasets/pipelines/random_down_sampling.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import numpy as np diff --git a/mmedit/datasets/pipelines/utils.py b/mmedit/datasets/pipelines/utils.py index 5eb4988bbd..42d470c349 100644 --- a/mmedit/datasets/pipelines/utils.py +++ b/mmedit/datasets/pipelines/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging import numpy as np diff --git a/mmedit/datasets/registry.py b/mmedit/datasets/registry.py index 2ebc753cd3..984580ef2b 100644 --- a/mmedit/datasets/registry.py +++ b/mmedit/datasets/registry.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import Registry DATASETS = Registry('dataset') diff --git a/mmedit/datasets/samplers/__init__.py b/mmedit/datasets/samplers/__init__.py index cffe4dcb0f..da09effaf2 100644 --- a/mmedit/datasets/samplers/__init__.py +++ b/mmedit/datasets/samplers/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .distributed_sampler import DistributedSampler __all__ = ['DistributedSampler'] diff --git a/mmedit/datasets/samplers/distributed_sampler.py b/mmedit/datasets/samplers/distributed_sampler.py index 1240129ba7..fced724c5a 100644 --- a/mmedit/datasets/samplers/distributed_sampler.py +++ b/mmedit/datasets/samplers/distributed_sampler.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from __future__ import division import math diff --git a/mmedit/datasets/sr_annotation_dataset.py b/mmedit/datasets/sr_annotation_dataset.py index 3e614f3d0e..9762e8d70c 100644 --- a/mmedit/datasets/sr_annotation_dataset.py +++ b/mmedit/datasets/sr_annotation_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_sr_dataset import BaseSRDataset diff --git a/mmedit/datasets/sr_facial_landmark_dataset.py b/mmedit/datasets/sr_facial_landmark_dataset.py index 1c37f02e47..77ad1401ff 100644 --- a/mmedit/datasets/sr_facial_landmark_dataset.py +++ b/mmedit/datasets/sr_facial_landmark_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import numpy as np diff --git a/mmedit/datasets/sr_folder_dataset.py b/mmedit/datasets/sr_folder_dataset.py index 5ff4c989fa..d82bed22ed 100644 --- a/mmedit/datasets/sr_folder_dataset.py +++ b/mmedit/datasets/sr_folder_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_sr_dataset import BaseSRDataset diff --git a/mmedit/datasets/sr_folder_gt_dataset.py b/mmedit/datasets/sr_folder_gt_dataset.py index 8966f9fc82..0ff4bfa153 100644 --- a/mmedit/datasets/sr_folder_gt_dataset.py +++ b/mmedit/datasets/sr_folder_gt_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_sr_dataset import BaseSRDataset from .registry import DATASETS diff --git a/mmedit/datasets/sr_folder_multiple_gt_dataset.py b/mmedit/datasets/sr_folder_multiple_gt_dataset.py index 2fece83ba4..71f45a2af3 100644 --- a/mmedit/datasets/sr_folder_multiple_gt_dataset.py +++ b/mmedit/datasets/sr_folder_multiple_gt_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import glob import os.path as osp diff --git a/mmedit/datasets/sr_folder_ref_dataset.py b/mmedit/datasets/sr_folder_ref_dataset.py index 835bc9b40a..2d1674f9db 100644 --- a/mmedit/datasets/sr_folder_ref_dataset.py +++ b/mmedit/datasets/sr_folder_ref_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_sr_dataset import BaseSRDataset diff --git a/mmedit/datasets/sr_folder_video_dataset.py b/mmedit/datasets/sr_folder_video_dataset.py index 81c02a6eaa..321aa23f69 100644 --- a/mmedit/datasets/sr_folder_video_dataset.py +++ b/mmedit/datasets/sr_folder_video_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import glob import os.path as osp from collections import defaultdict diff --git a/mmedit/datasets/sr_lmdb_dataset.py b/mmedit/datasets/sr_lmdb_dataset.py index 23dd5b9773..0f0aa09cad 100644 --- a/mmedit/datasets/sr_lmdb_dataset.py +++ b/mmedit/datasets/sr_lmdb_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_sr_dataset import BaseSRDataset diff --git a/mmedit/datasets/sr_reds_dataset.py b/mmedit/datasets/sr_reds_dataset.py index add7f821b6..9f4bcabcfe 100644 --- a/mmedit/datasets/sr_reds_dataset.py +++ b/mmedit/datasets/sr_reds_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_sr_dataset import BaseSRDataset from .registry import DATASETS diff --git a/mmedit/datasets/sr_reds_multiple_gt_dataset.py b/mmedit/datasets/sr_reds_multiple_gt_dataset.py index d27f98ed78..770bfeb03c 100644 --- a/mmedit/datasets/sr_reds_multiple_gt_dataset.py +++ b/mmedit/datasets/sr_reds_multiple_gt_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_sr_dataset import BaseSRDataset from .registry import DATASETS diff --git a/mmedit/datasets/sr_test_multiple_gt_dataset.py b/mmedit/datasets/sr_test_multiple_gt_dataset.py index 5ad625a588..e772b063b0 100644 --- a/mmedit/datasets/sr_test_multiple_gt_dataset.py +++ b/mmedit/datasets/sr_test_multiple_gt_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import glob import os.path as osp import warnings diff --git a/mmedit/datasets/sr_vid4_dataset.py b/mmedit/datasets/sr_vid4_dataset.py index 1c6227593f..50cae41749 100644 --- a/mmedit/datasets/sr_vid4_dataset.py +++ b/mmedit/datasets/sr_vid4_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections import defaultdict import numpy as np diff --git a/mmedit/datasets/sr_vimeo90k_dataset.py b/mmedit/datasets/sr_vimeo90k_dataset.py index c856936f5d..da0e434e4d 100644 --- a/mmedit/datasets/sr_vimeo90k_dataset.py +++ b/mmedit/datasets/sr_vimeo90k_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_sr_dataset import BaseSRDataset diff --git a/mmedit/datasets/sr_vimeo90k_multiple_gt_dataset.py b/mmedit/datasets/sr_vimeo90k_multiple_gt_dataset.py index 6541fe5004..578e66a771 100644 --- a/mmedit/datasets/sr_vimeo90k_multiple_gt_dataset.py +++ b/mmedit/datasets/sr_vimeo90k_multiple_gt_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from .base_sr_dataset import BaseSRDataset diff --git a/mmedit/models/__init__.py b/mmedit/models/__init__.py index eaa0a80417..5f8299aa82 100644 --- a/mmedit/models/__init__.py +++ b/mmedit/models/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .backbones import * # noqa: F401, F403 from .base import BaseModel from .builder import (build, build_backbone, build_component, build_loss, diff --git a/mmedit/models/backbones/__init__.py b/mmedit/models/backbones/__init__.py index de547fd339..a8f306bdc7 100644 --- a/mmedit/models/backbones/__init__.py +++ b/mmedit/models/backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .encoder_decoders import (VGG16, ContextualAttentionNeck, DeepFillDecoder, DeepFillEncoder, DeepFillEncoderDecoder, DepthwiseIndexBlock, FBADecoder, diff --git a/mmedit/models/backbones/encoder_decoders/__init__.py b/mmedit/models/backbones/encoder_decoders/__init__.py index 6155b171f6..652dd77c13 100644 --- a/mmedit/models/backbones/encoder_decoders/__init__.py +++ b/mmedit/models/backbones/encoder_decoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .decoders import (DeepFillDecoder, FBADecoder, GLDecoder, IndexedUpsample, IndexNetDecoder, PConvDecoder, PlainDecoder, ResGCADecoder, ResNetDec, ResShortcutDec) diff --git a/mmedit/models/backbones/encoder_decoders/decoders/__init__.py b/mmedit/models/backbones/encoder_decoders/decoders/__init__.py index 054bbc489d..45b05ac1da 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/__init__.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .deepfill_decoder import DeepFillDecoder from .fba_decoder import FBADecoder from .gl_decoder import GLDecoder diff --git a/mmedit/models/backbones/encoder_decoders/decoders/deepfill_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/deepfill_decoder.py index f8d38005d8..dd184eb6e6 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/deepfill_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/deepfill_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from functools import partial diff --git a/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py index 0d55c337a6..d1218125e8 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, constant_init, kaiming_init diff --git a/mmedit/models/backbones/encoder_decoders/decoders/gl_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/gl_decoder.py index 8ecfe750a7..78ab520ca7 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/gl_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/gl_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from functools import partial import torch diff --git a/mmedit/models/backbones/encoder_decoders/decoders/indexnet_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/indexnet_decoder.py index feeb5d7e22..4a14c8d9a9 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/indexnet_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/indexnet_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmedit/models/backbones/encoder_decoders/decoders/pconv_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/pconv_decoder.py index e18cb485ad..57d3d1b64a 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/pconv_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/pconv_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/mmedit/models/backbones/encoder_decoders/decoders/plain_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/plain_decoder.py index 4265734830..4660a4399a 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/plain_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/plain_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/mmedit/models/backbones/encoder_decoders/decoders/resnet_dec.py b/mmedit/models/backbones/encoder_decoders/decoders/resnet_dec.py index 8eaa4825c6..7af42e67c5 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/resnet_dec.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/resnet_dec.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule, constant_init diff --git a/mmedit/models/backbones/encoder_decoders/encoders/__init__.py b/mmedit/models/backbones/encoder_decoders/encoders/__init__.py index 2c1456e4c0..acf3c89188 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/__init__.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .deepfill_encoder import DeepFillEncoder from .fba_encoder import FBAResnetDilated from .gl_encoder import GLEncoder diff --git a/mmedit/models/backbones/encoder_decoders/encoders/deepfill_encoder.py b/mmedit/models/backbones/encoder_decoders/encoders/deepfill_encoder.py index 1b740a09d3..0a58e4bb5d 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/deepfill_encoder.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/deepfill_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmedit/models/backbones/encoder_decoders/encoders/fba_encoder.py b/mmedit/models/backbones/encoder_decoders/encoders/fba_encoder.py index d8f404d926..399b3b33f3 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/fba_encoder.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/fba_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmedit.models.registry import COMPONENTS from .resnet import ResNet diff --git a/mmedit/models/backbones/encoder_decoders/encoders/gl_encoder.py b/mmedit/models/backbones/encoder_decoders/encoders/gl_encoder.py index 46bd602c52..ae0546ece0 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/gl_encoder.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/gl_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmedit/models/backbones/encoder_decoders/encoders/indexnet_encoder.py b/mmedit/models/backbones/encoder_decoders/encoders/indexnet_encoder.py index 98b01cd217..963c349186 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/indexnet_encoder.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/indexnet_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from functools import partial import torch diff --git a/mmedit/models/backbones/encoder_decoders/encoders/pconv_encoder.py b/mmedit/models/backbones/encoder_decoders/encoders/pconv_encoder.py index 81265823fd..42d839ac3d 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/pconv_encoder.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/pconv_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.utils.parrots_wrapper import _BatchNorm diff --git a/mmedit/models/backbones/encoder_decoders/encoders/resnet.py b/mmedit/models/backbones/encoder_decoders/encoders/resnet.py index 79cb5fbbb0..e7c2b045e8 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/resnet.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/resnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn import torch.utils.checkpoint as cp from mmcv.cnn import (build_activation_layer, build_conv_layer, diff --git a/mmedit/models/backbones/encoder_decoders/encoders/resnet_enc.py b/mmedit/models/backbones/encoder_decoders/encoders/resnet_enc.py index 3f93ec4623..bfec7f6359 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/resnet_enc.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/resnet_enc.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, build_activation_layer, constant_init diff --git a/mmedit/models/backbones/encoder_decoders/encoders/vgg.py b/mmedit/models/backbones/encoder_decoders/encoders/vgg.py index 26fdc65ce3..9b4a441383 100644 --- a/mmedit/models/backbones/encoder_decoders/encoders/vgg.py +++ b/mmedit/models/backbones/encoder_decoders/encoders/vgg.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn.utils.weight_init import constant_init, xavier_init from mmcv.runner import load_checkpoint diff --git a/mmedit/models/backbones/encoder_decoders/gl_encoder_decoder.py b/mmedit/models/backbones/encoder_decoders/gl_encoder_decoder.py index 797c1013a3..9434e32416 100644 --- a/mmedit/models/backbones/encoder_decoders/gl_encoder_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/gl_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import auto_fp16, load_checkpoint diff --git a/mmedit/models/backbones/encoder_decoders/necks/__init__.py b/mmedit/models/backbones/encoder_decoders/necks/__init__.py index 9851f9db23..8db1cd00a6 100644 --- a/mmedit/models/backbones/encoder_decoders/necks/__init__.py +++ b/mmedit/models/backbones/encoder_decoders/necks/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .contextual_attention_neck import ContextualAttentionNeck from .gl_dilation import GLDilationNeck diff --git a/mmedit/models/backbones/encoder_decoders/necks/contextual_attention_neck.py b/mmedit/models/backbones/encoder_decoders/necks/contextual_attention_neck.py index e4638e18e5..28ffed4364 100644 --- a/mmedit/models/backbones/encoder_decoders/necks/contextual_attention_neck.py +++ b/mmedit/models/backbones/encoder_decoders/necks/contextual_attention_neck.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmedit/models/backbones/encoder_decoders/necks/gl_dilation.py b/mmedit/models/backbones/encoder_decoders/necks/gl_dilation.py index d1cfc27839..998f3900e1 100644 --- a/mmedit/models/backbones/encoder_decoders/necks/gl_dilation.py +++ b/mmedit/models/backbones/encoder_decoders/necks/gl_dilation.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmedit/models/backbones/encoder_decoders/pconv_encoder_decoder.py b/mmedit/models/backbones/encoder_decoders/pconv_encoder_decoder.py index 57983c400d..c4d5233b1f 100644 --- a/mmedit/models/backbones/encoder_decoders/pconv_encoder_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/pconv_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import auto_fp16, load_checkpoint diff --git a/mmedit/models/backbones/encoder_decoders/simple_encoder_decoder.py b/mmedit/models/backbones/encoder_decoders/simple_encoder_decoder.py index 4c6270ee27..69e8727cd9 100644 --- a/mmedit/models/backbones/encoder_decoders/simple_encoder_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/simple_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmedit.models.builder import build_component diff --git a/mmedit/models/backbones/encoder_decoders/two_stage_encoder_decoder.py b/mmedit/models/backbones/encoder_decoders/two_stage_encoder_decoder.py index a0d2975847..163e041722 100644 --- a/mmedit/models/backbones/encoder_decoders/two_stage_encoder_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/two_stage_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import constant_init, normal_init diff --git a/mmedit/models/backbones/generation_backbones/__init__.py b/mmedit/models/backbones/generation_backbones/__init__.py index 407c3b4ad7..e32c2dfb5d 100644 --- a/mmedit/models/backbones/generation_backbones/__init__.py +++ b/mmedit/models/backbones/generation_backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .resnet_generator import ResnetGenerator from .unet_generator import UnetGenerator diff --git a/mmedit/models/backbones/generation_backbones/resnet_generator.py b/mmedit/models/backbones/generation_backbones/resnet_generator.py index a946d1bb6f..389e1a16e8 100644 --- a/mmedit/models/backbones/generation_backbones/resnet_generator.py +++ b/mmedit/models/backbones/generation_backbones/resnet_generator.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule from mmcv.runner import load_checkpoint diff --git a/mmedit/models/backbones/generation_backbones/unet_generator.py b/mmedit/models/backbones/generation_backbones/unet_generator.py index 8d8a528f78..4e6622a4df 100644 --- a/mmedit/models/backbones/generation_backbones/unet_generator.py +++ b/mmedit/models/backbones/generation_backbones/unet_generator.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/backbones/sr_backbones/__init__.py b/mmedit/models/backbones/sr_backbones/__init__.py index 26d5e67bc0..224a3e3d7e 100644 --- a/mmedit/models/backbones/sr_backbones/__init__.py +++ b/mmedit/models/backbones/sr_backbones/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .basicvsr_net import BasicVSRNet from .basicvsr_pp import BasicVSRPlusPlus from .dic_net import DICNet diff --git a/mmedit/models/backbones/sr_backbones/basicvsr_net.py b/mmedit/models/backbones/sr_backbones/basicvsr_net.py index 21dbb18143..5794854ba6 100644 --- a/mmedit/models/backbones/sr_backbones/basicvsr_net.py +++ b/mmedit/models/backbones/sr_backbones/basicvsr_net.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/mmedit/models/backbones/sr_backbones/basicvsr_pp.py b/mmedit/models/backbones/sr_backbones/basicvsr_pp.py index 0e0417bc4c..579b12380a 100644 --- a/mmedit/models/backbones/sr_backbones/basicvsr_pp.py +++ b/mmedit/models/backbones/sr_backbones/basicvsr_pp.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import warnings import torch diff --git a/mmedit/models/backbones/sr_backbones/dic_net.py b/mmedit/models/backbones/sr_backbones/dic_net.py index 4ea8b47f49..3f4b8bfdb5 100644 --- a/mmedit/models/backbones/sr_backbones/dic_net.py +++ b/mmedit/models/backbones/sr_backbones/dic_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/backbones/sr_backbones/duf.py b/mmedit/models/backbones/sr_backbones/duf.py index 3294f3fc05..e38e23edea 100644 --- a/mmedit/models/backbones/sr_backbones/duf.py +++ b/mmedit/models/backbones/sr_backbones/duf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/mmedit/models/backbones/sr_backbones/edsr.py b/mmedit/models/backbones/sr_backbones/edsr.py index ca66ea8996..9bd470cdec 100644 --- a/mmedit/models/backbones/sr_backbones/edsr.py +++ b/mmedit/models/backbones/sr_backbones/edsr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import torch diff --git a/mmedit/models/backbones/sr_backbones/edvr_net.py b/mmedit/models/backbones/sr_backbones/edvr_net.py index ffe4f0f27f..5141764bc2 100644 --- a/mmedit/models/backbones/sr_backbones/edvr_net.py +++ b/mmedit/models/backbones/sr_backbones/edvr_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, constant_init, kaiming_init diff --git a/mmedit/models/backbones/sr_backbones/glean_styleganv2.py b/mmedit/models/backbones/sr_backbones/glean_styleganv2.py index 4dd3740d99..5976f793ae 100644 --- a/mmedit/models/backbones/sr_backbones/glean_styleganv2.py +++ b/mmedit/models/backbones/sr_backbones/glean_styleganv2.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/mmedit/models/backbones/sr_backbones/iconvsr.py b/mmedit/models/backbones/sr_backbones/iconvsr.py index eeaf0d544b..c1af811d31 100644 --- a/mmedit/models/backbones/sr_backbones/iconvsr.py +++ b/mmedit/models/backbones/sr_backbones/iconvsr.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/mmedit/models/backbones/sr_backbones/rrdb_net.py b/mmedit/models/backbones/sr_backbones/rrdb_net.py index 4ae4e2ff98..77d5d11b4e 100644 --- a/mmedit/models/backbones/sr_backbones/rrdb_net.py +++ b/mmedit/models/backbones/sr_backbones/rrdb_net.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/mmedit/models/backbones/sr_backbones/sr_resnet.py b/mmedit/models/backbones/sr_backbones/sr_resnet.py index 1916f855c6..c1f61242b7 100644 --- a/mmedit/models/backbones/sr_backbones/sr_resnet.py +++ b/mmedit/models/backbones/sr_backbones/sr_resnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/backbones/sr_backbones/srcnn.py b/mmedit/models/backbones/sr_backbones/srcnn.py index 9844056f5d..207c73b8bc 100644 --- a/mmedit/models/backbones/sr_backbones/srcnn.py +++ b/mmedit/models/backbones/sr_backbones/srcnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/backbones/sr_backbones/tdan_net.py b/mmedit/models/backbones/sr_backbones/tdan_net.py index e39e8d915d..238fc5b23d 100644 --- a/mmedit/models/backbones/sr_backbones/tdan_net.py +++ b/mmedit/models/backbones/sr_backbones/tdan_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, constant_init diff --git a/mmedit/models/backbones/sr_backbones/tof.py b/mmedit/models/backbones/sr_backbones/tof.py index 0348ae3092..c3b9beea03 100644 --- a/mmedit/models/backbones/sr_backbones/tof.py +++ b/mmedit/models/backbones/sr_backbones/tof.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/mmedit/models/base.py b/mmedit/models/base.py index a7b48e0f50..02327e2832 100644 --- a/mmedit/models/base.py +++ b/mmedit/models/base.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from abc import ABCMeta, abstractmethod from collections import OrderedDict diff --git a/mmedit/models/builder.py b/mmedit/models/builder.py index 9b0b320f1b..8606225aa6 100644 --- a/mmedit/models/builder.py +++ b/mmedit/models/builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv import build_from_cfg diff --git a/mmedit/models/common/__init__.py b/mmedit/models/common/__init__.py index 11b44b4a11..0828242362 100644 --- a/mmedit/models/common/__init__.py +++ b/mmedit/models/common/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .aspp import ASPP from .contextual_attention import ContextualAttentionModule from .conv import * # noqa: F401, F403 diff --git a/mmedit/models/common/aspp.py b/mmedit/models/common/aspp.py index 6b95080d03..c1e58e858a 100644 --- a/mmedit/models/common/aspp.py +++ b/mmedit/models/common/aspp.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.cnn import ConvModule from torch import nn diff --git a/mmedit/models/common/contextual_attention.py b/mmedit/models/common/contextual_attention.py index 3b28b4a71e..7dcf4099ee 100644 --- a/mmedit/models/common/contextual_attention.py +++ b/mmedit/models/common/contextual_attention.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from functools import partial import torch diff --git a/mmedit/models/common/conv.py b/mmedit/models/common/conv.py index e62412e954..8e03d0f7d7 100644 --- a/mmedit/models/common/conv.py +++ b/mmedit/models/common/conv.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.cnn import CONV_LAYERS from torch import nn diff --git a/mmedit/models/common/flow_warp.py b/mmedit/models/common/flow_warp.py index 05118677b1..b38f4bcdb8 100644 --- a/mmedit/models/common/flow_warp.py +++ b/mmedit/models/common/flow_warp.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn.functional as F diff --git a/mmedit/models/common/gated_conv_module.py b/mmedit/models/common/gated_conv_module.py index 33222245ff..b17b9ff469 100644 --- a/mmedit/models/common/gated_conv_module.py +++ b/mmedit/models/common/gated_conv_module.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import torch diff --git a/mmedit/models/common/gca_module.py b/mmedit/models/common/gca_module.py index 3588061b9d..75ab64e5b1 100644 --- a/mmedit/models/common/gca_module.py +++ b/mmedit/models/common/gca_module.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn import ConvModule, constant_init, xavier_init diff --git a/mmedit/models/common/generation_model_utils.py b/mmedit/models/common/generation_model_utils.py index c493f62f9d..21d45fbe34 100644 --- a/mmedit/models/common/generation_model_utils.py +++ b/mmedit/models/common/generation_model_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/mmedit/models/common/img_normalize.py b/mmedit/models/common/img_normalize.py index 38e3294644..1bd8f76aa6 100644 --- a/mmedit/models/common/img_normalize.py +++ b/mmedit/models/common/img_normalize.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn diff --git a/mmedit/models/common/linear_module.py b/mmedit/models/common/linear_module.py index 34b438db03..7445a31ba6 100644 --- a/mmedit/models/common/linear_module.py +++ b/mmedit/models/common/linear_module.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import build_activation_layer, kaiming_init diff --git a/mmedit/models/common/mask_conv_module.py b/mmedit/models/common/mask_conv_module.py index d7556477e3..e708686867 100644 --- a/mmedit/models/common/mask_conv_module.py +++ b/mmedit/models/common/mask_conv_module.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.cnn import ConvModule diff --git a/mmedit/models/common/model_utils.py b/mmedit/models/common/model_utils.py index 32cd0bbb36..62135c6e61 100644 --- a/mmedit/models/common/model_utils.py +++ b/mmedit/models/common/model_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch diff --git a/mmedit/models/common/partial_conv.py b/mmedit/models/common/partial_conv.py index 21083714b6..0664f2808f 100644 --- a/mmedit/models/common/partial_conv.py +++ b/mmedit/models/common/partial_conv.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/mmedit/models/common/separable_conv_module.py b/mmedit/models/common/separable_conv_module.py index cc8badc1d6..139a54e58c 100644 --- a/mmedit/models/common/separable_conv_module.py +++ b/mmedit/models/common/separable_conv_module.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule diff --git a/mmedit/models/common/sr_backbone_utils.py b/mmedit/models/common/sr_backbone_utils.py index a4bd8569e9..b4b0aad915 100644 --- a/mmedit/models/common/sr_backbone_utils.py +++ b/mmedit/models/common/sr_backbone_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import constant_init, kaiming_init from mmcv.utils.parrots_wrapper import _BatchNorm diff --git a/mmedit/models/common/upsample.py b/mmedit/models/common/upsample.py index 1dd55357d4..f39ec1a9e4 100644 --- a/mmedit/models/common/upsample.py +++ b/mmedit/models/common/upsample.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn import torch.nn.functional as F diff --git a/mmedit/models/components/__init__.py b/mmedit/models/components/__init__.py index c810032202..ce83d04cc2 100644 --- a/mmedit/models/components/__init__.py +++ b/mmedit/models/components/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .discriminators import (DeepFillv1Discriminators, GLDiscs, ModifiedVGG, MultiLayerDiscriminator, PatchDiscriminator) from .refiners import DeepFillRefiner, PlainRefiner diff --git a/mmedit/models/components/discriminators/__init__.py b/mmedit/models/components/discriminators/__init__.py index 6882561784..d533651fe6 100644 --- a/mmedit/models/components/discriminators/__init__.py +++ b/mmedit/models/components/discriminators/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .deepfill_disc import DeepFillv1Discriminators from .gl_disc import GLDiscs from .light_cnn import LightCNN diff --git a/mmedit/models/components/discriminators/deepfill_disc.py b/mmedit/models/components/discriminators/deepfill_disc.py index f679ab378f..0205b66c35 100644 --- a/mmedit/models/components/discriminators/deepfill_disc.py +++ b/mmedit/models/components/discriminators/deepfill_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import normal_init from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/discriminators/gl_disc.py b/mmedit/models/components/discriminators/gl_disc.py index 99cc5ff898..dec1493df6 100644 --- a/mmedit/models/components/discriminators/gl_disc.py +++ b/mmedit/models/components/discriminators/gl_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/discriminators/light_cnn.py b/mmedit/models/components/discriminators/light_cnn.py index 34b91bb639..a019d1dc61 100644 --- a/mmedit/models/components/discriminators/light_cnn.py +++ b/mmedit/models/components/discriminators/light_cnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/discriminators/modified_vgg.py b/mmedit/models/components/discriminators/modified_vgg.py index de3977303b..3aecaf9a40 100644 --- a/mmedit/models/components/discriminators/modified_vgg.py +++ b/mmedit/models/components/discriminators/modified_vgg.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/discriminators/multi_layer_disc.py b/mmedit/models/components/discriminators/multi_layer_disc.py index 04b5d4fd37..523ab99663 100644 --- a/mmedit/models/components/discriminators/multi_layer_disc.py +++ b/mmedit/models/components/discriminators/multi_layer_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/discriminators/patch_disc.py b/mmedit/models/components/discriminators/patch_disc.py index 7e14e5c4a6..aa08766521 100644 --- a/mmedit/models/components/discriminators/patch_disc.py +++ b/mmedit/models/components/discriminators/patch_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.cnn import ConvModule, build_conv_layer from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/discriminators/ttsr_disc.py b/mmedit/models/components/discriminators/ttsr_disc.py index 3441f36139..bc0c3e44f7 100644 --- a/mmedit/models/components/discriminators/ttsr_disc.py +++ b/mmedit/models/components/discriminators/ttsr_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/refiners/__init__.py b/mmedit/models/components/refiners/__init__.py index 3ddf6f0bdd..8b7da20788 100644 --- a/mmedit/models/components/refiners/__init__.py +++ b/mmedit/models/components/refiners/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .deepfill_refiner import DeepFillRefiner from .mlp_refiner import MLPRefiner from .plain_refiner import PlainRefiner diff --git a/mmedit/models/components/refiners/deepfill_refiner.py b/mmedit/models/components/refiners/deepfill_refiner.py index e5ad872524..8b42cb04fb 100644 --- a/mmedit/models/components/refiners/deepfill_refiner.py +++ b/mmedit/models/components/refiners/deepfill_refiner.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/mmedit/models/components/refiners/mlp_refiner.py b/mmedit/models/components/refiners/mlp_refiner.py index 1966f6b25b..1d6e09f338 100644 --- a/mmedit/models/components/refiners/mlp_refiner.py +++ b/mmedit/models/components/refiners/mlp_refiner.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/components/refiners/plain_refiner.py b/mmedit/models/components/refiners/plain_refiner.py index 74694718b6..1804a26cf8 100644 --- a/mmedit/models/components/refiners/plain_refiner.py +++ b/mmedit/models/components/refiners/plain_refiner.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.cnn.utils.weight_init import xavier_init diff --git a/mmedit/models/components/stylegan2/__init__.py b/mmedit/models/components/stylegan2/__init__.py index 7b20b345a4..8ccb2e9628 100644 --- a/mmedit/models/components/stylegan2/__init__.py +++ b/mmedit/models/components/stylegan2/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .generator_discriminator import (StyleGAN2Discriminator, StyleGANv2Generator) diff --git a/mmedit/models/components/stylegan2/common.py b/mmedit/models/components/stylegan2/common.py index 90b3ccc229..1be25c3e48 100644 --- a/mmedit/models/components/stylegan2/common.py +++ b/mmedit/models/components/stylegan2/common.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch diff --git a/mmedit/models/components/stylegan2/generator_discriminator.py b/mmedit/models/components/stylegan2/generator_discriminator.py index 64347df1ac..873d4a96de 100644 --- a/mmedit/models/components/stylegan2/generator_discriminator.py +++ b/mmedit/models/components/stylegan2/generator_discriminator.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import random import mmcv diff --git a/mmedit/models/components/stylegan2/modules.py b/mmedit/models/components/stylegan2/modules.py index a9c73e4711..455a0d6310 100644 --- a/mmedit/models/components/stylegan2/modules.py +++ b/mmedit/models/components/stylegan2/modules.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math from copy import deepcopy from functools import partial diff --git a/mmedit/models/extractors/__init__.py b/mmedit/models/extractors/__init__.py index 75f0a114f9..d93751781b 100644 --- a/mmedit/models/extractors/__init__.py +++ b/mmedit/models/extractors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .feedback_hour_glass import (FeedbackHourglass, Hourglass, reduce_to_five_heatmaps) from .lte import LTE diff --git a/mmedit/models/extractors/feedback_hour_glass.py b/mmedit/models/extractors/feedback_hour_glass.py index 75aaf3a448..3c8605763b 100644 --- a/mmedit/models/extractors/feedback_hour_glass.py +++ b/mmedit/models/extractors/feedback_hour_glass.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn diff --git a/mmedit/models/extractors/lte.py b/mmedit/models/extractors/lte.py index 30a2fa4487..786052d08d 100644 --- a/mmedit/models/extractors/lte.py +++ b/mmedit/models/extractors/lte.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/inpaintors/__init__.py b/mmedit/models/inpaintors/__init__.py index e51f90c83e..f3a4c093dd 100644 --- a/mmedit/models/inpaintors/__init__.py +++ b/mmedit/models/inpaintors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .deepfillv1 import DeepFillv1Inpaintor from .gl_inpaintor import GLInpaintor from .one_stage import OneStageInpaintor diff --git a/mmedit/models/inpaintors/deepfillv1.py b/mmedit/models/inpaintors/deepfillv1.py index 0a7d108ea6..e667eb5abc 100644 --- a/mmedit/models/inpaintors/deepfillv1.py +++ b/mmedit/models/inpaintors/deepfillv1.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from torch.nn.parallel import DataParallel, DistributedDataParallel diff --git a/mmedit/models/inpaintors/gl_inpaintor.py b/mmedit/models/inpaintors/gl_inpaintor.py index f1a77fc639..dee37ec2eb 100644 --- a/mmedit/models/inpaintors/gl_inpaintor.py +++ b/mmedit/models/inpaintors/gl_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from ..common import extract_around_bbox, extract_bbox_patch, set_requires_grad diff --git a/mmedit/models/inpaintors/one_stage.py b/mmedit/models/inpaintors/one_stage.py index d5de992fee..607477cf95 100644 --- a/mmedit/models/inpaintors/one_stage.py +++ b/mmedit/models/inpaintors/one_stage.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from pathlib import Path diff --git a/mmedit/models/inpaintors/pconv_inpaintor.py b/mmedit/models/inpaintors/pconv_inpaintor.py index eff6f9b8d3..f9ef7b4091 100644 --- a/mmedit/models/inpaintors/pconv_inpaintor.py +++ b/mmedit/models/inpaintors/pconv_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from pathlib import Path diff --git a/mmedit/models/inpaintors/two_stage.py b/mmedit/models/inpaintors/two_stage.py index 7f48770cc8..dbf4d59853 100644 --- a/mmedit/models/inpaintors/two_stage.py +++ b/mmedit/models/inpaintors/two_stage.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from pathlib import Path diff --git a/mmedit/models/losses/__init__.py b/mmedit/models/losses/__init__.py index 0eba5bb667..5f3ae4e674 100644 --- a/mmedit/models/losses/__init__.py +++ b/mmedit/models/losses/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .composition_loss import (CharbonnierCompLoss, L1CompositionLoss, MSECompositionLoss) from .feature_loss import LightCNNFeatureLoss diff --git a/mmedit/models/losses/composition_loss.py b/mmedit/models/losses/composition_loss.py index cc0d3ace00..5d6b7a9d1e 100644 --- a/mmedit/models/losses/composition_loss.py +++ b/mmedit/models/losses/composition_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch.nn as nn from ..registry import LOSSES diff --git a/mmedit/models/losses/feature_loss.py b/mmedit/models/losses/feature_loss.py index 1791f2554a..f80eb30107 100644 --- a/mmedit/models/losses/feature_loss.py +++ b/mmedit/models/losses/feature_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn from mmcv.runner import load_checkpoint diff --git a/mmedit/models/losses/gan_loss.py b/mmedit/models/losses/gan_loss.py index 469716303a..affc0065cc 100644 --- a/mmedit/models/losses/gan_loss.py +++ b/mmedit/models/losses/gan_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.autograd as autograd import torch.nn as nn diff --git a/mmedit/models/losses/gradient_loss.py b/mmedit/models/losses/gradient_loss.py index a692593d91..b556c1e5c4 100644 --- a/mmedit/models/losses/gradient_loss.py +++ b/mmedit/models/losses/gradient_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/mmedit/models/losses/perceptual_loss.py b/mmedit/models/losses/perceptual_loss.py index 809615225e..bb6daddc3c 100644 --- a/mmedit/models/losses/perceptual_loss.py +++ b/mmedit/models/losses/perceptual_loss.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn import torchvision.models.vgg as vgg diff --git a/mmedit/models/losses/pixelwise_loss.py b/mmedit/models/losses/pixelwise_loss.py index 632f343877..2f2435b8d3 100644 --- a/mmedit/models/losses/pixelwise_loss.py +++ b/mmedit/models/losses/pixelwise_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/mmedit/models/losses/utils.py b/mmedit/models/losses/utils.py index e300fb9732..2f536d9243 100644 --- a/mmedit/models/losses/utils.py +++ b/mmedit/models/losses/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import functools import torch.nn.functional as F diff --git a/mmedit/models/mattors/__init__.py b/mmedit/models/mattors/__init__.py index 498de9578d..02074f1f8c 100644 --- a/mmedit/models/mattors/__init__.py +++ b/mmedit/models/mattors/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .base_mattor import BaseMattor from .dim import DIM from .gca import GCA diff --git a/mmedit/models/mattors/base_mattor.py b/mmedit/models/mattors/base_mattor.py index c83b2de323..76279a33f3 100644 --- a/mmedit/models/mattors/base_mattor.py +++ b/mmedit/models/mattors/base_mattor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging import os.path as osp from abc import abstractmethod diff --git a/mmedit/models/mattors/dim.py b/mmedit/models/mattors/dim.py index 4f95eb4c28..c06d80864d 100644 --- a/mmedit/models/mattors/dim.py +++ b/mmedit/models/mattors/dim.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.runner import auto_fp16 diff --git a/mmedit/models/mattors/gca.py b/mmedit/models/mattors/gca.py index 46fbf309b2..2f7cc6d875 100644 --- a/mmedit/models/mattors/gca.py +++ b/mmedit/models/mattors/gca.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.runner import auto_fp16 diff --git a/mmedit/models/mattors/indexnet.py b/mmedit/models/mattors/indexnet.py index e886df3759..22bd053698 100644 --- a/mmedit/models/mattors/indexnet.py +++ b/mmedit/models/mattors/indexnet.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv.runner import auto_fp16 diff --git a/mmedit/models/mattors/utils.py b/mmedit/models/mattors/utils.py index 4edbc5b587..5ff9da05c2 100644 --- a/mmedit/models/mattors/utils.py +++ b/mmedit/models/mattors/utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch diff --git a/mmedit/models/registry.py b/mmedit/models/registry.py index 2936de960d..41fae9d263 100644 --- a/mmedit/models/registry.py +++ b/mmedit/models/registry.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.utils import Registry MODELS = Registry('model') diff --git a/mmedit/models/restorers/__init__.py b/mmedit/models/restorers/__init__.py index 1f5e41eb89..0f69055a65 100644 --- a/mmedit/models/restorers/__init__.py +++ b/mmedit/models/restorers/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .basic_restorer import BasicRestorer from .basicvsr import BasicVSR from .dic import DIC diff --git a/mmedit/models/restorers/basic_restorer.py b/mmedit/models/restorers/basic_restorer.py index 9d6052b350..5114b8e9af 100644 --- a/mmedit/models/restorers/basic_restorer.py +++ b/mmedit/models/restorers/basic_restorer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/restorers/basicvsr.py b/mmedit/models/restorers/basicvsr.py index 6f69ca0adc..631c782cb2 100644 --- a/mmedit/models/restorers/basicvsr.py +++ b/mmedit/models/restorers/basicvsr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/restorers/dic.py b/mmedit/models/restorers/dic.py index 7b32eb747b..df62385268 100644 --- a/mmedit/models/restorers/dic.py +++ b/mmedit/models/restorers/dic.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/restorers/edvr.py b/mmedit/models/restorers/edvr.py index 8b582329d5..e97a5833f2 100644 --- a/mmedit/models/restorers/edvr.py +++ b/mmedit/models/restorers/edvr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/restorers/esrgan.py b/mmedit/models/restorers/esrgan.py index 823101c528..4e787c0e44 100644 --- a/mmedit/models/restorers/esrgan.py +++ b/mmedit/models/restorers/esrgan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from ..common import set_requires_grad diff --git a/mmedit/models/restorers/glean.py b/mmedit/models/restorers/glean.py index f69b2600e9..ed0b6f29e9 100644 --- a/mmedit/models/restorers/glean.py +++ b/mmedit/models/restorers/glean.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/restorers/liif.py b/mmedit/models/restorers/liif.py index 0ca88dd241..f9462ca26e 100644 --- a/mmedit/models/restorers/liif.py +++ b/mmedit/models/restorers/liif.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math import numbers import os.path as osp diff --git a/mmedit/models/restorers/srgan.py b/mmedit/models/restorers/srgan.py index 21ffa46088..3a9e7c045d 100644 --- a/mmedit/models/restorers/srgan.py +++ b/mmedit/models/restorers/srgan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from mmcv.runner import auto_fp16 from ..builder import build_backbone, build_component, build_loss diff --git a/mmedit/models/restorers/tdan.py b/mmedit/models/restorers/tdan.py index fc1895c5ab..10ff383e8b 100644 --- a/mmedit/models/restorers/tdan.py +++ b/mmedit/models/restorers/tdan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/restorers/ttsr.py b/mmedit/models/restorers/ttsr.py index 617e509823..fceab07580 100644 --- a/mmedit/models/restorers/ttsr.py +++ b/mmedit/models/restorers/ttsr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numbers import os.path as osp diff --git a/mmedit/models/synthesizers/__init__.py b/mmedit/models/synthesizers/__init__.py index e680b54f54..fcc328e4fc 100644 --- a/mmedit/models/synthesizers/__init__.py +++ b/mmedit/models/synthesizers/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .cycle_gan import CycleGAN from .pix2pix import Pix2Pix diff --git a/mmedit/models/synthesizers/cycle_gan.py b/mmedit/models/synthesizers/cycle_gan.py index af68e4304e..819c7093e4 100644 --- a/mmedit/models/synthesizers/cycle_gan.py +++ b/mmedit/models/synthesizers/cycle_gan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/mmedit/models/synthesizers/pix2pix.py b/mmedit/models/synthesizers/pix2pix.py index daa404a33a..8ab5147085 100644 --- a/mmedit/models/synthesizers/pix2pix.py +++ b/mmedit/models/synthesizers/pix2pix.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import mmcv diff --git a/mmedit/models/transformers/__init__.py b/mmedit/models/transformers/__init__.py index 103d6aa8b9..488957ee17 100644 --- a/mmedit/models/transformers/__init__.py +++ b/mmedit/models/transformers/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .search_transformer import SearchTransformer __all__ = ['SearchTransformer'] diff --git a/mmedit/models/transformers/search_transformer.py b/mmedit/models/transformers/search_transformer.py index db7f2be26a..1bcfbc9ce8 100644 --- a/mmedit/models/transformers/search_transformer.py +++ b/mmedit/models/transformers/search_transformer.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/mmedit/utils/__init__.py b/mmedit/utils/__init__.py index 013f1dfbcb..12eed4cfe8 100644 --- a/mmedit/utils/__init__.py +++ b/mmedit/utils/__init__.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from .logger import get_root_logger __all__ = ['get_root_logger'] diff --git a/mmedit/utils/collect_env.py b/mmedit/utils/collect_env.py index faae63b698..af22082c76 100644 --- a/mmedit/utils/collect_env.py +++ b/mmedit/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/mmedit/utils/logger.py b/mmedit/utils/logger.py index 53216d4e33..cdc340f18b 100644 --- a/mmedit/utils/logger.py +++ b/mmedit/utils/logger.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging from mmcv.utils import get_logger diff --git a/tests/data/inpaintor_config/gl_test.py b/tests/data/inpaintor_config/gl_test.py index 8979638e5b..08568e9f42 100644 --- a/tests/data/inpaintor_config/gl_test.py +++ b/tests/data/inpaintor_config/gl_test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. global_disc_cfg = dict( in_channels=3, max_channels=512, diff --git a/tests/data/inpaintor_config/one_stage_gl.py b/tests/data/inpaintor_config/one_stage_gl.py index 392d826853..67bfdb8dd6 100644 --- a/tests/data/inpaintor_config/one_stage_gl.py +++ b/tests/data/inpaintor_config/one_stage_gl.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. model = dict( type='OneStageInpaintor', encdec=dict( diff --git a/tests/data/inpaintor_config/pconv_inpaintor_test.py b/tests/data/inpaintor_config/pconv_inpaintor_test.py index f46f0d3f69..35baa70f42 100644 --- a/tests/data/inpaintor_config/pconv_inpaintor_test.py +++ b/tests/data/inpaintor_config/pconv_inpaintor_test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. model = dict( type='PConvInpaintor', encdec=dict( diff --git a/tests/test_data/test_datasets/test_generation_datasets.py b/tests/test_data/test_datasets/test_generation_datasets.py index e643033af0..347588d64a 100644 --- a/tests/test_data/test_datasets/test_generation_datasets.py +++ b/tests/test_data/test_datasets/test_generation_datasets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from pathlib import Path import pytest diff --git a/tests/test_data/test_datasets/test_matting_datasets.py b/tests/test_data/test_datasets/test_matting_datasets.py index b270141414..7333f38156 100644 --- a/tests/test_data/test_datasets/test_matting_datasets.py +++ b/tests/test_data/test_datasets/test_matting_datasets.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from pathlib import Path diff --git a/tests/test_data/test_datasets/test_repeat_dataset.py b/tests/test_data/test_datasets/test_repeat_dataset.py index 136623c574..f0ae7a11b4 100644 --- a/tests/test_data/test_datasets/test_repeat_dataset.py +++ b/tests/test_data/test_datasets/test_repeat_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from torch.utils.data import Dataset from mmedit.datasets import RepeatDataset diff --git a/tests/test_data/test_datasets/test_sr_dataset.py b/tests/test_data/test_datasets/test_sr_dataset.py index 5d264ecf12..28a95c9bad 100644 --- a/tests/test_data/test_datasets/test_sr_dataset.py +++ b/tests/test_data/test_datasets/test_sr_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from pathlib import Path from unittest.mock import patch diff --git a/tests/test_data/test_pipelines/test_augmentation.py b/tests/test_data/test_pipelines/test_augmentation.py index 9810e478d6..22c4c4f8dc 100644 --- a/tests/test_data/test_pipelines/test_augmentation.py +++ b/tests/test_data/test_pipelines/test_augmentation.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import numpy as np diff --git a/tests/test_data/test_pipelines/test_compose.py b/tests/test_data/test_pipelines/test_compose.py index aa8cc69c93..712cac8d6a 100644 --- a/tests/test_data/test_pipelines/test_compose.py +++ b/tests/test_data/test_pipelines/test_compose.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest diff --git a/tests/test_data/test_pipelines/test_crop.py b/tests/test_data/test_pipelines/test_crop.py index 2db7c7f380..d437cfcf36 100644 --- a/tests/test_data/test_pipelines/test_crop.py +++ b/tests/test_data/test_pipelines/test_crop.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import numpy as np diff --git a/tests/test_data/test_pipelines/test_formating.py b/tests/test_data/test_pipelines/test_formating.py index 4be3999bac..9c61d5e45d 100644 --- a/tests/test_data/test_pipelines/test_formating.py +++ b/tests/test_data/test_pipelines/test_formating.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest import torch diff --git a/tests/test_data/test_pipelines/test_generate_assistant.py b/tests/test_data/test_pipelines/test_generate_assistant.py index 3037ccb271..fce5d0e373 100644 --- a/tests/test_data/test_pipelines/test_generate_assistant.py +++ b/tests/test_data/test_pipelines/test_generate_assistant.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.datasets.pipelines import (GenerateCoordinateAndCell, diff --git a/tests/test_data/test_pipelines/test_loading.py b/tests/test_data/test_pipelines/test_loading.py index 3d760d9ca2..6e9a51f542 100644 --- a/tests/test_data/test_pipelines/test_loading.py +++ b/tests/test_data/test_pipelines/test_loading.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os.path as osp from pathlib import Path diff --git a/tests/test_data/test_pipelines/test_normalization.py b/tests/test_data/test_pipelines/test_normalization.py index 0e69317993..7d879854cc 100644 --- a/tests/test_data/test_pipelines/test_normalization.py +++ b/tests/test_data/test_pipelines/test_normalization.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest diff --git a/tests/test_data/test_pipelines/test_pipeline_utils.py b/tests/test_data/test_pipelines/test_pipeline_utils.py index b5997feef6..14b8bc13d3 100644 --- a/tests/test_data/test_pipelines/test_pipeline_utils.py +++ b/tests/test_data/test_pipelines/test_pipeline_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_data/test_pipelines/test_random_down_sampling.py b/tests/test_data/test_pipelines/test_random_down_sampling.py index 7b6749026d..4417be157b 100644 --- a/tests/test_data/test_pipelines/test_random_down_sampling.py +++ b/tests/test_data/test_pipelines/test_random_down_sampling.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np from mmedit.datasets.pipelines import RandomDownSampling diff --git a/tests/test_data/test_pipelines/test_trimap.py b/tests/test_data/test_pipelines/test_trimap.py index 1aa40fe975..19cd68d00e 100644 --- a/tests/test_data/test_pipelines/test_trimap.py +++ b/tests/test_data/test_pipelines/test_trimap.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import cv2 import numpy as np import pytest diff --git a/tests/test_inference.py b/tests/test_inference.py index 737e81e3e8..c55e0e103d 100644 --- a/tests/test_inference.py +++ b/tests/test_inference.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_metrics/test_metrics.py b/tests/test_metrics/test_metrics.py index 60e92499e3..9eef594319 100644 --- a/tests/test_metrics/test_metrics.py +++ b/tests/test_metrics/test_metrics.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import numpy as np import pytest diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_decoders.py b/tests/test_models/test_backbones/test_encoder_decoders/test_decoders.py index 7f34f1ae35..cf2dcb9d7f 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_decoders.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_decoders.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_backbones/test_encoder_decoders/test_deepfill_decoder.py b/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_decoder.py index 08367f73c0..26d6958b1c 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_decoder.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models.backbones import DeepFillDecoder diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encdec.py b/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encdec.py index 609b90e111..337b26ee66 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encdec.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encdec.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models.backbones import DeepFillEncoderDecoder, GLEncoderDecoder diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encoder.py b/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encoder.py index e11d3385fd..fd6dabf7b3 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encoder.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_deepfill_encoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models.backbones import ContextualAttentionNeck, DeepFillEncoder diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_encoder_decoder.py b/tests/test_models/test_backbones/test_encoder_decoders/test_encoder_decoder.py index 3d73d6e9ac..e34e6cd33a 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_encoder_decoder.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_encoder_decoder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_encoders.py b/tests/test_models/test_backbones/test_encoder_decoders/test_encoders.py index 9c82a429e5..285c5b254a 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_encoders.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_encoders.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from collections.abc import Iterable import numpy as np diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_gl_model.py b/tests/test_models/test_backbones/test_encoder_decoders/test_gl_model.py index 89f571bbb2..3cef0ce522 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_gl_model.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_gl_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_encoder_decoders/test_pconv_encdec.py b/tests/test_models/test_backbones/test_encoder_decoders/test_pconv_encdec.py index dd0924f4ec..c075d37d41 100644 --- a/tests/test_models/test_backbones/test_encoder_decoders/test_pconv_encdec.py +++ b/tests/test_models/test_backbones/test_encoder_decoders/test_pconv_encdec.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch from mmcv.utils.parrots_wrapper import _BatchNorm diff --git a/tests/test_models/test_backbones/test_generation_backbones/test_generators.py b/tests/test_models/test_backbones/test_generation_backbones/test_generators.py index 11382d4946..a6e406983d 100644 --- a/tests/test_models/test_backbones/test_generation_backbones/test_generators.py +++ b/tests/test_models/test_backbones/test_generation_backbones/test_generators.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import numpy as np diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_net.py b/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_net.py index 95458e55ec..ddb74930f0 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_net.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_plusplus.py b/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_plusplus.py index 5d7e8626f2..0c88543a09 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_plusplus.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_basicvsr_plusplus.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_dic_net.py b/tests/test_models/test_backbones/test_sr_backbones/test_dic_net.py index 4cdbdfaae2..a4aad791b0 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_dic_net.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_dic_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch import torch.nn as nn diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_duf.py b/tests/test_models/test_backbones/test_sr_backbones/test_duf.py index 06b2803070..5d85fe6920 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_duf.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_duf.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_edvr_net.py b/tests/test_models/test_backbones/test_sr_backbones/test_edvr_net.py index 9fc2194669..96e6e05448 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_edvr_net.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_edvr_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_glean_net.py b/tests/test_models/test_backbones/test_sr_backbones/test_glean_net.py index 2a75120e90..f0edcfd427 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_glean_net.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_glean_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_iconvsr.py b/tests/test_models/test_backbones/test_sr_backbones/test_iconvsr.py index b581c127aa..701a767292 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_iconvsr.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_iconvsr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_liif_net.py b/tests/test_models/test_backbones/test_sr_backbones/test_liif_net.py index f825c04dee..2101a635bf 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_liif_net.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_liif_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models import build_backbone diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_rdn.py b/tests/test_models/test_backbones/test_sr_backbones/test_rdn.py index ccf07fa9c0..487502413c 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_rdn.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_rdn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_sr_backbones.py b/tests/test_models/test_backbones/test_sr_backbones/test_sr_backbones.py index 31662b8894..6c7692ca95 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_sr_backbones.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_sr_backbones.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_backbones/test_sr_backbones/test_tdan_net.py b/tests/test_models/test_backbones/test_sr_backbones/test_tdan_net.py index de36a52c21..08462070b2 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_tdan_net.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_tdan_net.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_backbones/test_sr_backbones/test_tof.py b/tests/test_models/test_backbones/test_sr_backbones/test_tof.py index 5c767b2653..0d1f4bc7e0 100644 --- a/tests/test_models/test_backbones/test_sr_backbones/test_tof.py +++ b/tests/test_models/test_backbones/test_sr_backbones/test_tof.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_base_model.py b/tests/test_models/test_base_model.py index 6c7d1a9717..3d3febad41 100644 --- a/tests/test_models/test_base_model.py +++ b/tests/test_models/test_base_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import unittest from unittest.mock import patch diff --git a/tests/test_models/test_common/test_common_module.py b/tests/test_models/test_common/test_common_module.py index cd648407d0..e74d0a7194 100644 --- a/tests/test_models/test_common/test_common_module.py +++ b/tests/test_models/test_common/test_common_module.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch import torch.nn as nn diff --git a/tests/test_models/test_common/test_flow_warp.py b/tests/test_models/test_common/test_flow_warp.py index 36017b7ffe..aff69e5bf0 100644 --- a/tests/test_models/test_common/test_flow_warp.py +++ b/tests/test_models/test_common/test_flow_warp.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_common/test_img_normalize.py b/tests/test_models/test_common/test_img_normalize.py index d8a16f17e2..02921b5f56 100644 --- a/tests/test_models/test_common/test_img_normalize.py +++ b/tests/test_models/test_common/test_img_normalize.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models.common import ImgNormalize diff --git a/tests/test_models/test_common/test_model_utils.py b/tests/test_models/test_common/test_model_utils.py index 4a4c77cde5..2910ca5fb4 100644 --- a/tests/test_models/test_common/test_model_utils.py +++ b/tests/test_models/test_common/test_model_utils.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import numpy as np diff --git a/tests/test_models/test_components/test_discriminators/test_deepfill_disc.py b/tests/test_models/test_components/test_discriminators/test_deepfill_disc.py index 03b70ac6be..ff1c4dbf3c 100644 --- a/tests/test_models/test_components/test_discriminators/test_deepfill_disc.py +++ b/tests/test_models/test_components/test_discriminators/test_deepfill_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_components/test_discriminators/test_discriminators.py b/tests/test_models/test_components/test_discriminators/test_discriminators.py index 26deb1846f..55d3f3af6c 100644 --- a/tests/test_models/test_components/test_discriminators/test_discriminators.py +++ b/tests/test_models/test_components/test_discriminators/test_discriminators.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import numpy as np diff --git a/tests/test_models/test_components/test_discriminators/test_light_cnn.py b/tests/test_models/test_components/test_discriminators/test_light_cnn.py index 8ad69b4670..95a6ae3847 100644 --- a/tests/test_models/test_components/test_discriminators/test_light_cnn.py +++ b/tests/test_models/test_components/test_discriminators/test_light_cnn.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_components/test_discriminators/test_multi_layer_disc.py b/tests/test_models/test_components/test_discriminators/test_multi_layer_disc.py index c3933e9130..7f712e5ada 100644 --- a/tests/test_models/test_components/test_discriminators/test_multi_layer_disc.py +++ b/tests/test_models/test_components/test_discriminators/test_multi_layer_disc.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch import torch.nn as nn diff --git a/tests/test_models/test_components/test_refiners/test_deepfill_refiner.py b/tests/test_models/test_components/test_refiners/test_deepfill_refiner.py index f0920c5a5a..b3b2984a7e 100644 --- a/tests/test_models/test_components/test_refiners/test_deepfill_refiner.py +++ b/tests/test_models/test_components/test_refiners/test_deepfill_refiner.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models import (ContextualAttentionNeck, DeepFillDecoder, diff --git a/tests/test_models/test_components/test_refiners/test_matting_refiners.py b/tests/test_models/test_components/test_refiners/test_matting_refiners.py index 6a21293c60..1bd1c877a6 100644 --- a/tests/test_models/test_components/test_refiners/test_matting_refiners.py +++ b/tests/test_models/test_components/test_refiners/test_matting_refiners.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch diff --git a/tests/test_models/test_components/test_refiners/test_mlp_refiner.py b/tests/test_models/test_components/test_refiners/test_mlp_refiner.py index 5c93436fbc..15114ead41 100644 --- a/tests/test_models/test_components/test_refiners/test_mlp_refiner.py +++ b/tests/test_models/test_components/test_refiners/test_mlp_refiner.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn diff --git a/tests/test_models/test_components/test_stylegan2.py b/tests/test_models/test_components/test_stylegan2.py index 6edddcadb3..e7a42c0ede 100644 --- a/tests/test_models/test_components/test_stylegan2.py +++ b/tests/test_models/test_components/test_stylegan2.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from copy import deepcopy import pytest diff --git a/tests/test_models/test_extractors/test_feedback_hour_glass.py b/tests/test_models/test_extractors/test_feedback_hour_glass.py index 1c55153721..515f5c545d 100644 --- a/tests/test_models/test_extractors/test_feedback_hour_glass.py +++ b/tests/test_models/test_extractors/test_feedback_hour_glass.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_extractors/test_lte.py b/tests/test_models/test_extractors/test_lte.py index d3adc04fc4..1332265a2c 100644 --- a/tests/test_models/test_extractors/test_lte.py +++ b/tests/test_models/test_extractors/test_lte.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import pytest import torch diff --git a/tests/test_models/test_inpaintors/test_deepfill_inpaintor.py b/tests/test_models/test_inpaintors/test_deepfill_inpaintor.py index 1228dc289e..8836e87cfd 100644 --- a/tests/test_models/test_inpaintors/test_deepfill_inpaintor.py +++ b/tests/test_models/test_inpaintors/test_deepfill_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os import tempfile diff --git a/tests/test_models/test_inpaintors/test_gl_inpaintor.py b/tests/test_models/test_inpaintors/test_gl_inpaintor.py index 192fdcdf0b..47b63d8593 100644 --- a/tests/test_models/test_inpaintors/test_gl_inpaintor.py +++ b/tests/test_models/test_inpaintors/test_gl_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmcv import Config diff --git a/tests/test_models/test_inpaintors/test_one_stage_inpaintor.py b/tests/test_models/test_inpaintors/test_one_stage_inpaintor.py index 2acf864dd6..a1b942112d 100644 --- a/tests/test_models/test_inpaintors/test_one_stage_inpaintor.py +++ b/tests/test_models/test_inpaintors/test_one_stage_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os import tempfile diff --git a/tests/test_models/test_inpaintors/test_pconv_inpaintor.py b/tests/test_models/test_inpaintors/test_pconv_inpaintor.py index 8c3fb59e3b..e49d2585f5 100644 --- a/tests/test_models/test_inpaintors/test_pconv_inpaintor.py +++ b/tests/test_models/test_inpaintors/test_pconv_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import tempfile from unittest.mock import patch diff --git a/tests/test_models/test_inpaintors/test_two_stage_inpaintor.py b/tests/test_models/test_inpaintors/test_two_stage_inpaintor.py index 942679dbf6..202209ade1 100644 --- a/tests/test_models/test_inpaintors/test_two_stage_inpaintor.py +++ b/tests/test_models/test_inpaintors/test_two_stage_inpaintor.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy import os import tempfile diff --git a/tests/test_models/test_losses/test_feature_loss.py b/tests/test_models/test_losses/test_feature_loss.py index 7cbe8282df..ecb098df3a 100644 --- a/tests/test_models/test_losses/test_feature_loss.py +++ b/tests/test_models/test_losses/test_feature_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_losses.py b/tests/test_models/test_losses/test_losses.py index 2dddce7a8a..8e133ad54d 100644 --- a/tests/test_models/test_losses/test_losses.py +++ b/tests/test_models/test_losses/test_losses.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math from unittest.mock import patch diff --git a/tests/test_models/test_mattors/test_mattors.py b/tests/test_models/test_mattors/test_mattors.py index fbecd94c42..2ea1dff91e 100644 --- a/tests/test_models/test_mattors/test_mattors.py +++ b/tests/test_models/test_mattors/test_mattors.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp from unittest.mock import patch diff --git a/tests/test_models/test_restorers/test_basic_restorer.py b/tests/test_models/test_restorers/test_basic_restorer.py index dc158cf1b4..f50bf41516 100644 --- a/tests/test_models/test_restorers/test_basic_restorer.py +++ b/tests/test_models/test_restorers/test_basic_restorer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile import mmcv diff --git a/tests/test_models/test_restorers/test_basicvsr_model.py b/tests/test_models/test_restorers/test_basicvsr_model.py index 1d4cc8d0b4..f7b14433af 100644 --- a/tests/test_models/test_restorers/test_basicvsr_model.py +++ b/tests/test_models/test_restorers/test_basicvsr_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile import mmcv diff --git a/tests/test_models/test_restorers/test_dic_model.py b/tests/test_models/test_restorers/test_dic_model.py index 4070e40c33..1560273661 100644 --- a/tests/test_models/test_restorers/test_dic_model.py +++ b/tests/test_models/test_restorers/test_dic_model.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_restorers/test_edvr_model.py b/tests/test_models/test_restorers/test_edvr_model.py index 1573609228..13463a19a4 100644 --- a/tests/test_models/test_restorers/test_edvr_model.py +++ b/tests/test_models/test_restorers/test_edvr_model.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile import mmcv diff --git a/tests/test_models/test_restorers/test_esrgan.py b/tests/test_models/test_restorers/test_esrgan.py index 2e67271b21..66ea33a973 100644 --- a/tests/test_models/test_restorers/test_esrgan.py +++ b/tests/test_models/test_restorers/test_esrgan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. from unittest.mock import patch import torch diff --git a/tests/test_models/test_restorers/test_glean.py b/tests/test_models/test_restorers/test_glean.py index 19eadb42a2..f79c5fa215 100644 --- a/tests/test_models/test_restorers/test_glean.py +++ b/tests/test_models/test_restorers/test_glean.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import mmcv import pytest import torch diff --git a/tests/test_models/test_restorers/test_liif.py b/tests/test_models/test_restorers/test_liif.py index 741af48cff..cca423e6af 100644 --- a/tests/test_models/test_restorers/test_liif.py +++ b/tests/test_models/test_restorers/test_liif.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch import torch.nn as nn diff --git a/tests/test_models/test_restorers/test_srgan.py b/tests/test_models/test_restorers/test_srgan.py index 66598d1859..8f7a442f73 100644 --- a/tests/test_models/test_restorers/test_srgan.py +++ b/tests/test_models/test_restorers/test_srgan.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_restorers/test_tdan.py b/tests/test_models/test_restorers/test_tdan.py index 1a451b8c50..43ef4bcc6b 100644 --- a/tests/test_models/test_restorers/test_tdan.py +++ b/tests/test_models/test_restorers/test_tdan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import tempfile import mmcv diff --git a/tests/test_models/test_restorers/test_ttsr.py b/tests/test_models/test_restorers/test_ttsr.py index d0ad99051f..6b7a20d32d 100644 --- a/tests/test_models/test_restorers/test_ttsr.py +++ b/tests/test_models/test_restorers/test_ttsr.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import torch from mmcv.runner import obj_from_dict diff --git a/tests/test_models/test_synthesizers/test_cyclegan.py b/tests/test_models/test_synthesizers/test_cyclegan.py index 7b29c68838..e934b48be3 100644 --- a/tests/test_models/test_synthesizers/test_cyclegan.py +++ b/tests/test_models/test_synthesizers/test_cyclegan.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from unittest.mock import patch diff --git a/tests/test_models/test_transformer/test_search_transformer.py b/tests/test_models/test_transformer/test_search_transformer.py index 29d802acde..1c7a26bd8a 100644 --- a/tests/test_models/test_transformer/test_search_transformer.py +++ b/tests/test_models/test_transformer/test_search_transformer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch from mmedit.models.builder import build_component diff --git a/tests/test_runtime/test_dataset_builder.py b/tests/test_runtime/test_dataset_builder.py index ffe1aa6958..dce0ac267d 100644 --- a/tests/test_runtime/test_dataset_builder.py +++ b/tests/test_runtime/test_dataset_builder.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import math from torch.utils.data import ConcatDataset, RandomSampler, SequentialSampler diff --git a/tests/test_runtime/test_eval_hook.py b/tests/test_runtime/test_eval_hook.py index 67387b0b61..6c7dd6f0d0 100644 --- a/tests/test_runtime/test_eval_hook.py +++ b/tests/test_runtime/test_eval_hook.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import logging import tempfile from unittest.mock import MagicMock diff --git a/tests/test_runtime/test_optimizer.py b/tests/test_runtime/test_optimizer.py index cb396c8305..9b88cf6077 100644 --- a/tests/test_runtime/test_optimizer.py +++ b/tests/test_runtime/test_optimizer.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import torch import torch.nn as nn diff --git a/tests/test_runtime/test_visual_hook.py b/tests/test_runtime/test_visual_hook.py index 8de925f37e..22e6808ab5 100644 --- a/tests/test_runtime/test_visual_hook.py +++ b/tests/test_runtime/test_visual_hook.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os.path as osp import tempfile from unittest.mock import MagicMock diff --git a/tests/test_utils/test_mask_generation.py b/tests/test_utils/test_mask_generation.py index fba7cd0239..bb0182b11a 100644 --- a/tests/test_utils/test_mask_generation.py +++ b/tests/test_utils/test_mask_generation.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest diff --git a/tests/test_utils/test_onnx_wraper.py b/tests/test_utils/test_onnx_wraper.py index 29bce6cb0f..d0e4811052 100644 --- a/tests/test_utils/test_onnx_wraper.py +++ b/tests/test_utils/test_onnx_wraper.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import os import mmcv diff --git a/tests/test_utils/test_pix2pix.py b/tests/test_utils/test_pix2pix.py index 7e72328ea0..2b9fca2895 100644 --- a/tests/test_utils/test_pix2pix.py +++ b/tests/test_utils/test_pix2pix.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import copy from unittest.mock import patch diff --git a/tests/test_utils/test_tensor2img.py b/tests/test_utils/test_tensor2img.py index 3da46f9213..8698e6652a 100644 --- a/tests/test_utils/test_tensor2img.py +++ b/tests/test_utils/test_tensor2img.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import numpy as np import pytest import torch diff --git a/tools/data/matting/bgm/preprocess_bgm_dataset.py b/tools/data/matting/bgm/preprocess_bgm_dataset.py index d073d8fb6b..9604d9b314 100644 --- a/tools/data/matting/bgm/preprocess_bgm_dataset.py +++ b/tools/data/matting/bgm/preprocess_bgm_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/data/matting/comp1k/check_extended_fg.py b/tools/data/matting/comp1k/check_extended_fg.py index a9b0d91051..b0238a4868 100755 --- a/tools/data/matting/comp1k/check_extended_fg.py +++ b/tools/data/matting/comp1k/check_extended_fg.py @@ -1,4 +1,5 @@ #!/usr/bin/env python +# Copyright (c) OpenMMLab. All rights reserved. # This script checks the alpha-foreground difference between # the extended fg and the original fg diff --git a/tools/data/matting/comp1k/extend_fg.py b/tools/data/matting/comp1k/extend_fg.py index 6812394612..5e1c7d8919 100644 --- a/tools/data/matting/comp1k/extend_fg.py +++ b/tools/data/matting/comp1k/extend_fg.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/data/matting/comp1k/filter_comp1k_anno.py b/tools/data/matting/comp1k/filter_comp1k_anno.py index e10ea7127b..9a447b409c 100644 --- a/tools/data/matting/comp1k/filter_comp1k_anno.py +++ b/tools/data/matting/comp1k/filter_comp1k_anno.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os.path as osp diff --git a/tools/data/matting/comp1k/preprocess_comp1k_dataset.py b/tools/data/matting/comp1k/preprocess_comp1k_dataset.py index 625dcc85b6..0b2df609e5 100644 --- a/tools/data/matting/comp1k/preprocess_comp1k_dataset.py +++ b/tools/data/matting/comp1k/preprocess_comp1k_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import math import os.path as osp diff --git a/tools/data/super-resolution/div2k/preprocess_div2k_dataset.py b/tools/data/super-resolution/div2k/preprocess_div2k_dataset.py index 95e961e3a3..548ae4f709 100644 --- a/tools/data/super-resolution/div2k/preprocess_div2k_dataset.py +++ b/tools/data/super-resolution/div2k/preprocess_div2k_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os import os.path as osp diff --git a/tools/data/super-resolution/reds/preprocess_reds_dataset.py b/tools/data/super-resolution/reds/preprocess_reds_dataset.py index 5d59e45c64..ce3c81cacf 100644 --- a/tools/data/super-resolution/reds/preprocess_reds_dataset.py +++ b/tools/data/super-resolution/reds/preprocess_reds_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import glob import os diff --git a/tools/data/super-resolution/vimeo90k/preprocess_vimeo90k_dataset.py b/tools/data/super-resolution/vimeo90k/preprocess_vimeo90k_dataset.py index 29a49642f4..0dc5c505b7 100644 --- a/tools/data/super-resolution/vimeo90k/preprocess_vimeo90k_dataset.py +++ b/tools/data/super-resolution/vimeo90k/preprocess_vimeo90k_dataset.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os.path as osp import sys diff --git a/tools/deploy_test.py b/tools/deploy_test.py index 1b4c7cd9dc..ba4cff25d7 100644 --- a/tools/deploy_test.py +++ b/tools/deploy_test.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import warnings from typing import Any diff --git a/tools/evaluate_comp1k.py b/tools/evaluate_comp1k.py index bb7bce1aa9..aae1f08997 100644 --- a/tools/evaluate_comp1k.py +++ b/tools/evaluate_comp1k.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import os.path as osp import re diff --git a/tools/get_flops.py b/tools/get_flops.py index 6bcb15126b..3a5efb9c37 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/onnx2tensorrt.py b/tools/onnx2tensorrt.py index 2a3bd1bc9c..de4b0c6d98 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/publish_model.py b/tools/publish_model.py index 88625aaa95..aad82c7f91 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 33074c6027..6114e1babe 100644 --- a/tools/pytorch2onnx.py +++ b/tools/pytorch2onnx.py @@ -1,3 +1,4 @@ +# Copyright (c) OpenMMLab. All rights reserved. import argparse import cv2 diff --git a/tools/test.py b/tools/test.py index 3f55e89a36..ddc1ef0b12 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 b197f2f6e2..ce0c05bc33 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