linfan 1 年之前
父節點
當前提交
06680d1e7f
共有 1 個文件被更改,包括 1 次插入0 次删除
  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: