mirror of
https://gitee.com/sui-feng-cb/AzurLaneAutoScript1
synced 2026-03-09 18:39:04 +08:00
Add: Mask class
- Copy module/map/ui_mask.png to assets/mask/MASK_MAP_UI.png
This commit is contained in:
parent
4ed424fc20
commit
f16fda59de
BIN
assets/mask/MASK_MAP_UI.png
Normal file
BIN
assets/mask/MASK_MAP_UI.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 7.8 KiB |
17
module/base/mask.py
Normal file
17
module/base/mask.py
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
import cv2
|
||||||
|
import numpy as np
|
||||||
|
from PIL import Image
|
||||||
|
|
||||||
|
from module.base.template import Template
|
||||||
|
|
||||||
|
|
||||||
|
class Mask(Template):
|
||||||
|
@property
|
||||||
|
def image(self):
|
||||||
|
if self._image is None:
|
||||||
|
self._image = np.array(Image.open(self.file).convert('L'))
|
||||||
|
|
||||||
|
return self._image
|
||||||
|
|
||||||
|
def apply(self, image):
|
||||||
|
return cv2.bitwise_and(image, self.image)
|
||||||
@ -17,7 +17,7 @@ class Template:
|
|||||||
@property
|
@property
|
||||||
def image(self):
|
def image(self):
|
||||||
if self._image is None:
|
if self._image is None:
|
||||||
self._image = np.array(Image.open(self.file))
|
self._image = np.array(Image.open(self.file).convert('RGB'))
|
||||||
|
|
||||||
return self._image
|
return self._image
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user