Bye Bye Moore

PoCソルジャーな零細事業主が作業メモを残すブログ

OpenCV for pythonでMJPEG用のテスト画像をつくる

import cv2

class MyVideoCapture():
    def createTestImage(self,count):
        # 画像の解像度
        image = np.zeros((480, 640, 3), dtype=np.uint8)

        # テキストのフォントとスケール
        font = cv2.FONT_HERSHEY_SIMPLEX
        font_scale = 2.0

        # テキストの色(白)
        text_color = (255, 255, 255)

        # テキストのサイズと位置を計算
        text = str(count)
        text_size, _ = cv2.getTextSize(text, font, font_scale, thickness=1)
        text_x = (image.shape[1] - text_size[0]) // 2
        text_y = (image.shape[0] + text_size[1]) // 2

        # 背景色を設定(countの値に基づいて色を選択)
        colors = [(55, 0, 0), (0, 0, 55), (0, 55, 55), (55, 55, 55)] 
        background_color = colors[count % len(colors)]
        image[:] = background_color

        # テキストを描画
        cv2.putText(image, text, (text_x, text_y), font, font_scale, text_color, thickness=2, lineType=cv2.LINE_AA)

        return image

#...