From 55ad855f26a8d1000a50d7950c49077c5ef1e7b7 Mon Sep 17 00:00:00 2001
From: gexy5 <xinyu.ge@zilliz.com>
Date: Tue, 14 Jun 2022 15:08:48 +0800
Subject: [PATCH] add omnivore

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
---
 omnivore.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/omnivore.py b/omnivore.py
index 1d2ab94..584c919 100644
--- a/omnivore.py
+++ b/omnivore.py
@@ -99,9 +99,9 @@ class Omnivore(NNOperator):
                     )
         inputs = data.to(self.device)[None, ...]
 
-        outs = self.model(inputs)
+        outs = self.model(inputs,input_type="video")
         post_act = torch.nn.Softmax(dim=1)
-        preds = post_act(outs,input_type="video")
+        preds = post_act(outs)
         pred_scores, pred_classes = preds.topk(k=self.topk)
         labels = [self.classmap[int(i)] for i in pred_classes[0]]
         scores = [round(float(x), 5) for x in pred_scores[0]]