Update total/Model.py
Browse files- total/Model.py +0 -1
total/Model.py
CHANGED
|
@@ -74,5 +74,4 @@ class Combine(Reduction):
|
|
| 74 |
bg = (1.0 - sum_fg).clamp(min=1e-6)
|
| 75 |
|
| 76 |
probs = torch.cat([bg, fg_all], dim=1)
|
| 77 |
-
print(probs.shape)
|
| 78 |
return probs / probs.sum(dim=1, keepdim=True)
|
|
|
|
| 74 |
bg = (1.0 - sum_fg).clamp(min=1e-6)
|
| 75 |
|
| 76 |
probs = torch.cat([bg, fg_all], dim=1)
|
|
|
|
| 77 |
return probs / probs.sum(dim=1, keepdim=True)
|