├── .gitignore ├── LICENSE ├── README.md ├── configs ├── BRVE_LLRVD.py └── BRVE_SMOID.py ├── data_list ├── LLRVD_test.txt ├── LLRVD_train.txt ├── LLRVD_val.txt ├── test_list_DarkRaw2BrightRaw.txt ├── train_list_DarkRaw2BrightRaw.txt └── val_list_DarkRaw2BrightRaw.txt ├── figures ├── LLRVD.gif ├── SMOID.gif └── framework.png ├── mmedit ├── .mim │ ├── configs │ └── tools ├── __init__.py ├── apis │ ├── __init__.py │ ├── generation_inference.py │ ├── inpainting_inference.py │ ├── matting_inference.py │ ├── restoration_face_inference.py │ ├── restoration_inference.py │ ├── restoration_video_inference.py │ ├── test.py │ ├── train.py │ └── video_interpolation_inference.py ├── core │ ├── __init__.py │ ├── distributed_wrapper.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_hooks.py │ │ ├── metric_utils.py │ │ ├── metrics.py │ │ └── niqe_pris_params.npz │ ├── export │ │ ├── __init__.py │ │ └── wrappers.py │ ├── hooks │ │ ├── __init__.py │ │ ├── ema.py │ │ └── visualization.py │ ├── mask.py │ ├── misc.py │ ├── optimizer │ │ ├── __init__.py │ │ └── builder.py │ ├── scheduler │ │ ├── __init__.py │ │ └── lr_updater.py │ └── utils │ │ ├── __init__.py │ │ └── dist_utils.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── base_sr_dataset.py │ ├── builder.py │ ├── dataset_wrappers.py │ ├── ll_smoid_dataset.py │ ├── llrvd_dataset.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── augmentation.py │ │ ├── blur_kernels.py │ │ ├── compose.py │ │ ├── crop.py │ │ ├── formating.py │ │ ├── generate_assistant.py │ │ ├── llrvd_pipeline.py │ │ ├── loading.py │ │ ├── matlab_like_resize.py │ │ ├── matting_aug.py │ │ ├── normalization.py │ │ ├── random_degradations.py │ │ ├── random_down_sampling.py │ │ ├── smoidnpy_pipeline.py │ │ └── utils.py │ ├── registry.py │ └── samplers │ │ ├── __init__.py │ │ └── distributed_sampler.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── DABC.py │ │ ├── __init__.py │ │ └── bnnvd.py │ ├── base.py │ ├── builder.py │ ├── common │ │ ├── __init__.py │ │ ├── aspp.py │ │ ├── contextual_attention.py │ │ ├── conv.py │ │ ├── downsample.py │ │ ├── ensemble.py │ │ ├── flow_warp.py │ │ ├── gated_conv_module.py │ │ ├── gca_module.py │ │ ├── generation_model_utils.py │ │ ├── img_normalize.py │ │ ├── linear_module.py │ │ ├── mask_conv_module.py │ │ ├── model_utils.py │ │ ├── partial_conv.py │ │ ├── separable_conv_module.py │ │ ├── sr_backbone_utils.py │ │ └── upsample.py │ ├── losses │ │ ├── __init__.py │ │ ├── gan_loss.py │ │ ├── perceptual_loss.py │ │ ├── pixelwise_loss.py │ │ └── utils.py │ ├── registry.py │ └── restorers │ │ ├── __init__.py │ │ ├── basic_restorer.py │ │ └── brve.py ├── utils │ ├── __init__.py │ ├── bnn_flops_counter.py │ ├── cli.py │ ├── collect_env.py │ ├── flops_counter.py │ ├── isp.py │ ├── llrvd_isp.py │ ├── logger.py │ ├── padding.py │ └── setup_env.py └── version.py ├── requirements.txt ├── requirements ├── brve.txt ├── docs.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── setup.cfg ├── setup.py └── tools ├── demo.py ├── deploy_test.py ├── deployment ├── mmedit2torchserve.py ├── mmedit_handler.py └── test_torchserver.py ├── dist_test.sh ├── dist_train.sh ├── evaluate_comp1k.py ├── get_flops.py ├── get_flops_bnn.py ├── onnx2tensorrt.py ├── publish_model.py ├── pytorch2onnx.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/README.md -------------------------------------------------------------------------------- /configs/BRVE_LLRVD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/configs/BRVE_LLRVD.py -------------------------------------------------------------------------------- /configs/BRVE_SMOID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/configs/BRVE_SMOID.py -------------------------------------------------------------------------------- /data_list/LLRVD_test.txt: -------------------------------------------------------------------------------- 1 | 35 2 | 42 3 | 49 4 | 56 5 | 63 6 | 70 -------------------------------------------------------------------------------- /data_list/LLRVD_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/data_list/LLRVD_train.txt -------------------------------------------------------------------------------- /data_list/LLRVD_val.txt: -------------------------------------------------------------------------------- 1 | 7 2 | 14 3 | 21 4 | 28 -------------------------------------------------------------------------------- /data_list/test_list_DarkRaw2BrightRaw.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/data_list/test_list_DarkRaw2BrightRaw.txt -------------------------------------------------------------------------------- /data_list/train_list_DarkRaw2BrightRaw.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/data_list/train_list_DarkRaw2BrightRaw.txt -------------------------------------------------------------------------------- /data_list/val_list_DarkRaw2BrightRaw.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/data_list/val_list_DarkRaw2BrightRaw.txt -------------------------------------------------------------------------------- /figures/LLRVD.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/figures/LLRVD.gif -------------------------------------------------------------------------------- /figures/SMOID.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/figures/SMOID.gif -------------------------------------------------------------------------------- /figures/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/figures/framework.png -------------------------------------------------------------------------------- /mmedit/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /mmedit/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /mmedit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/__init__.py -------------------------------------------------------------------------------- /mmedit/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/__init__.py -------------------------------------------------------------------------------- /mmedit/apis/generation_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/generation_inference.py -------------------------------------------------------------------------------- /mmedit/apis/inpainting_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/inpainting_inference.py -------------------------------------------------------------------------------- /mmedit/apis/matting_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/matting_inference.py -------------------------------------------------------------------------------- /mmedit/apis/restoration_face_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/restoration_face_inference.py -------------------------------------------------------------------------------- /mmedit/apis/restoration_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/restoration_inference.py -------------------------------------------------------------------------------- /mmedit/apis/restoration_video_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/restoration_video_inference.py -------------------------------------------------------------------------------- /mmedit/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/test.py -------------------------------------------------------------------------------- /mmedit/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/train.py -------------------------------------------------------------------------------- /mmedit/apis/video_interpolation_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/apis/video_interpolation_inference.py -------------------------------------------------------------------------------- /mmedit/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/__init__.py -------------------------------------------------------------------------------- /mmedit/core/distributed_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/distributed_wrapper.py -------------------------------------------------------------------------------- /mmedit/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmedit/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /mmedit/core/evaluation/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/evaluation/metric_utils.py -------------------------------------------------------------------------------- /mmedit/core/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/evaluation/metrics.py -------------------------------------------------------------------------------- /mmedit/core/evaluation/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/evaluation/niqe_pris_params.npz -------------------------------------------------------------------------------- /mmedit/core/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/export/__init__.py -------------------------------------------------------------------------------- /mmedit/core/export/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/export/wrappers.py -------------------------------------------------------------------------------- /mmedit/core/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/hooks/__init__.py -------------------------------------------------------------------------------- /mmedit/core/hooks/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/hooks/ema.py -------------------------------------------------------------------------------- /mmedit/core/hooks/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/hooks/visualization.py -------------------------------------------------------------------------------- /mmedit/core/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/mask.py -------------------------------------------------------------------------------- /mmedit/core/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/misc.py -------------------------------------------------------------------------------- /mmedit/core/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/optimizer/__init__.py -------------------------------------------------------------------------------- /mmedit/core/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/optimizer/builder.py -------------------------------------------------------------------------------- /mmedit/core/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/scheduler/__init__.py -------------------------------------------------------------------------------- /mmedit/core/scheduler/lr_updater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/scheduler/lr_updater.py -------------------------------------------------------------------------------- /mmedit/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/utils/__init__.py -------------------------------------------------------------------------------- /mmedit/core/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/core/utils/dist_utils.py -------------------------------------------------------------------------------- /mmedit/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/__init__.py -------------------------------------------------------------------------------- /mmedit/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/base_dataset.py -------------------------------------------------------------------------------- /mmedit/datasets/base_sr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/base_sr_dataset.py -------------------------------------------------------------------------------- /mmedit/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/builder.py -------------------------------------------------------------------------------- /mmedit/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mmedit/datasets/ll_smoid_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/ll_smoid_dataset.py -------------------------------------------------------------------------------- /mmedit/datasets/llrvd_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/llrvd_dataset.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/augmentation.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/blur_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/blur_kernels.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/crop.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/generate_assistant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/generate_assistant.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/llrvd_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/llrvd_pipeline.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/matlab_like_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/matlab_like_resize.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/matting_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/matting_aug.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/normalization.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/random_degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/random_degradations.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/random_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/random_down_sampling.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/smoidnpy_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/smoidnpy_pipeline.py -------------------------------------------------------------------------------- /mmedit/datasets/pipelines/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/pipelines/utils.py -------------------------------------------------------------------------------- /mmedit/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/registry.py -------------------------------------------------------------------------------- /mmedit/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /mmedit/datasets/samplers/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/datasets/samplers/distributed_sampler.py -------------------------------------------------------------------------------- /mmedit/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/__init__.py -------------------------------------------------------------------------------- /mmedit/models/backbones/DABC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/backbones/DABC.py -------------------------------------------------------------------------------- /mmedit/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmedit/models/backbones/bnnvd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/backbones/bnnvd.py -------------------------------------------------------------------------------- /mmedit/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/base.py -------------------------------------------------------------------------------- /mmedit/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/builder.py -------------------------------------------------------------------------------- /mmedit/models/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/__init__.py -------------------------------------------------------------------------------- /mmedit/models/common/aspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/aspp.py -------------------------------------------------------------------------------- /mmedit/models/common/contextual_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/contextual_attention.py -------------------------------------------------------------------------------- /mmedit/models/common/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/conv.py -------------------------------------------------------------------------------- /mmedit/models/common/downsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/downsample.py -------------------------------------------------------------------------------- /mmedit/models/common/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/ensemble.py -------------------------------------------------------------------------------- /mmedit/models/common/flow_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/flow_warp.py -------------------------------------------------------------------------------- /mmedit/models/common/gated_conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/gated_conv_module.py -------------------------------------------------------------------------------- /mmedit/models/common/gca_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/gca_module.py -------------------------------------------------------------------------------- /mmedit/models/common/generation_model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/generation_model_utils.py -------------------------------------------------------------------------------- /mmedit/models/common/img_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/img_normalize.py -------------------------------------------------------------------------------- /mmedit/models/common/linear_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/linear_module.py -------------------------------------------------------------------------------- /mmedit/models/common/mask_conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/mask_conv_module.py -------------------------------------------------------------------------------- /mmedit/models/common/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/model_utils.py -------------------------------------------------------------------------------- /mmedit/models/common/partial_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/partial_conv.py -------------------------------------------------------------------------------- /mmedit/models/common/separable_conv_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/separable_conv_module.py -------------------------------------------------------------------------------- /mmedit/models/common/sr_backbone_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/sr_backbone_utils.py -------------------------------------------------------------------------------- /mmedit/models/common/upsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/common/upsample.py -------------------------------------------------------------------------------- /mmedit/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/losses/__init__.py -------------------------------------------------------------------------------- /mmedit/models/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/losses/gan_loss.py -------------------------------------------------------------------------------- /mmedit/models/losses/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/losses/perceptual_loss.py -------------------------------------------------------------------------------- /mmedit/models/losses/pixelwise_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/losses/pixelwise_loss.py -------------------------------------------------------------------------------- /mmedit/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/losses/utils.py -------------------------------------------------------------------------------- /mmedit/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/registry.py -------------------------------------------------------------------------------- /mmedit/models/restorers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/restorers/__init__.py -------------------------------------------------------------------------------- /mmedit/models/restorers/basic_restorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/restorers/basic_restorer.py -------------------------------------------------------------------------------- /mmedit/models/restorers/brve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/models/restorers/brve.py -------------------------------------------------------------------------------- /mmedit/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/__init__.py -------------------------------------------------------------------------------- /mmedit/utils/bnn_flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/bnn_flops_counter.py -------------------------------------------------------------------------------- /mmedit/utils/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/cli.py -------------------------------------------------------------------------------- /mmedit/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/collect_env.py -------------------------------------------------------------------------------- /mmedit/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/flops_counter.py -------------------------------------------------------------------------------- /mmedit/utils/isp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/isp.py -------------------------------------------------------------------------------- /mmedit/utils/llrvd_isp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/llrvd_isp.py -------------------------------------------------------------------------------- /mmedit/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/logger.py -------------------------------------------------------------------------------- /mmedit/utils/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/padding.py -------------------------------------------------------------------------------- /mmedit/utils/setup_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/utils/setup_env.py -------------------------------------------------------------------------------- /mmedit/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/mmedit/version.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/brve.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/requirements/brve.txt -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/requirements/runtime.txt -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/setup.py -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/deploy_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/deploy_test.py -------------------------------------------------------------------------------- /tools/deployment/mmedit2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/deployment/mmedit2torchserve.py -------------------------------------------------------------------------------- /tools/deployment/mmedit_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/deployment/mmedit_handler.py -------------------------------------------------------------------------------- /tools/deployment/test_torchserver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/deployment/test_torchserver.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/evaluate_comp1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/evaluate_comp1k.py -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/get_flops_bnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/get_flops_bnn.py -------------------------------------------------------------------------------- /tools/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/onnx2tensorrt.py -------------------------------------------------------------------------------- /tools/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/publish_model.py -------------------------------------------------------------------------------- /tools/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/pytorch2onnx.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanggengchen/BRVE/HEAD/tools/train.py --------------------------------------------------------------------------------