Add: Add initialization operation of networks.

chairc · May 6, 2024 · 3639b6e · 3639b6e
1 parent 6e46b54
commit 3639b6e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/model/networks/__init__.py b/model/networks/__init__.py
@@ -5,3 +5,11 @@
     @Author : chairc
     @Site   : https://github.com/chairc
 """
+# Diffusion model network
+from .base import BaseNet
+from .cspdarkunet import CSPDarkUnet
+from .unet import UNet
+from .unetv2 import UNetV2
+
+# Super resolution network
+from .sr.srv1 import SRv1