Как добавить изображения в формате jpeg в конвейер сервера gstreamer rtsp, используя python - PullRequest
Новая
       31

Как добавить изображения в формате jpeg в конвейер сервера gstreamer rtsp, используя python

0 голосов
/ 30 апреля 2019

Я столкнулся с проблемой при добавлении изображений opencv в конвейер gst rtsp сервера.Потоковые изображения, но что-то не так с кодеком.Я не могу понять, как создать простой сервер RTSP, используя Python, который будет передавать изображения из каталога.Я следовал: Запись кадров opencv в конвейер сервера gtsreamer rtsp , но не повезло. Пожалуйста, помогите. 

import gi

gi.require_version('Gst', '1.0')
gi.require_version('GstRtspServer', '1.0')
from gi.repository import Gst, GstRtspServer, GObject


# '! videoconvert ! video/x-raw,format=I420 ' \
class SensorFactory(GstRtspServer.RTSPMediaFactory):
    def __init__(self, **properties):
        super(SensorFactory, self).__init__(**properties)
        self.number_frames = 0
        self.fps = 5
        self.duration = 1 / self.fps * Gst.SECOND  # duration of a frame in nanoseconds
        self.launch_string = 'appsrc name=source is-live=true block=true format=GST_FORMAT_TIME ' \
                             'caps=video/x-raw,format=BGR,width=640,height=480,framerate={}/1 ' \
                             '! videoconvert ! video/x-raw,format=I420 ' \
                             '! x264enc speed-preset=ultrafast tune=zerolatency ' \
                             '! rtph264pay config-interval=1 name=pay0 pt=96'.format(self.fps)

        # self.launch_string = 'appsrc name=source videotestsrc ! ximagesink'

    def on_need_data(self, src, lenght):

        cv2.imwrite('{}.jpg'.format(self.number_frames), frame)
        frm = cv2.imread('cover.jpg')
        data = frm.tostring()
        buf = Gst.Buffer.new_allocate(None, len(data), None)
        buf.fill(0, data)
        buf.duration = self.duration
        timestamp = self.number_frames * self.duration
        buf.pts = buf.dts = int(timestamp)
        buf.offset = timestamp
        self.number_frames += 1
        retval = src.emit('push-buffer', buf)
        print('pushed buffer, frame {}, duration {} ns, durations {} s'.format(self.number_frames,
                                                                                self.duration,
                                                                                self.duration / Gst.SECOND))
        if retval != Gst.FlowReturn.OK:
            print(retval)

    def do_create_element(self, url):
        return Gst.parse_launch(self.launch_string)

    def do_configure(self, rtsp_media):
        self.number_frames = 0
        appsrc = rtsp_media.get_element().get_child_by_name('source')
        appsrc.connect('need-data', self.on_need_data)


class GstServer(GstRtspServer.RTSPServer):
    def __init__(self, **properties):
        super(GstServer, self).__init__(**properties)
        self.factory = SensorFactory()
        self.factory.set_shared(True)

        self.props.service = "3000"

        self.get_mount_points().add_factory("/test", self.factory)
        self.attach(None)


GObject.threads_init()
Gst.init(None)

server = GstServer()

loop = GObject.MainLoop()
loop.run()
...