File size: 1,886 Bytes
57bc6ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
"""画像処理ユーティリティ"""
import base64
import io
from typing import Tuple
import cv2
import numpy as np
from PIL import Image
def resize_frame(
frame: np.ndarray, width: int = 640, height: int = 480
) -> np.ndarray:
"""フレームをリサイズ"""
return cv2.resize(frame, (width, height), interpolation=cv2.INTER_AREA)
def frame_to_base64(frame: np.ndarray, quality: int = 85) -> str:
"""フレームをBase64エンコード"""
# BGR -> RGB
if len(frame.shape) == 3 and frame.shape[2] == 3:
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
else:
frame_rgb = frame
img = Image.fromarray(frame_rgb)
buffer = io.BytesIO()
img.save(buffer, format="JPEG", quality=quality)
return base64.b64encode(buffer.getvalue()).decode("utf-8")
def frame_to_pil(frame: np.ndarray) -> Image.Image:
"""NumPy配列をPIL Imageに変換"""
if len(frame.shape) == 3 and frame.shape[2] == 3:
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
else:
frame_rgb = frame
return Image.fromarray(frame_rgb)
def pil_to_frame(img: Image.Image) -> np.ndarray:
"""PIL ImageをNumPy配列に変換"""
frame = np.array(img)
if len(frame.shape) == 3 and frame.shape[2] == 3:
frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
return frame
def rotate_frame(frame: np.ndarray, angle: int) -> np.ndarray:
"""フレームを回転（0, 90, 180, 270度）"""
if angle == 90:
return cv2.rotate(frame, cv2.ROTATE_90_CLOCKWISE)
elif angle == 180:
return cv2.rotate(frame, cv2.ROTATE_180)
elif angle == 270:
return cv2.rotate(frame, cv2.ROTATE_90_COUNTERCLOCKWISE)
return frame
def get_frame_dimensions(frame: np.ndarray) -> Tuple[int, int]:
"""フレームの寸法を取得 (width, height)"""
return frame.shape[1], frame.shape[0]
|