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: