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-08-27 22:03:13 -07:00 committed by GitHub
parent c5da300d44
commit e7c38f4902
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -430,10 +430,6 @@ class ISNetGTEncoder(nn.Module):
self.side5 = nn.Conv2d(512,out_ch,3,padding=1)
self.side6 = nn.Conv2d(512,out_ch,3,padding=1)
def compute_loss_max(self, preds, targets, fs):
return muti_loss_fusion_max(preds, targets,fs)
def compute_loss(self, preds, targets):
return muti_loss_fusion(preds,targets)