From 996c9495d40a82f850f1f658b095b108f590a434 Mon Sep 17 00:00:00 2001 From: Xuebin Qin Date: Thu, 17 Nov 2022 20:57:13 -0800 Subject: [PATCH] Update isnet.py --- IS-Net/models/isnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)