codeformer

2136e796 · mashun1 · 2136e796 · 2136e796 · 2136e796 · 2136e796
Commit 2136e796 authored Jan 08, 2024 by mashun1
20 changed files
--- a/basicsr/metrics/metric_util.py
+++ b/basicsr/metrics/metric_util.py
--- a/basicsr/metrics/psnr_ssim.py
+++ b/basicsr/metrics/psnr_ssim.py
--- a/basicsr/models/__init__.py
+++ b/basicsr/models/__init__.py
--- a/basicsr/models/base_model.py
+++ b/basicsr/models/base_model.py
--- a/basicsr/models/codeformer_idx_model.py
+++ b/basicsr/models/codeformer_idx_model.py
--- a/basicsr/models/codeformer_joint_model.py
+++ b/basicsr/models/codeformer_joint_model.py
--- a/basicsr/models/codeformer_model.py
+++ b/basicsr/models/codeformer_model.py
--- a/basicsr/models/lr_scheduler.py
+++ b/basicsr/models/lr_scheduler.py
--- a/basicsr/models/sr_model.py
+++ b/basicsr/models/sr_model.py
--- a/basicsr/models/vqgan_model.py
+++ b/basicsr/models/vqgan_model.py
--- a/basicsr/ops/__init__.py
+++ b/basicsr/ops/__init__.py
--- a/basicsr/ops/dcn/__init__.py
+++ b/basicsr/ops/dcn/__init__.py
--- a/basicsr/ops/dcn/deform_conv.py
+++ b/basicsr/ops/dcn/deform_conv.py
--- a/basicsr/ops/dcn/src/deform_conv_cuda.cpp
+++ b/basicsr/ops/dcn/src/deform_conv_cuda.cpp
--- a/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu
+++ b/basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu
--- a/basicsr/ops/dcn/src/deform_conv_ext.cpp
+++ b/basicsr/ops/dcn/src/deform_conv_ext.cpp
--- a/basicsr/ops/fused_act/__init__.py
+++ b/basicsr/ops/fused_act/__init__.py
+from .fused_act import FusedLeakyReLU, fused_leaky_relu
+
+__all__ = ['FusedLeakyReLU', 'fused_leaky_relu']
--- a/basicsr/ops/fused_act/fused_act.py
+++ b/basicsr/ops/fused_act/fused_act.py
--- a/basicsr/ops/fused_act/src/fused_bias_act.cpp
+++ b/basicsr/ops/fused_act/src/fused_bias_act.cpp
--- a/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu
+++ b/basicsr/ops/fused_act/src/fused_bias_act_kernel.cu