linfan 2 年 前
コミット
0d6bbddfa1
1 ファイル変更3 行追加3 行削除
  1. 3 3
      video_rank.py

+ 3 - 3
video_rank.py

@@ -689,7 +689,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, flowConfig=None):
                     rank_result.extend(flow_recall_rank[:size - top_K - i])
                     return rank_result[:size]
             i += 1
-    return rank_result[:size]
+    return rank_result[:size], flow_result
 
 def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None, flowConfig=None):
     """
@@ -797,8 +797,8 @@ def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None
     if flowConfig=="1" and len(rov_recall_rank)>0:
         for recall_item in rank_result:
             flow_recall_name = recall_item.get("flowPool", '')
-            print("recall_item:", recall_item)
-            print("flow_recall_name:", flow_recall_name)
+            #print("recall_item:", recall_item)
+            #print("flow_recall_name:", flow_recall_name)
             if flow_recall_name is not None and flow_recall_name.find("#")>-1:
                 flow_result = flow_result+1
         if flow_result>0: