VideoCaptureのgrab()とretrieve()で最新画像を描画する
0.概要
スレッドを利用して取り込みと描画を分けているコードはいくつか見つかったのですが、所望の機能のものがなかったため、以下のようなコードを考えました。
1.コード
#! /usr/bin/python3
import cv2, threading, time
class Measure():
def __init__(self, regular):
self._regular = regular
self._number = 1
self._start = time.time()
def elapsed(self):
return (time.time() - self._start)
def difference(self):
diff = (self._regular * self._number) - self.elapsed()
return diff if diff > 0 else 0
def fps(self):
return self._number / self.elapsed()
def next(self):
self._number += 1
class VideoThread():
def __init__(self, index, name):
self._index = index
self._camera = cv2.VideoCapture(index)
self._name = name
number = 0
while number != 10: # Wait for the transfer to stabilize.
if self._camera.grab():
number += 1
self._alive = True
self._thread = threading.Thread(target=self.grabber)
self._thread.daemon = True
self._thread.start()
def name(self):
return self._name + "[" + str(self._index) + "]"
def grabber(self):
measure = Measure(regular=1/120)
while self._alive and self._camera.grab():
#time.sleep(measure.difference())
print("fps(" + self.name() + ") {0:.3f}".format(measure.fps()))
measure.next()
def retrieve(self):
_, frame = self._camera.retrieve()
return frame
def close(self):
self._alive = False
self._thread.join()
self._camera.release()
def main():
video_thread_0 = VideoThread(0, "camera")
video_thread_1 = VideoThread(1, "camera")
# Wait for the window to stabilize.
cv2.imshow(video_thread_0.name(), video_thread_0.retrieve())
cv2.imshow(video_thread_1.name(), video_thread_1.retrieve())
cv2.waitKey(1)
measure = Measure(regular=1/120)
while chr(cv2.waitKey(1) & 255) != 'q':
cv2.imshow(video_thread_0.name(), video_thread_0.retrieve())
cv2.imshow(video_thread_1.name(), video_thread_1.retrieve())
time.sleep(measure.difference())
print("fps(draw) {0:.3f}".format(measure.fps()))
measure.next()
video_thread_0.close()
video_thread_1.close()
if __name__ == "__main__":
main()
2.説明
cv::VideoCapture Class Reference
grab()
retrieve()
「# Wait for ~」の部分については、「カメラの転送開始時」や「Windowの生成直後」に負荷が大きく、正確な時間計測ができないため、安定するまで待たせる処理を適宜挟んでいます。
Author And Source
この問題について(VideoCaptureのgrab()とretrieve()で最新画像を描画する), 我々は、より多くの情報をここで見つけました https://zenn.dev/kaiyu/articles/6294df3b5038bad435e1著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Collection and Share based on the CC protocol