fix img_metas bug (#152)

2021-01-26 11:24:08 +08:00 · 2021-01-26 11:24:08 +08:00 · a225cb6bdb
parent 3e5a9513be
commit a225cb6bdb
1 changed files with 1 additions and 1 deletions
--- a/mmcls/models/classifiers/image.py
+++ b/mmcls/models/classifiers/image.py
@ -62,7 +62,7 @@ class ImageClassifier(BaseClassifier):

        return losses

-    def simple_test(self, img):
+    def simple_test(self, img, img_metas):
        """Test without augmentation."""
        x = self.extract_feat(img)
        return self.head.simple_test(x)