Commit c82f0b26 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add segmentation and background function

parent d2535557

docs/source/_libs/plot.py

+4 −5
Original line number Diff line number Diff line
@@ -3,21 +3,21 @@ from typing import Tuple 
import matplotlib.pyplot as plt

from PIL import Image



from imgutils.data import load_image

from imgutils.data import load_image, grid_transparent

from imgutils.validate.truncate import _mock_load_truncated_images





def _image_input_process(img) -> Tuple[Image.Image, str]:

    if isinstance(img, tuple):

        img_file, label = img

        image = load_image(img_file)

        image = load_image(img_file, force_background=None)

    elif isinstance(img, str):

        label = img

        image = load_image(img)

        image = load_image(img, force_background=None)

    else:

        raise TypeError(f'Unknown type of img - {img!r}.')



    return image.convert('RGB'), label

    return grid_transparent(image), label





@_mock_load_truncated_images(True)
@@ -34,7 +34,6 @@ def image_plot(*images, save_as: str, columns=2, keep_axis: bool = False, figsiz 
        xi, yi = i // columns, i % columns

        image, label = _image_input_process(img)

        ax = axs[yi] if rows == 1 else axs[xi, yi]

        print(image, label)

        ax.imshow(image)

        ax.set_title(label)

        if not keep_axis:

docs/source/api_doc/data/background.rst

0 → 100644
+22 −0
Original line number Diff line number Diff line 
imgutils.data.background

=============================



.. currentmodule:: imgutils.data.background



.. automodule:: imgutils.data.background





grid_background

--------------------------------------



.. autofunction:: grid_background







grid_transparent

--------------------------------------



.. autofunction:: grid_transparent

docs/source/api_doc/data/dori.png

0 → 100644
+1.13 MiB
Loading image diff...

docs/source/api_doc/data/grid_transparent.demo.py

0 → 100644
+10 −0
Original line number Diff line number Diff line 
from plot import image_plot



if __name__ == '__main__':

    image_plot(

        'nian.png',

        'dori.png',

        save_as='grid_transparent.dat.svg',

        columns=2,

        figsize=(12, 16),

    )

docs/source/api_doc/data/index.rst

+1 −0
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ imgutils.data 
.. toctree::

    :maxdepth: 3



    background

    encode

    decode

    image