Clone/DIS
1
0
Fork 0
mirror of https://github.com/xuebinqin/DIS.git synced 2024-11-26 08:43:17 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update isnet.py

Browse Source
This commit is contained in:
Xuebin Qin 2022-11-17 20:57:13 -08:00 committed by GitHub
parent 10657f42e5
commit 996c9495d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IS-Net/models/isnet.py
View File

@ -544,7 +544,7 @@ class ISNetDIS(nn.Module):
hx = x hx = x
hxin = self.conv_in(hx) hxin = self.conv_in(hx)
hx = self.pool_in(hxin) #hx = self.pool_in(hxin)
#stage 1 #stage 1
hx1 = self.stage1(hxin) hx1 = self.stage1(hxin)