PythonでARマーカーを利用する（OpenCV、ArUco Markers）

py -3.10 -m venv pyenv-ar
cd .\pyenv-ar\
.\Scripts\activate

pip install opencv-python opencv-contrib-python

import cv2
import cv2.aruco as aruco

# 使用する辞書を取得
aruco_dict = aruco.getPredefinedDictionary(aruco.DICT_4X4_50)

# マーカーIDとサイズの設定
marker_id = 0
marker_size = 200  # ピクセル単位

# マーカー画像の生成
marker_img = aruco.generateImageMarker(aruco_dict, marker_id, marker_size)

# 画像の保存
cv2.imwrite("aruco_marker_0.png", marker_img)
print("aruco_marker_0.png を保存しました。")

import cv2
import cv2.aruco as aruco
import os

# 保存先のディレクトリを作成
save_dir = "marker"
if not os.path.exists(save_dir):
    os.makedirs(save_dir)

# 使用する辞書を取得
aruco_dict = aruco.getPredefinedDictionary(aruco.DICT_4X4_50)

# マーカーサイズの設定
marker_size = 200  # ピクセル単位

# 0から49までのマーカーIDを生成して保存
for marker_id in range(50):
    # マーカー画像の生成
    marker_img = aruco.generateImageMarker(aruco_dict, marker_id, marker_size)
    
    # 画像の保存
    filename = os.path.join(save_dir, f"aruco_marker_{marker_id}.png")
    cv2.imwrite(filename, marker_img)
    print(f"{filename} を保存しました。")

import cv2
import cv2.aruco as aruco

# カメラの読み込み
cap = cv2.VideoCapture(1)

# ArUco辞書の取得
aruco_dict = aruco.getPredefinedDictionary(aruco.DICT_4X4_50)

# ディテクターパラメータの設定
parameters = aruco.DetectorParameters()

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # マーカーの検出
    corners, ids, rejected_img_points = aruco.detectMarkers(frame, aruco_dict, parameters=parameters)

    if ids is not None:
        # マーカーに枠とIDを描画
        aruco.drawDetectedMarkers(frame, corners, ids)

    # 結果の表示
    cv2.imshow("AR Marker Detection", frame)

    # 'q'キーで終了
    if cv2.waitKey(1) & 0xFF == ord("q"):
        break

# リソースの解放
cap.release()
cv2.destroyAllWindows()