Question

我是Python的新手，我想提取视频帧，检测egde并使用Python对其进行流传输。我尝试过，但是什么也没出现。此代码有什么问题？

app = Flask(__name__)

def gen():
    cap = cv2.VideoCapture('a.mp4')
    while(True):
        img = cap.read()
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        edge = cv2.Canny(gray, 10, 200, apertureSize=3)
        vis = img.copy()
        vis /=2
        vis[edge != 0] = (0, 255, 0)
        merge = np.concatenate((img, vis), axis=1)
        cv2.imwrite('out.png', merge)
        frame = open('out.png', 'rb').read()
        yield (b'--frame\r\n'
               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')


@app.route('/')
def video_feed():
   return Response(gen(),
                mimetype='multipart/x-mixed-replace; boundary=frame')

if __name__ == '__main__':
      app.run(host='0.0.0.0', debug=True)

谢谢。

Answer 1

您的代码中有两个问题：

第一个在此行中：

img = cap.read()

这是因为cv2.VideoCapture(src).read()返回2个参数。第一个是布尔值，当图像存在时为True，否则为False。第二个参数是当前帧。因此，您必须用类似以下的内容替换行： ret, img = cap.read()或img = cap.read()[1]

此行还导致TypeError：

 vis /=2

之所以发生这种情况，是因为除以2后的奇数整数获得浮点型。要成功运行角检测器，您必须导入numpy库并将此行替换为另一行，该行不会将图像数组元素转换为float：

vis = numpy.trunc(vis / 2)

Python-使用Flask提取和流式传输视频帧

1 个答案: