pythonでデータ型が変わってしまう

Question

以下のコードで顔検出をし、結果を辞書形式にしたいのですが、

import cv2
import json

face_cascade_path = './haarcascades/haarcascade_frontalface_default.xml'

face_cascade = cv2.CascadeClassifier(face_cascade_path)

cap = cv2.VideoCapture(0)

while True:
    file = open('sample2.json','w')
    key = ["x","y","w","h"]
    ret, img = cap.read()
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.3, minNeighbors=5)
    for x, y, w, h in faces:
        cv2.rectangle(img, (x, y), (x + w, y + h), (255, 0, 0), 2)
        face = img[y: y + h, x: x + w]
        face_gray = gray[y: y + h, x: x + w]
    cv2.imshow('video image', img)
    target = list(faces)
    dic = dict(zip(key,target))
    print(faces)
    key = cv2.waitKey(10)
    if key == 27:  # ESCキーで終了
        break

cap.release()
cv2.destroyAllWindows()

tuple型とlist型で変わってしまいます。

()
()
()
[[978 441 176 176]]
()
()
()
[[672 287 430 430]]
()
()
[[638 280 430 430]]
[[643 280 430 430]]
()
[[528 259  52  52]]

どうにも、空の時はtuple型らしいです。
tuple型だと辞書にできないそうです
何卒、ご教授のほどよろしくお願いします

Answer 1

1枚の画像から複数の顔が検出される可能性があるので、1フレームあたりの検出結果はキーが x, y, w, h の dict の list として保持されてはどうでしょうか。
この場合、1つも顔が検出されない場合は [] となりますし、1つ以上検出された場合は

[{"x": <x座標>, "y": <y座標>, "w": <幅>, "h": <高さ>},
 {"x": <x座標>, "y": <y座標>, "w": <幅>, "h": <高さ>},
 ...
]

となります。

import cv2
import json


face_cascade_path = "haarcascade_frontalface_default.xml"
face_cascade = cv2.CascadeClassifier(face_cascade_path)

cap = cv2.VideoCapture(0)

detections = []

while True:
    ret, img = cap.read()

    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.3, minNeighbors=5)

    det = []
    for x, y, w, h in faces:
        cv2.rectangle(img, (x, y), (x + w, y + h), (255, 0, 0), 2)
        face = img[y : y + h, x : x + w]
        face_gray = gray[y : y + h, x : x + w]

        det.append({"x": x, "y": y, "w": w, "h": h})
    detections.append(det)
    print(det)

    cv2.imshow("video image", img)

    key = cv2.waitKey(10)
    if key == 27:
        break  # ESCキーで終了

cap.release()
cv2.destroyAllWindows()

# 検出結果を書き込む。
with open("detections.json", "w") as f:
    json.dump(detections, f, indent=4)

Comments

[tomoki_takaba]

ありがとうございます！！！
うまくいきました！！
本当にありがとうございます！！！