refactor evaluation and add cross view gallery

2022-11-24 22:07:50 +08:00
parent c5b4fb0806
commit 793405ad7c
18 changed files with 388 additions and 309 deletions
@@ -203,15 +203,3 @@ def get_ddp_module(module, **kwargs):
 def params_count(net):
    n_parameters = sum(p.numel() for p in net.parameters())
    return 'Parameters Count: {:.5f}M'.format(n_parameters / 1e6)
-
-
-def MeanIOU(msk1, msk2, eps=1.0e-9):
-    if not is_tensor(msk1):
-        msk1 = torch.from_numpy(msk1).cuda()
-    if not is_tensor(msk2):
-        msk2 = torch.from_numpy(msk2).cuda()
-    n = msk1.size(0)
-    inter = msk1 * msk2
-    union = ((msk1 + msk2) > 0.).float()
-    MeIOU = inter.view(n, -1).sum(-1) / (union.view(n, -1).sum(-1) + eps)
-    return MeIOU