Upload u2net.py

model/u2net.py

@@ -1,4 +1,25 @@
 
-# Здесь должна быть модель U2NET
-# Предполагается, что файл model/u2net.py уже содержит реализацию U2NET
-# Файл будет скопирован позже
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+
+class REBNCONV(nn.Module):
+    def __init__(self, in_ch=3, out_ch=3, dirate=1):
+        super(REBNCONV, self).__init__()
+        self.conv_s1 = nn.Conv2d(in_ch, out_ch, 3, padding=1*dirate, dilation=1*dirate)
+        self.bn_s1 = nn.BatchNorm2d(out_ch)
+        self.relu_s1 = nn.ReLU(inplace=True)
+
+    def forward(self, x):
+        return self.relu_s1(self.bn_s1(self.conv_s1(x)))
+
+# Простой U2NET-заглушка
+class U2NET(nn.Module):
+    def __init__(self, in_ch=3, out_ch=1):
+        super(U2NET, self).__init__()
+        self.rebnconv = REBNCONV(in_ch, out_ch)
+
+    def forward(self, x):
+        return self.rebnconv(x)
+
+__all__ = ["U2NET"]