seal_rec.py 705 B

12345678910111213141516171819
  1. from pathlib import Path
  2. from .base import BaseResult
  3. class SealResult(BaseResult):
  4. """SealResult"""
  5. def save_to_img(self, save_path):
  6. if not save_path.lower().endswith((".jpg", ".png")):
  7. input_path = self["input_path"]
  8. save_path = Path(save_path) / f"{Path(input_path).stem}"
  9. else:
  10. save_path = Path(save_path).stem
  11. layout_save_path = f"{save_path}_layout.jpg"
  12. layout_result = self["layout_result"]
  13. layout_result.save_to_img(layout_save_path)
  14. for idx, seal_result in enumerate(self["ocr_result"]):
  15. ocr_save_path = f"{save_path}_{idx}_seal_ocr.jpg"
  16. seal_result.save_to_img(ocr_save_path)