38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import cv2
|
|
import numpy as np
|
|
from pdf2image import pdf2image
|
|
|
|
# import pandas as pd
|
|
from PIL import Image
|
|
import timeit
|
|
from os import path
|
|
from cv_analysis.locations import METADATA_TESTFILES, PNG_FOR_TESTING, PNG_FIGURES_DETECTED
|
|
from cv_analysis.utils.detection import detect_large_coherent_structures
|
|
from cv_analysis.utils.display import show_mpl
|
|
from cv_analysis.utils.draw import draw_rectangles
|
|
from cv_analysis.utils.post_processing import remove_included
|
|
from cv_analysis.utils.filters import is_large_enough, has_acceptable_format
|
|
from cv_analysis.utils.structures import Rectangle
|
|
from cv_analysis.utils.text import remove_primary_text_regions
|
|
from cv_analysis.utils.visual_logging import vizlogger
|
|
|
|
|
|
def is_likely_figure(cont, min_area=5000, max_width_to_hight_ratio=6):
|
|
return is_large_enough(cont, min_area) and has_acceptable_format(cont, max_width_to_hight_ratio)
|
|
|
|
|
|
def detect_figures(image: np.array):
|
|
|
|
image = image.copy()
|
|
vizlogger.debug(image, "figures01_start.png")
|
|
|
|
image = remove_primary_text_regions(image)
|
|
vizlogger.debug(image, "figures02_remove_text.png")
|
|
cnts = detect_large_coherent_structures(image)
|
|
|
|
cnts = filter(is_likely_figure, cnts)
|
|
rects = map(cv2.boundingRect, cnts)
|
|
rects = remove_included(rects)
|
|
|
|
return list(map(Rectangle.from_xywh, rects))
|