add omnivore

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
3 years ago · 3f8e118fc8
1 changed files with 1 additions and 1 deletions
--- a/omnivore.py
+++ b/omnivore.py
@ -101,7 +101,7 @@ class Omnivore(NNOperator):
                    )
        inputs = data.to(self.device)[None, ...]

-        feats = self.model.forward_features(inputs, input_type = self.input_type)
+        feats = self.model.forward_features(inputs)
        features = feats.to('cpu').squeeze(0).detach().numpy()

        outs = self.model.head(feats, input_type = self.input_type)