human/render/video_render.py

#encoding = utf8
import copy
import logging
import time

import cv2
import numpy as np

from .base_render import BaseRender


class VideoRender(BaseRender):
    def __init__(self, play_clock, context, human_render):
        super().__init__(play_clock, context, 'Video')
        self._human_render = human_render
        self.index = 0

    def render(self, frame, ps):
        if self._human_render is not None:
            self._human_render.put_image(frame)

        # image = cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA)