使用yield而不是return创建一个函数来连续从http流生成帧

我想创建一个函数，使用以下命令从 HTTP 流中读取帧requests http://docs.python-requests.org/en/master/并返回每一帧。但由于流读取器基于迭代器对象（如果我理解正确的话），返回帧会破坏流。

我正在使用的代码（工作得很好，来自this https://stackoverflow.com/a/42904920/6331998回答）：

import cv2
import requests
import numpy as np

r = requests.get('http://roofcam.warwick.ac.uk/cgi-bin/faststream.jpg', stream=True)
if(r.status_code == 200):
    bytes_buffer = bytes()
    for chunk in r.iter_content(chunk_size=1024):
        bytes += chunk
        a = bytes_buffer.find(b'\xff\xd8')
        b = bytes_buffer.find(b'\xff\xd9')
        if a != -1 and b != -1:
            jpg = bytes_buffer[a:b+2]
            bytes_buffer = bytes_buffer[b+2:]
            i = cv2.imdecode(np.fromstring(jpg, dtype=np.uint8), cv2.IMREAD_COLOR)
            cv2.imshow('i', i)
            if cv2.waitKey(1) == 27:
                exit(0)
else:
    print("Received unexpected status code {}".format(r.status_code))

我想要做的事情的概念（如果它是一个 while 函数而不是 for using 迭代器，我想要工作的返回）：

import cv2
import requests
import numpy as np

r = requests.get('http://roofcam.warwick.ac.uk/cgi-bin/faststream.jpg', stream=True)

def get_frame_from_stream(r):
    if(r.status_code == 200):
        bytes_buffer = bytes()
        for chunk in r.iter_content(chunk_size=1024):
            bytes_buffer += chunk
            a = bytes_buffer.find(b'\xff\xd8')
            b = bytes_buffer.find(b'\xff\xd9')
            if a != -1 and b != -1:
                jpg = bytes_buffer[a:b + 2]
                bytes_buffer = bytes_buffer[b + 2:]
                i = cv2.imdecode(np.fromstring(
                    jpg, dtype=np.uint8), cv2.IMREAD_COLOR)
                return i

    else:
        print("Received unexpected status code {}".format(r.status_code))
        return None

while True:
    if img is not None:
        img = get_frame_from_stream(r)
        cv2.imshow('i', img)
        cv2.waitKey(0)
    else:
        break

所以基本上我想返回原始代码显示帧的每个帧，以便我可以对其执行一些处理。但我不明白 iter_content 到底是如何让原始代码连续工作的。

（我不知道如何命名这个问题 - 欢迎更好的标题）

Python有一个非常美丽的概念generators https://wiki.python.org/moin/Generators，这可能是您正在寻找的东西，所以基本上是为了制造一个generator我们使用关键字yield https://docs.python.org/2/reference/simple_stmts.html#the-yield-statement代替return https://docs.python.org/2/reference/simple_stmts.html#the-return-statement。这两个关键字之间的基本区别在于return语句一旦遇到就会停止函数执行，而另一方面yield关键字让执行继续并继续生成值直到存活。它可以用一个简单的例子来形象化：

def sample_function():
    # The body would be replaced by image generating code.
    for i in xrange(20):
        yield i**2

for x in sample_function():
    print x, 
>>> 0 1 4 9 16 25 36 49 64 81 100 121 144 169 196 225 256 289 324 361

如果我们使用以下命令重现相同的场景return语句，那么它可能看起来像：

def sample_function(i):
    return i**2

for i in xrange(0, 20):
    print sample_function(i),

所以在上面的代码中如果你替换return关键字与yield，那么您可以将帧迭代为：

for frame in get_frame_from_stream(r):
    cv2.imshow('i', frame)

没有必要while循环就是这种情况，除非并且直到该流处于活动状态，该方法将继续生成帧。