diff --git a/IS-Net/models/isnet.py b/IS-Net/models/isnet.py index 182a8d4..8be74bb 100644 --- a/IS-Net/models/isnet.py +++ b/IS-Net/models/isnet.py @@ -544,7 +544,7 @@ class ISNetDIS(nn.Module): hx = x hxin = self.conv_in(hx) - hx = self.pool_in(hxin) + #hx = self.pool_in(hxin) #stage 1 hx1 = self.stage1(hxin)