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

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
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 re
import numpy as np
from PIL import Image from PIL import Image
from module.base.mask import Mask from module.base.mask import Mask
@@ -27,17 +28,17 @@ def put_image_mask(image, mask):
def handle_sensitive_image(image): def handle_sensitive_image(image):
""" """
Args: Args:
image (PIL.Image.Image): image:
Returns: Returns:
PIL.Image.Image: PIL.Image.Image:
""" """
if PLAYER_CHECK.match(image, offset=(30, 30)): 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)): 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): def handle_sensitive_text(text):