sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-09 18:39:04 +08:00
Code

Fix: handle_sensitive_image

Browse Source
This commit is contained in:
LmeSzinc 2020-09-26 15:36:17 +08:00
parent 6defbbefc3
commit 6f7ff6eb0d
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
module/handler/sensitive_info.py
View File

@ -1,5 +1,6 @@
import re
import numpy as np
from PIL import Image
from module.base.mask import Mask
@ -27,17 +28,17 @@ def put_image_mask(image, mask):
def handle_sensitive_image(image):
"""
Args:
image (PIL.Image.Image):
image:
Returns:
PIL.Image.Image:
"""
if PLAYER_CHECK.match(image, offset=(30, 30)):
image = MASK_PLAYER.apply(image)
image = Image.fromarray(MASK_PLAYER.apply(image), mode='RGB')
if MAIN_CHECK.match(image, offset=(30, 30)):
image = MASK_MAIN.apply(image)
image = Image.fromarray(MASK_MAIN.apply(image), mode='RGB')
return Image.fromarray(image, mode='RGB')
return image
def handle_sensitive_text(text):