feat: support HAT (#29)

TensoRaws · Dec 12, 2024 · 731bade · 731bade
1 parent 0bc0a93
commit 731bade
Show file tree

Hide file tree

Showing 10 changed files with 1,547 additions and 0 deletions.
diff --git a/ccrestoration/arch/__init__.py b/ccrestoration/arch/__init__.py
@@ -15,3 +15,4 @@
 from ccrestoration.arch.scunet_arch import SCUNet  # noqa
 from ccrestoration.arch.dat_arch import DAT  # noqa
 from ccrestoration.arch.srcnn_arch import SRCNN  # noqa
+from ccrestoration.arch.hat_arch import HAT  # noqa