linfan vor 1 Jahr
Ursprung
Commit
06680d1e7f
1 geänderte Dateien mit 1 neuen und 0 gelöschten Zeilen
  1. 1 0
      video_rank.py

+ 1 - 0
video_rank.py

@@ -663,6 +663,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, flowConfig=None):
             all_recall_rank = rov_recall_rank + flow_recall_rank
             if flow_num > 0:
                 rank_result.extend(all_recall_rank[:size - top_K])
+                return  rank_result, flow_num
             else:
                 i = 0
                 while i < size - top_K: