更换文档检测模型

photo_review/auto_photo_review.py

@@ -15,7 +15,7 @@ from sqlalchemy import update
 from db import MysqlSession
 from db.mysql import BdYljg, BdYlks, ZxIeResult, ZxIeCost, ZxIeDischarge, ZxIeSettlement, ZxPhhd, ZxPhrec
 from doc_dewarp import dewarp
-from object_detection import doc_detect
+from paddle_detection import detector
 from photo_review import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \
     PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR, \
     ADMISSION_ID, SETTLEMENT_ID, AGE, OCR, SETTLEMENT_IE, DISCHARGE_IE, COST_IE, PHHD_BATCH_SIZE, SLEEP_MINUTES, \
@@ -82,7 +82,7 @@ def information_extraction(ie, phrecs, identity):
 
         image = image_util.read(img_path)
         target_images = []
-        target_images += doc_detect.capture_target_area(image)  # 识别文档区域并裁剪
+        target_images += detector.get_book_areas(image)  # 识别文档区域并裁剪
         if not target_images:
             target_images.append(image)  # 识别失败
         angle_count = defaultdict(int, {"0": 0})  # 分割后图片的最优角度统计