Update vg_evaluator.py

vg_evaluator.py

@@ -347,8 +347,8 @@ def evaluation_for_challenge(gt_data,pred_data):
             batch_result["pred_scores"] = torch.zeros(0,9)
             batch_result["pred_bboxes"] =  torch.zeros(0,)
         else:
-            batch_result["pred_scores"] = torch.tensor(pred_data[sample_ID]["score"])
-            batch_result["pred_bboxes"] = torch.tensor(trun_box(pred_data[sample_ID]["pred_bboxes"]))
+            batch_result["pred_scores"] = torch.tensor(pred_data[sample_ID]["score"][:100])
+            batch_result["pred_bboxes"] = torch.tensor(trun_box(pred_data[sample_ID]["pred_bboxes"][:100]))
 
         batch_result["gt_bboxes"] = torch.tensor(gt_data[sample_ID])
         batch_result["subclass"] = sample_ID.split('__')[0]