minor fix

jkowalk · Apr 28, 2024 · 55df4ea · 55df4ea
1 parent 8e4fc64
commit 55df4ea
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/challenge/README.md b/challenge/README.md
@@ -308,10 +308,6 @@ Chances are that you are not logged in to the current competition space.
 
 Please refresh the page, click `Login with Hugging Face` at the bottom of the left panel.
 
-### If I encounter a reshape error, what should I do?
-
-You should first refer to this [location](https://github.com/OpenDriveLab/DriveLM/blob/main/challenge/evaluation.py#L90). Most of the reshape errors occur here.
-
 ### Finally, which dataset do we submit to the competition?
 
 Please refrain from using demo data. Instead, utilize the [validation data](https://drive.google.com/file/d/1fsVP7jOpvChcpoXVdypaZ4HREX1gA7As/view?usp=sharing) for inference and submission to the evaluation server.

diff --git a/challenge/evaluation.py b/challenge/evaluation.py
@@ -84,6 +84,8 @@ def match_result(self, answer, GT):
  answer_nums = re.findall(r'\d+\.\d+', answer)
  GT_nums = re.findall(r'\d+\.\d+', GT)
  # transform string into float
+ if len(answer_nums) % 2 != 0:
+ answer_nums = answer_nums[:-1]
  answer_nums = np.array([list(map(float, x.split()))[0] for x in answer_nums]).reshape(-1, 2)
  GT_nums = np.array([list(map(float, x.split()))[0] for x in GT_nums]).reshape(-1, 2)
  length = len(GT_nums)