From 3639b6ed7645ab5da135816e2cca219ce371a043 Mon Sep 17 00:00:00 2001
From: ChairC <974833488@qq.com>
Date: Mon, 6 May 2024 11:07:59 +0800
Subject: [PATCH] Add: Add initialization operation of networks.

---
 model/networks/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/model/networks/__init__.py b/model/networks/__init__.py
index 9278aee..c6e6a30 100644
--- 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