Commit 08bcdf51 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add star censor demo, test skip

parent c6ba1b12
Loading
Loading
Loading
Loading
+25 −0
Original line number Diff line number Diff line
from PIL import Image

from imgutils.operate import ImageBasedCensor, register_censor_method
from imgutils.operate import censor_areas
from plot import image_plot

if __name__ == '__main__':
    register_censor_method('star', ImageBasedCensor, images=['star.png'])

    image = Image.open('genshin_post.jpg')
    censor_img = Image.open('star.png')
    areas = [
        (967, 143, 1084, 261),
        (246, 208, 331, 287),
        (662, 466, 705, 514),
        (479, 283, 523, 326)
    ]

    image_plot(
        (image, 'origin'),
        (censor_img, 'censor_img'),
        (censor_areas(image, 'star', areas), 'star_censored'),
        columns=3,
        figsize=(10, 4),
    )
+9.24 KiB
Loading image diff...