Update demo.py

demo.py

@@ -239,8 +239,8 @@ class VideoCLSModel(VideoModel):
     def predict(self, idx=0):
         all_outputs, all_targets = self.forward(idx)
         preds, targets = all_outputs.numpy(), all_targets.numpy()
-        #sel = np.where(np.cumsum(sorted(preds[0].tolist(), reverse=True)) > 0.06)[0][0]
-        sel = 5
+        sel = np.where(np.cumsum(sorted(preds[0].tolist(), reverse=True)) > 0.055)[0][0]
+        #sel = 5
         df = pd.DataFrame(self.labels)
         pred_action = df.iloc[preds[0].argsort()[-sel:]].values.tolist()
         gt_action = df.iloc[np.where(targets[0])[0]].values.tolist()