diff --git a/mmtrack/models/reid/base_reid.py b/mmtrack/models/reid/base_reid.py index 90a3603b1..f180631c2 100644 --- a/mmtrack/models/reid/base_reid.py +++ b/mmtrack/models/reid/base_reid.py @@ -18,7 +18,7 @@ def forward_train(self, img, gt_label, **kwargs): # change the shape of label tensor from NxS to NS gt_label = gt_label.view(-1) x = self.extract_feat(img) - head_outputs = self.head.forward_train(x) + head_outputs = self.head.forward_train(x[0]) losses = dict() reid_loss = self.head.loss(gt_label, *head_outputs) @@ -30,7 +30,7 @@ def simple_test(self, img, **kwargs): """Test without augmentation.""" if img.nelement() > 0: x = self.extract_feat(img) - head_outputs = self.head.forward_train(x) + head_outputs = self.head.forward_train(x[0]) feats = head_outputs[0] return feats else: diff --git a/requirements/runtime.txt b/requirements/runtime.txt index 8125a5ae0..5c0857542 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -1,6 +1,6 @@ dotty_dict matplotlib -mmcls>=0.14.0 +mmcls>=0.16.0 motmetrics packaging pycocotools