文档检测

2024-08-20 13:18:45 +08:00
parent e6891257b9
commit 299b762cad
59 changed files with 4966 additions and 0 deletions
--- a/object_detection/doc_detect.py
+++ b/object_detection/doc_detect.py
@@ -0,0 +1,16 @@
+import numpy as np
+
+from object_detection import CornerNet_Saccade
+from util import image_util
+
+
+def capture_target_area(image, target="book"):
+    detector = CornerNet_Saccade()
+    bboxes = detector(image)
+    target_images = []
+    keep_inds = bboxes[target][:, -1] > 0.5
+    for bbox in bboxes[target][keep_inds]:
+        bbox = bbox[0:4].astype(np.int32)
+        bbox = np.clip(bbox, 0, None)
+        target_images.append(image_util.capture(image, bbox))
+    return target_images