Sweep: refactor e2e_eval from tools/end2end/eval_end2end.py to make it more modular and add unit tests in eval_end2end_tests.py

Here's the PR! https://github.com/kevinlu1248/PaddleOCR/pull/17.

💎 Sweep Pro: I'm using GPT-4. You have unlimited GPT-4 tickets.

Actions (click)

[ ] ↻ Restart Sweep

Step 1: 🔎 Searching

I found the following snippets in your repository. I will now analyze these snippets and come up with a plan.

Some code snippets I looked at (click to expand). If some file is missing from here, you can mention the path in the ticket description.

https://github.com/kevinlu1248/PaddleOCR/blob/47ff250c837ab72890c3ff98d6fa3f68e61f713d/tools/end2end/eval_end2end.py#L1-L175 https://github.com/kevinlu1248/PaddleOCR/blob/47ff250c837ab72890c3ff98d6fa3f68e61f713d/applications/PCB字符识别/PCB字符识别.md#L545-L625 https://github.com/kevinlu1248/PaddleOCR/blob/47ff250c837ab72890c3ff98d6fa3f68e61f713d/applications/轻量级车牌识别.md#L635-L745 https://github.com/kevinlu1248/PaddleOCR/blob/47ff250c837ab72890c3ff98d6fa3f68e61f713d/tools/end2end/readme.md#L5-L62 https://github.com/kevinlu1248/PaddleOCR/blob/47ff250c837ab72890c3ff98d6fa3f68e61f713d/tools/end2end/eval_end2end.py#L125-L192

Step 2: ⌨️ Coding

[X] Extract tools/end2end/eval_end2end.py ✓ https://github.com/kevinlu1248/PaddleOCR/commit/a89267cdd5bf3191b68eeb37c4ad68434304c4b7

Extract tools/end2end/eval_end2end.py with contents: 1. Extract the code block that calculates the Intersection over Union (IoU) between two polygons into a new function named `calculate_iou`. 2. Extract the code block that calculates the edit distance between two strings into a new function named `calculate_edit_distance`. 3. Extract the code block that calculates the precision, recall, and fmeasure into a new function named `calculate_metrics`. 4. Extract the code block that handles the matching of ground truth and detected text into a new function named `match_gt_and_dt`.
[X] Create tools/end2end/eval_end2end_tests.py ✓ https://github.com/kevinlu1248/PaddleOCR/commit/236b30f4a1bb5b4ed354c5198eccfd4bb55df7eb

Create tools/end2end/eval_end2end_tests.py with contents: 1. Create a test for the `calculate_iou` function. The test should include cases where the polygons do not intersect, partially intersect, and fully intersect. 2. Create a test for the `calculate_edit_distance` function. The test should include cases where the strings are identical, completely different, and partially different. 3. Create a test for the `calculate_metrics` function. The test should include cases with different values of hit, dt_count, and gt_count to cover different possible scenarios. 4. Create a test for the `match_gt_and_dt` function. The test should include cases where there are unmatched ground truth and detected text, and cases where all ground truth and detected text are matched.
[X] Check tools/end2end/eval_end2end_tests.py ✓

Sandbox logs for

Step 3: 🔁 Code Review

I have finished reviewing the code for completeness. I did not find errors for sweep/refactor-e2e-eval-and-add-tests.

🎉 Latest improvements to Sweep:

Sweep now uses the rope library to refactor Python! Check out Large Language Models are Bad at Refactoring Code. To have Sweep refactor your code, try sweep: Refactor <your_file>.py to be more modular!
Sweep finds and fixes bugs in your code by writing unit tests! Check out Having GPT-4 Iterate on Unit Tests like a Human.

💡 To recreate the pull request edit the issue title or description. To tweak the pull request, leave a comment on the pull request. ^{Join Our Discord}

kevinlu1248 / PaddleOCR