gina9726 commited on
Commit
d32d7bb
·
verified ·
1 Parent(s): d598fd0

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -378,7 +378,7 @@ class VideoMIRModel(VideoModel):
378
  all_outputs = self.forward_video(text_features, t2v=True)
379
  preds = all_outputs.numpy()
380
  relevancy = self.relevancy_mat_t2v[idx]
381
- sel = 2
382
  pred_video = self.video_samples[(-preds[0]).argsort()[:sel]]
383
  gt_video = np.where(relevancy == 1)[0]
384
  return pred_video, gt_video
 
378
  all_outputs = self.forward_video(text_features, t2v=True)
379
  preds = all_outputs.numpy()
380
  relevancy = self.relevancy_mat_t2v[idx]
381
+ sel = 1
382
  pred_video = self.video_samples[(-preds[0]).argsort()[:sel]]
383
  gt_video = np.where(relevancy == 1)[0]
384
  return pred_video, gt_video