Commit 2723463a authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add auto censor

parent 17bb1dcd
Loading
Loading
Loading
Loading
+11 −1
Original line number Diff line number Diff line
@@ -5,6 +5,8 @@ from PIL import Image

from cli import _wrap_func_as_cli
from imgutils.data import load_image, grid_transparent, ImageTyping
from imgutils.detect import detect_censors
from imgutils.operate.censor_ import censor_areas
from imgutils.validate.truncate import _mock_load_truncated_images

INCHES_TO_PIXELS = 96
@@ -20,7 +22,15 @@ def _image_input_process(img) -> Tuple[Image.Image, str]:
    else:
        raise TypeError(f'Unknown type of img - {img!r}.')

    return grid_transparent(image), label
    image = grid_transparent(image)
    label = label.rstrip()

    detection = detect_censors(image)
    if detection:
        image = censor_areas(image, method='emoji', areas=[area for area, _, _ in detection])
        label = f'{label}\n(Censored)'

    return image, label


@_wrap_func_as_cli
+0 −867

File deleted.

Preview size limit exceeded, changes collapsed.