python 図形やテキストの描画

Posted 2021-05-09

import numpy as np
import cv2

# 真っ黒な画像を作成
img = np.zeros((512, 512, 3), np.uint8)

# 5pxの太さの青い線を引く
# 色はBGRなので注意
# 戻り値は代入しなくてもOK
img = cv2.line(img, (0, 0), (511, 511), (255, 0, 0), 5)

# 長方形の描画
img = cv2.rectangle(img, (384, 0), (510, 128), (0, 255, 0), 3)

# 円の描画
# 太さに-1を入れると塗りつぶす
img = cv2.circle(img, (447, 63), 63, (0, 0, 255), -1)

# 楕円の描画
img = cv2.ellipse(img, (256, 256), (100, 50), 0, 0, 180, 255, -1)

# 多角形の描画
pts = np.array([[10, 5], [20, 30], [70, 20], [50, 10]], np.int32)
# 頂点数 x 1 x 2の配列にする
pts = pts.reshape((-1, 1, 2))
img = cv2.polylines(img, [pts], True, (0, 255, 255))

# テキストの追加
font = cv2.FONT_HERSHEY_SIMPLEX
cv2.putText(img, 'OpenCV', (10, 500), font, 4, (255, 255, 255), 2, cv2.LINE_AA)

cv2.imshow("img", img)
cv2.waitKey(0)
cv2.destroyAllWindows()