283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-04-18 21:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

impr: comments & fixes from ChatGPT

Browse Source
This commit is contained in:
283375 2023-06-10 17:27:02 +08:00
parent f9968ae8b3
commit c7e870f42e
1 changed files with 27 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/arcaea_offline_ocr/ocr.py
View File

@ -54,12 +54,17 @@ def filter_digit_results(
): ):
result_sorted_by_x_pos: Dict[ result_sorted_by_x_pos: Dict[
int, List[FilterDigitResultDict] int, List[FilterDigitResultDict]
] = {} # dict[x_pos, dict[int, list[result]]] ] = {} # Dictionary to store results sorted by x-position
# Iterate over each digit and its match results
for digit, match_results in results.items(): for digit, match_results in results.items():
if match_results: if match_results:
# Iterate over each match result
for result in match_results: for result in match_results:
x_pos = result["xywh"][0] x_pos = result["xywh"][0] # Extract x-position from result
_dict = {**result, "digit": digit} _dict = {**result, "digit": digit} # Add digit information to result
# Store result in result_sorted_by_x_pos dictionary
if result_sorted_by_x_pos.get(x_pos) is None: if result_sorted_by_x_pos.get(x_pos) is None:
result_sorted_by_x_pos[x_pos] = [_dict] result_sorted_by_x_pos[x_pos] = [_dict]
else: else:
@ -67,22 +72,32 @@ def filter_digit_results(
x_poses_grouped: List[List[int]] = group_numbers( x_poses_grouped: List[List[int]] = group_numbers(
list(result_sorted_by_x_pos), threshold list(result_sorted_by_x_pos), threshold
) ) # Group x-positions based on threshold
final_result: Dict[ final_result: Dict[
int, List[MatchTemplateMultipleResult] int, List[MatchTemplateMultipleResult]
] = {} # dict[digit, list[Results]] ] = {} # Dictionary to store final filtered results
# Iterate over each group of x-positions
for x_poses in x_poses_grouped: for x_poses in x_poses_grouped:
possible_results = [] possible_results = []
# Iterate over each x-position in the group
for x_pos in x_poses: for x_pos in x_poses:
# Retrieve all results associated with the x-position
possible_results.extend(result_sorted_by_x_pos.get(x_pos, [])) possible_results.extend(result_sorted_by_x_pos.get(x_pos, []))
result = sorted(possible_results, key=lambda d: d["max_val"], reverse=True)[0]
result_digit = result["digit"] if possible_results:
result.pop("digit", None) # Sort the results based on "max_val" in descending order and select the top result
if final_result.get(result_digit) is None: result = sorted(possible_results, key=lambda d: d["max_val"], reverse=True)[0]
final_result[result_digit] = [result] result_digit = result["digit"] # Get the digit value from the result
else: result.pop("digit", None) # Remove the digit key from the result
final_result[result_digit].append(result)
# Store the result in the final_result dictionary
if final_result.get(result_digit) is None:
final_result[result_digit] = [result]
else:
final_result[result_digit].append(result)
return final_result return final_result