refactor evaluation and add cross view gallery

opengait/modeling/base_model.py

@@ -30,7 +30,7 @@ from data.dataset import DataSet
 import data.sampler as Samplers
 from utils import Odict, mkdir, ddp_all_gather
 from utils import get_valid_args, is_list, is_dict, np2var, ts2np, list2var, get_attr_from
-from utils import evaluation as eval_functions
+from evaluation import evaluator as eval_functions
 from utils import NoOp
 from utils import get_msg_mgr
 

opengait/modeling/losses/bce.py

@@ -1,6 +1,6 @@
 import torch
 from .base import BaseLoss
-from utils import MeanIOU
+from evaluation import mean_iou
 
 
 class BinaryCrossEntropyLoss(BaseLoss):
@@ -24,7 +24,7 @@ class BinaryCrossEntropyLoss(BaseLoss):
         loss = loss.view(n, -1)
         mean_loss = loss.mean()
         hard_loss = loss.max()
-        miou = MeanIOU((logits > 0.5).float(), labels)
+        miou = mean_iou((logits > 0.5).float(), labels)
         self.info.update({
             'loss': mean_loss.detach().clone(),
             'hard_loss': hard_loss.detach().clone(),