Unverified Commit 8ba44973 authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #44 from deepghs/dev/ocr

dev(narugo): add text detection
parents c7a5c158 73c8b403
Loading
Loading
Loading
Loading
+9 −0
Original line number Diff line number Diff line
from typing import Tuple, List

import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

from cli import _wrap_func_as_cli
@@ -44,6 +45,14 @@ def image_plot(*images, save_as: str, columns=2, keep_axis: bool = False, figsiz
    n = len(images)
    rows = (n + columns - 1) // columns
    fig, axs = plt.subplots(rows, columns, figsize=figsize)
    if rows == 1 and columns == 1:
        axs = np.array([[axs]])
    elif rows == 1:
        axs = axs[None, ...]
    elif columns == 1:
        axs = axs[..., None]
    else:
        pass
    plt.subplots_adjust(wspace=0.2, hspace=0.15)
    for i, img in enumerate(images, start=0):
        xi, yi = i // columns, i % columns
+1 −0
Original line number Diff line number Diff line
@@ -16,5 +16,6 @@ imgutils.detect
    hand
    head
    person
    text
    visual
+14 −0
Original line number Diff line number Diff line
imgutils.detect.text
==========================

.. currentmodule:: imgutils.detect.text

.. automodule:: imgutils.detect.text


detect_text
------------------------------

.. autofunction:: detect_text

+32.4 KiB
Loading image diff...
+75.2 KiB
Loading image diff...
Loading