Merge pull request #49 from OpenDriveLab/46-prediction-format-issue

fix 46-prediction-format-issue

Author: faikit

plugin/mmdet3d/baseline/datasets/openlane_v2_dataset.py

@@ -383,8 +383,7 @@ def format_preds(self, results):
             # lc
 
             pred_lc = result['pred_lc']
-            sorted_index = np.argsort(pred_lc[1][:, 0])[:100]
-            lanes, confidences = pred_lc[0][sorted_index], pred_lc[1][:, 0][sorted_index]
+            lanes, confidences = pred_lc[0], pred_lc[1][:, 0]
 
             lanes = lanes.reshape(-1, lanes.shape[-1] // 3, 3)