47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
|
|
||
|
|
import cv2
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import json
|
||
|
|
from paddleocr import PaddleOCR
|
||
|
|
|
||
|
|
def run_sanity():
|
||
|
|
img_path = "seal_cropped.png"
|
||
|
|
if not os.path.exists(img_path):
|
||
|
|
print(f"Error: {img_path} not found")
|
||
|
|
return
|
||
|
|
|
||
|
|
img = cv2.imread(img_path)
|
||
|
|
h, w = img.shape[:2]
|
||
|
|
print(f"Image: {img_path}, Shape: {h}x{w}")
|
||
|
|
|
||
|
|
ocr = PaddleOCR(use_angle_cls=True, lang='ch')
|
||
|
|
|
||
|
|
# OCR on Raw
|
||
|
|
raw_res = ocr.ocr(img_path)
|
||
|
|
|
||
|
|
# Manual Unwarp
|
||
|
|
cx, cy = w//2, h//2
|
||
|
|
radius = min(cx, cy)
|
||
|
|
out_w = int(radius * 3.14 * 2)
|
||
|
|
out_h = radius
|
||
|
|
unwarped = cv2.warpPolar(img, (out_w, out_h), (cx, cy), radius, cv2.WARP_FILL_OUTLIERS + cv2.WARP_POLAR_LINEAR)
|
||
|
|
unwarp_path = "debug_unwarp_sanity.png"
|
||
|
|
cv2.imwrite(unwarp_path, unwarped)
|
||
|
|
|
||
|
|
# OCR on Unwarp
|
||
|
|
unwarp_res = ocr.ocr(unwarp_path)
|
||
|
|
|
||
|
|
results = {
|
||
|
|
"raw": str(raw_res),
|
||
|
|
"unwarp": str(unwarp_res)
|
||
|
|
}
|
||
|
|
|
||
|
|
with open("sanity_results.json", "w", encoding="utf-8") as f:
|
||
|
|
json.dump(results, f, ensure_ascii=False, indent=2)
|
||
|
|
|
||
|
|
print("Sanity results saved to sanity_results.json")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
run_sanity()
|