Создание видео-виджетов OpenCV для использования в Kivy - PullRequest
Новая
       6

Создание видео-виджетов OpenCV для использования в Kivy

0 голосов
/ 19 февраля 2020

У меня есть от 2 до 4 потоков RTSP, которые я хочу отобразить в макете коробки в Kivy, но я не могу понять, как использовать язык kv для достижения этой цели. Я могу получить отображение одной камеры с помощью этого кода: 

class KivyCamera(Image):
    def __init__(self, capture, fps, **kwargs):
        super(KivyCamera, self).__init__(**kwargs)
        self.capture = capture
        Clock.schedule_interval(self.update, 1.0 / fps)

    def update(self, dt):
        ret, frame = self.capture.read()
        if ret:
            # convert it to texture
            buf1 = cv2.flip(frame, 0)
            buf = buf1.tostring()
            image_texture = Texture.create(
                size=(frame.shape[1], frame.shape[0]), colorfmt='bgr')
            image_texture.blit_buffer(buf, colorfmt='bgr', bufferfmt='ubyte')
            # display image from the texture
            self.texture = image_texture


class CamApp(App):
    def build(self):
        self.capture = cv2.VideoCapture('stream address removed for stack overflow')
        self.my_camera = KivyCamera(capture=self.capture, fps=30)
        return self.my_camera

    def on_stop(self):
        #without this, app will not exit even if the window is closed
        self.capture.release()


if __name__ == '__main__':
    CamApp().run()

Однако это только использование .py. Я хотел бы иметь возможность использовать язык kv для создания экземпляров KivyCamera и устанавливать разные значения capture для каждого.

Например: 

BoxLayout:
        orientation: 'horizontal'

        KivyCamera:
            capture: 'myvalue'
            pos_hint: {'left': 1}

        KivyCamera:
            capture: 'anothervalue'
            pos_hint: {'right': 1}

1 Ответ

1 голос
/ 19 февраля 2020

Вы можете сделать свойства fps и источник (индекс, имя файла и т. Д. c) вместо VideoCapture:

main.py 

from kivy.app import App
from kivy.uix.image import Image
from kivy.clock import Clock
from kivy.graphics.texture import Texture
from kivy.properties import ObjectProperty, NumericProperty

import cv2

class KivyCamera(Image):
    source = ObjectProperty()
    fps = NumericProperty(30)

    def __init__(self, **kwargs):
        super(KivyCamera, self).__init__(**kwargs)
        self._capture = None
        if self.source is not None:
            self._capture = cv2.VideoCapture(self.source)
        Clock.schedule_interval(self.update, 1.0 / self.fps)

    def on_source(self, *args):
        if self._capture is not None:
            self._capture.release()
        self._capture = cv2.VideoCapture(self.source)

    @property
    def capture(self):
        return self._capture

    def update(self, dt):
        ret, frame = self.capture.read()
        if ret:
            buf1 = cv2.flip(frame, 0)
            buf = buf1.tostring()
            image_texture = Texture.create(
                size=(frame.shape[1], frame.shape[0]), colorfmt="bgr"
            )
            image_texture.blit_buffer(buf, colorfmt="bgr", bufferfmt="ubyte")
            self.texture = image_texture


class CamApp(App):
    pass


if __name__ == "__main__":
    CamApp().run()

cam.kv 

BoxLayout:
    orientation: 'horizontal'

    KivyCamera:
        source: "stream address removed for stack overflow"
...