42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
import numpy as np
|
||
import cv2
|
||
|
||
|
||
def convert_to_grayscale(image: np.ndarray) -> np.ndarray:
|
||
"""
|
||
画像をグレースケールに変換します。
|
||
OpenCVのcv2.cvtColorを使用して変換を行います。
|
||
|
||
Args:
|
||
image (np.ndarray): 入力画像（RGBまたはRGBA）
|
||
|
||
Returns:
|
||
np.ndarray: グレースケール化された画像（入力と同じチャンネル数）
|
||
"""
|
||
# アルファチャンネルの有無を確認
|
||
has_alpha = image.shape[2] == 4
|
||
alpha = image[..., -1] if has_alpha else None
|
||
|
||
# uint8に変換
|
||
img_uint8 = (image * 255).astype(np.uint8)
|
||
|
||
# RGBからBGRに変換
|
||
if has_alpha:
|
||
img_bgr = cv2.cvtColor(img_uint8[..., :3], cv2.COLOR_RGB2BGR)
|
||
else:
|
||
img_bgr = cv2.cvtColor(img_uint8, cv2.COLOR_RGB2BGR)
|
||
|
||
# グレースケールに変換
|
||
gray = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY)
|
||
|
||
# 出力画像を作成
|
||
result_image = np.zeros_like(image)
|
||
for i in range(3):
|
||
result_image[:, :, i] = gray.astype(np.float32) / 255.0
|
||
|
||
# アルファチャンネルがある場合は保持
|
||
if has_alpha:
|
||
result_image[:, :, 3] = alpha
|
||
|
||
return result_image
|