linfan 1 năm trước cách đây
mục cha
commit
06680d1e7f
1 tập tin đã thay đổi với 1 bổ sung0 xóa
  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: