modify features

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
3 years ago · c6ec91eee7
3 changed files with 2 additions and 1 deletions
--- a/movinet.py
+++ b/movinet.py
@ -106,7 +106,8 @@ class Movinet(NNOperator):

        self.model.clean_activation_buffers()
        feats = self.model.forward_features(inputs)
-        features = feats.to('cpu').squeeze(0).detach().numpy()
+        features = feats.flatten(1)
+        features = features.to('cpu').squeeze(0).detach().numpy()

        outs = self.model.head(feats)
        post_act = torch.nn.Softmax(dim=1)
--- a/result1.png
+++ b/result1.png
--- a/result2.png
+++ b/result2.png