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

Opt: Move rgb2gray from to module/base/utils.py

Browse Source
This commit is contained in:
LmeSzinc
2020-09-03 19:40:33 +08:00
parent 857adacfbf
commit ffa36b0fc2
8 changed files with 19 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
module/base/utils.py
View File

@@ -231,6 +231,21 @@ def crop(image, area):
return cv2.copyMakeBorder(image[y1:y2, x1:x2], *border, borderType=cv2.BORDER_CONSTANT, value=(0, 0, 0))
def rgb2gray(image):
"""
Args:
image (np.ndarray): Shape (height, width, channel)
Returns:
np.ndarray: Shape (height, width)
"""
r, g, b = cv2.split(image)
return cv2.add(
cv2.multiply(cv2.max(cv2.max(r, g), b), 0.5),
cv2.multiply(cv2.min(cv2.min(r, g), b), 0.5)
)
def get_color(image, area):
"""Calculate the average color of a particular area of the image.