modify video render

This commit is contained in:
jiegeaiai 2024-11-22 00:29:37 +08:00
parent f5e84127cc
commit e3646f4e71

View File

@ -8,7 +8,6 @@ import cv2
import numpy as np import numpy as np
from .base_render import BaseRender from .base_render import BaseRender
from human.message_type import MessageType
class VideoRender(BaseRender): class VideoRender(BaseRender):
@ -31,9 +30,10 @@ class VideoRender(BaseRender):
except: except:
print('resize error') print('resize error')
return return
cv2.imwrite(f'res_frame_{ self.index }.png', res_frame) # cv2.imwrite(f'./images/res_frame_{ self.index }.png', res_frame)
self.index = self.index + 1
combine_frame[y1:y2, x1:x2] = res_frame combine_frame[y1:y2, x1:x2] = res_frame
# cv2.imwrite(f'/combine_frame_{self.index}.png', combine_frame)
# self.index = self.index + 1
image = combine_frame image = combine_frame
# image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)