Update demo.py

demo.py

@@ -378,7 +378,7 @@ class VideoMIRModel(VideoModel):
         all_outputs = self.forward_video(text_features, t2v=True)
         preds = all_outputs.numpy()
         relevancy = self.relevancy_mat_t2v[idx]
-        sel = 3
+        sel = 2
         pred_video = self.video_samples[(-preds[0]).argsort()[:sel]]
         gt_video = np.where(relevancy == 1)[0]
         return pred_video, gt_video
@@ -446,14 +446,14 @@ def main():
         lavila = VideoCLSModel(f"configs/{args.dataset}/zeroshot.yml")
         egovpa = VideoCLSModel(f"configs/{args.dataset}/egovpa.yml")
     elif args.dataset == 'ek100_mir':
-        #lavila = VideoMIRModel(f"configs/{args.dataset}/zeroshot.yml")
+        lavila = VideoMIRModel(f"configs/{args.dataset}/zeroshot.yml")
         egovpa = VideoMIRModel(f"configs/{args.dataset}/egovpa.yml")
     else:
         raise NotImplementedError
 
-    #lavila.evaluate()
-    #egovpa.evaluate()
-    egovpa.predict_t2v(idx=0, sid=2119)
+    lavila.evaluate()
+    egovpa.evaluate()
+    #egovpa.predict_t2v(idx=0, sid=2119)
 
 
 if __name__ == '__main__':