Commit ca55cbdd authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): optimize model usage

parent c2408445

docs/source/api_doc/edge/edge.demo.py

+0 −3
Original line number Diff line number Diff line 
from functools import partial



from hbutils.reflection import nested_for

from tqdm.auto import tqdm
@@ -11,7 +9,6 @@ if __name__ == '__main__': 
    funcs = [

        ('canny', edge_image_with_canny),

        ('lineart', edge_image_with_lineart),

        ('lineart\n(coarse)', partial(edge_image_with_lineart, coarse=True)),

        ('lineart\nanime', edge_image_with_lineart_anime)

    ]

    table = [[item] for item in demo_images]

docs/source/api_doc/edge/lineart.demo.py

+1 −5
Original line number Diff line number Diff line
@@ -10,13 +10,9 @@ if __name__ == '__main__': 
    demo_images = ['hutao.png', 'skadi.jpg']

    funcs = [

        ('lineart', partial(edge_image)),

        ('lineart\ncoarse', partial(edge_image, coarse=True)),

        ('lineart\nbackcolor=transparent', partial(edge_image, backcolor='transparent')),

        ('lineart\ncoarse\nbackcolor=transparent', partial(edge_image, backcolor='transparent', coarse=True)),

        ('lineart\nbackcolor=white\nforecolor=black',

         partial(edge_image, backcolor='white', forecolor='black')),

        ('lineart\ncoarse\nbackcolor=white\nforecolor=black',

         partial(edge_image, backcolor='white', forecolor='black', coarse=True)),

    ]

    table = [[item] for item in demo_images]

    for (xi, origin_image), (_, func) in \
@@ -28,5 +24,5 @@ if __name__ == '__main__': 
        columns=['origin', *(name for name, _ in funcs)],

        rows=['' for _ in demo_images],

        save_as='lineart.dat.svg',

        figsize=(2400, 980),

        figsize=(1600, 980),

    )

imgutils/edge/lineart.py

+5 −0
Original line number Diff line number Diff line
@@ -64,6 +64,11 @@ def edge_image_with_lineart(image: ImageTyping, coarse: bool = False, detect_res 
    Examples::

        .. image:: lineart.dat.svg

            :align: center



        When ``coarse`` is used:



        .. image:: lineart_coarse.dat.svg

            :align: center

    """

    return _get_image_edge(

        image,