Python3 параллельный процесс opencv видеокадры

Question

Python3 параллельный процесс opencv видеокадры

1

У меня есть видеофайл, который мне нужно обрабатывать по кадре, и мне нужно показать результаты в кадрах впоследствии. В настоящее время я делаю обработку последовательно и показывая кадры один за другим.

Теперь я хотел бы обрабатывать фреймы параллельно, а не последовательно. После того, как X количество кадров было обработано, необходимо показать cv2.imshow и показать обработанные кадры в правильном порядке.

В настоящее время мой последовательный код выглядит так

import cv2
import requests


def process_frame(bgr_image, jpg_as_text):
    try:
        # Post to api for processing and get the results
        # result = requests.post("example.com", data={"jpg": jpg_as_text})

        # Add results to bgr_image
        # cv2.putText()
    except Exception as e:
        print(e)
        pass
    # Show the frame
    cv2.imshow("frame", bgr_image)


video = cv2.VideoCapture("video.mp4")
i = 0


while video.isOpened():
    ret, bgr_image = video.read()
    if ret == True:
        img_height, img_width, _ = bgr_image.shape
        jpg_as_text = cv2.imencode(".jpg", bgr_image)[1].tostring()
        process_frame(bgr_image, jpg_as_text)
        print(i)
        i += 1
    else:
        break
    if cv2.waitKey(1) & 0xFF == ord("q"):
        break

video.release()
cv2.destroyAllWindows()

Теперь, что я должен реорганизовать для параллельной обработки и предварительного просмотра кадров после обработки X кадров.

Kumaran 18 июнь 2018, в 16:36

Источник

Теги:

python

opencv

multithreading

asynchronous

multiprocessing

1 ответ

Ещё вопросы

Michael James · Accepted Answer · 2018-06-18T11-33-00.000Z

OpenCV имеет пример многопоточной обработки видео в своем реестре github.

https://github.com/opencv/opencv/blob/master/samples/python/video_threaded.py

Импорт ThreadPool из multiprocessing.pool и запуск нового потока для каждого ядра процессора. OpenCV имеет функцию getNumberOfCPUs()

Пример:

from __future__ import print_function

import numpy as np
import cv2 as cv

from multiprocessing.pool import ThreadPool
from collections import deque

from common import clock, draw_str, StatValue
import video


class DummyTask:
    def __init__(self, data):
    self.data = data
def ready(self):
    return True
def get(self):
    return self.data

if __name__ == '__main__':
    import sys

    print(__doc__)

    try:
        fn = sys.argv[1]
    except:
        fn = 0
    cap = video.create_capture(fn)


    def process_frame(frame, t0):
    # some intensive computation...
        frame = cv.medianBlur(frame, 19)
        frame = cv.medianBlur(frame, 19)
        return frame, t0

    threadn = cv.getNumberOfCPUs()
    pool = ThreadPool(processes = threadn)
    pending = deque()

    threaded_mode = True

    latency = StatValue()
    frame_interval = StatValue()
    last_frame_time = clock()
    while True:
        while len(pending) > 0 and pending[0].ready():
            res, t0 = pending.popleft().get()
            latency.update(clock() - t0)
            draw_str(res, (20, 20), "threaded      :  " +             str(threaded_mode))
            draw_str(res, (20, 40), "latency        :  %.1f ms" %     (latency.value*1000))
            draw_str(res, (20, 60), "frame interval :  %.1f ms" % (frame_interval.value*1000))
            cv.imshow('threaded video', res)
        if len(pending) < threadn:
            ret, frame = cap.read()
            t = clock()
            frame_interval.update(t - last_frame_time)
            last_frame_time = t
            if threaded_mode:
                task = pool.apply_async(process_frame, (frame.copy(), t))
            else:
                task = DummyTask(process_frame(frame, t))
            pending.append(task)
        ch = cv.waitKey(1)
        if ch == ord(' '):
            threaded_mode = not threaded_mode
        if ch == 27:
            break
    cv.destroyAllWindows()

Вы должны иметь возможность использовать этот примерный код и поместить обработку изображения в функцию process_frame.

Добавьте счетчик в свой цикл и вызовите cv2.imshow, когда count == X

Не размещайте ссылку как ответ. Что если страницы больше не существует, когда кто-то ищет ответ?
@Cid, спасибо, я отредактировал, чтобы включить образец.