Update demo.py
Browse files
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 =
|
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
|