python - 从 ArDrone 2.0 获取图像，并使用 cv2

Question

我正在尝试从 ArDrone 2.0 获取图像并使用 Python OpenCV、cv2 来处理图像并创建 Controller 。

这是我的代码:

import cv2
import numpy as np
from pydrone import libardrone

drone = libardrone.ARDrone()
cap = drone.image

while(True):
    ret, frame = cap
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    cv2.imshow('frame',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

但是，我一直遇到这个问题:

Traceback (most recent call last):
  File "ArDrn/7/drony.py", line 14, in <module>
    ret, frame = cap
TypeError: 'NoneType' object is not iterable

有人可以告诉我我做错了什么吗？

Answer 1

在我看来，您可能需要稍微更改逻辑以在尝试使用之前检查 cap 是否为 None:

while(True):
    cap = drone.image
    if not cap:
        time.sleep(0.1)   # or something to save just a touch of CPU spin time, optional
        continue
    ret, frame = cap
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    cv2.imshow('frame',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

可能是在切换图片的过程中或者类似的某个时候，有一段时间cap是None。此外，您似乎没有更新 cap 以在每个循环中指向一个新图像，因此这也可能是一个问题。