From b31db9fd888773b6c6e949cefdff3c854bee45d0 Mon Sep 17 00:00:00 2001 From: gexy5 Date: Tue, 14 Jun 2022 15:04:46 +0800 Subject: [PATCH] add omnivore Signed-off-by: gexy5 --- omnivore.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/omnivore.py b/omnivore.py index 167524a..bfd5eac 100644 --- a/omnivore.py +++ b/omnivore.py @@ -99,16 +99,7 @@ class Omnivore(NNOperator): ) inputs = data.to(self.device)[None, ...] - feats = self.model.forward_features(inputs) - if self.model.reshape: - if self.model.is_shift and self.model.temporal_pool: - base_out = feats.view((-1, self.model.num_segments // 2) + feats.size()[1:]) - else: - base_out = feats.view((-1, self.model.num_segments) + feats.size()[1:]) - output = self.model.consensus(base_out) - features = output.to('cpu').squeeze(0).detach().numpy() - - outs = self.model.head(feats) + outs = self.model.head(inputs) post_act = torch.nn.Softmax(dim=1) preds = post_act(outs) pred_scores, pred_classes = preds.topk(k=self.topk)