adapat to mmcls >= 0.16.0 (#281)

open-mmlab · Oct 1, 2021 · ae53c70 · ae53c70
1 parent b66ef6f
commit ae53c70
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -1,6 +1,6 @@
 dotty_dict
 matplotlib
-mmcls>=0.14.0
+mmcls>=0.16.0
 motmetrics
 packaging
 pycocotools