gitadmin
/
tuoheng_alg_move_20240715
Следить 1
В избранное 0
Форкнуть 0
Код Задачи 0 Pull Request'ы 0 Релизы 11 Вики Активность
Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.
61 коммит
8 Ветки
Дерево: 80557c5e0b
Ветки Теги
${ item.name }
Создать ветку ${ searchTerm }
из '80557c5e0b'
${ noResults }
tuoheng_alg_move_20240715/util/Cv2Utils.py

529 lines
25KB
Исходник Blame История 

  1. # -*- coding: utf-8 -*-

  2. import json

  3. import time

  4. 

  5. import cv2

  6. import subprocess as sp

  7. 

  8. import numpy as np

  9. from loguru import logger

  10. from exception.CustomerException import ServiceException

  11. from enums.ExceptionEnum import ExceptionType

  12. 

  13. 

  14. class Cv2Util():

  15. 

  16.     def __init__(self, pullUrl, pushUrl=None, orFilePath=None, aiFilePath=None, requestId=None):

  17.         self.pullUrl = pullUrl

  18.         self.pushUrl = pushUrl

  19.         self.orFilePath = orFilePath

  20.         self.aiFilePath = aiFilePath

  21.         self.cap = None

  22.         self.p = None

  23.         self.or_video_file = None

  24.         self.ai_video_file = None

  25.         self.fps = None

  26.         self.width = None

  27.         self.height = None

  28.         self.wah = None

  29.         self.wh = None

  30.         self.h = None

  31.         self.hn = None

  32.         self.w = None

  33.         self.all_frames = None

  34.         self.bit_rate = None

  35.         self.pull_p = None

  36.         self.requestId = requestId

  37.         self.p_push_retry_num = 0

  38.         self.resize_status = False

  39.         self.current_frame = 0

  40. 

  41.     def getFrameConfig(self, fps, width, height):

  42.         if self.fps is None:

  43.             self.fps = fps

  44.             self.width = width

  45.             self.height = height

  46.             if width > 1600:

  47.                 self.wh = int(width * height * 3 // 8)

  48.                 self.wah = '%sx%s' % (int(self.width / 2), int(self.height / 2))

  49.                 self.h = int(self.height * 3 // 4)

  50.                 self.w = int(self.width // 2)

  51.                 self.hn = int(self.height // 2)

  52.                 self.wn = int(self.width // 2)

  53.                 w_f = self.wh != width * height * 3 / 8

  54.                 h_f = self.h != self.height * 3 / 4

  55.                 wd_f = self.w != self.width / 2

  56.                 if w_f or h_f or wd_f:

  57.                     self.resize_status = True

  58.                     self.wh = int(width * height * 3 // 2)

  59.                     self.wah = '%sx%s' % (int(self.width), int(self.height))

  60.                     self.h = int(self.height * 3 // 2)

  61.                     self.w = int(self.width)

  62.             else:

  63.                 self.wh = int(width * height * 3 // 2)

  64.                 self.wah = '%sx%s' % (int(self.width), int(self.height))

  65.                 self.h = int(self.height * 3 // 2)

  66.                 self.w = int(self.width)

  67.                 self.hn = int(self.height)

  68.                 self.wn = int(self.width)

  69. 

  70.     def clear_video_info(self):

  71.         self.fps = None

  72.         self.width = None

  73.         self.height = None

  74. 

  75. 

  76.     '''

  77.         获取视频信息

  78.     '''

  79. 

  80.     def get_video_info(self):

  81.         try:

  82.             if self.pullUrl is None:

  83.                 logger.error("拉流地址不能为空, requestId:{}", self.requestId)

  84.                 raise ServiceException(ExceptionType.PULL_STREAM_URL_EXCEPTION.value[0],

  85.                                        ExceptionType.PULL_STREAM_URL_EXCEPTION.value[1])

  86.             args = ['ffprobe', '-show_format', '-show_streams', '-of', 'json', self.pullUrl]

  87.             p = sp.Popen(args, stdout=sp.PIPE, stderr=sp.PIPE)

  88.             out, err = p.communicate(timeout=20)

  89.             if p.returncode != 0:

  90.                 raise Exception("未获取视频信息！！！！！requestId:" + self.requestId)

  91.             probe = json.loads(out.decode('utf-8'))

  92.             if probe is None or probe.get("streams") is None:

  93.                 raise Exception("未获取视频信息！！！！！requestId:" + self.requestId)

  94.             # 视频大小

  95.             # format = probe['format']

  96.             # size = int(format['size'])/1024/1024

  97.             video_stream = next((stream for stream in probe['streams'] if stream.get('codec_type') == 'video'), None)

  98.             if video_stream is None:

  99.                 raise Exception("未获取视频信息！！！！！requestId:" + self.requestId)

  100.             width = video_stream.get('width')

  101.             height = video_stream.get('height')

  102.             nb_frames = video_stream.get('nb_frames')

  103.             fps = video_stream.get('r_frame_rate')

  104.             # duration = video_stream.get('duration')

  105.             # bit_rate = video_stream.get('bit_rate')

  106.             self.width = int(width)

  107.             self.height = int(height)

  108.             if width > 1600:

  109.                 self.wh = int(width * height * 3 // 8)

  110.                 self.wah = '%sx%s' % (int(self.width / 2), int(self.height / 2))

  111.                 self.h = int(self.height * 3 // 4)

  112.                 self.w = int(self.width / 2)

  113.                 self.hn = int(self.height / 2)

  114.                 self.wn = int(self.width // 2)

  115.                 w_f = self.wh != width * height * 3 / 8

  116.                 h_f = self.h != self.height * 3 / 4

  117.                 wd_f = self.w != self.width / 2

  118.                 if w_f or h_f or wd_f:

  119.                     self.resize_status = True

  120.                     self.wh = int(width * height * 3 // 2)

  121.                     self.wah = '%sx%s' % (int(self.width), int(self.height))

  122.                     self.h = int(self.height * 3 // 2)

  123.                     self.w = int(self.width)

  124.             else:

  125.                 self.wh = int(width * height * 3 // 2)

  126.                 self.wah = '%sx%s' % (int(self.width), int(self.height))

  127.                 self.h = int(self.height * 3 // 2)

  128.                 self.w = int(self.width)

  129.                 self.hn = int(self.height)

  130.                 self.wn = int(self.width)

  131.             if nb_frames:

  132.                 self.all_frames = int(nb_frames)

  133.             up, down = str(fps).split('/')

  134.             self.fps = int(eval(up) / eval(down))

  135.             # if duration:

  136.             #    self.duration = float(video_stream['duration'])

  137.             # self.bit_rate = int(bit_rate) / 1000

  138.             logger.info("视频信息, width:{}|height:{}|fps:{}|all_frames:{}|bit_rate:{}, requestId:{}", self.width,

  139.                         self.height, self.fps, self.all_frames, self.bit_rate, self.requestId)

  140.         except ServiceException as s:

  141.             logger.error("获取视频信息异常: {}, requestId:{}", s.msg, self.requestId)

  142.             self.clear_video_info()

  143.             raise s

  144.         except Exception as e:

  145.             logger.exception("获取视频信息异常：{}, requestId:{}", e, self.requestId)

  146.             self.clear_video_info()

  147. 

  148.     '''

  149.         拉取视频

  150.     '''

  151. 

  152.     def build_pull_p(self):

  153.         try:

  154.             if self.wah is None:

  155.                 return

  156.             if self.pull_p:

  157.                 logger.info("重试, 关闭拉流管道, requestId:{}", self.requestId)

  158.                 self.pull_p.stdout.close()

  159.                 self.pull_p.terminate()

  160.                 self.pull_p.wait()

  161.             # command = ['ffmpeg',

  162.             #            # '-b:v', '3000k',

  163.             #            '-i', self.pullUrl,

  164.             #            '-f', 'rawvideo',

  165.             #            '-vcodec', 'rawvideo',

  166.             #            '-pix_fmt', 'bgr24',

  167.             #            # '-s', "{}x{}".format(int(width), int(height)),

  168.             #            '-an',

  169.             #            '-']

  170.             # input_config = {'c:v': 'h264_cuvid', 'resize': self.wah}

  171.             # process = (

  172.             #     ffmpeg

  173.             #         .input(self.pullUrl, **input_config)

  174.             #         .output('pipe:', format='rawvideo', r=str(self.fps)) # pix_fmt='bgr24'

  175.             #         .overwrite_output()

  176.             #         .global_args('-an')

  177.             #         .run_async(pipe_stdout=True)

  178.             # )

  179.             command = ['ffmpeg',

  180.                        '-re',

  181.                        '-y',

  182.                        '-c:v', 'h264_cuvid',

  183.                        '-resize', self.wah,

  184.                        '-i', self.pullUrl,

  185.                        '-f', 'rawvideo',

  186.                        '-an',

  187.                        '-']

  188.             self.pull_p = sp.Popen(command, stdout=sp.PIPE)

  189.             # self.pull_p = sp.Popen(command, stdout=sp.PIPE, stderr=sp.PIPE)

  190.             # self.pull_p = process

  191.         except ServiceException as s:

  192.             logger.exception("构建拉流管道异常: {}, requestId:{}", s, self.requestId)

  193.             if self.pull_p:

  194.                 logger.info("重试, 关闭拉流管道, requestId:{}", self.requestId)

  195.                 self.pull_p.stdout.close()

  196.                 self.pull_p.terminate()

  197.                 self.pull_p.wait()

  198.                 self.pull_p = None

  199.             raise s

  200.         except Exception as e:

  201.             logger.exception("构建拉流管道异常：{}, requestId:{}", e, self.requestId)

  202.             if self.pull_p:

  203.                 logger.info("重试, 关闭拉流管道, requestId:{}", self.requestId)

  204.                 self.pull_p.stdout.close()

  205.                 self.pull_p.terminate()

  206.                 self.pull_p.wait()

  207.                 self.pull_p = None

  208. 

  209.     def checkconfig(self):

  210.         if self.fps is None or self.width is None or self.height is None:

  211.             return True

  212.         return False

  213. 

  214.     def read(self):

  215.         result = None

  216.         try:

  217.             # if self.pull_p is None:

  218.             #     logger.error("拉流管道为空, requestId:{}", self.requestId)

  219.             #     raise ServiceException(ExceptionType.PULL_PIPELINE_INIT_EXCEPTION.value[0],

  220.             #                            ExceptionType.PULL_PIPELINE_INIT_EXCEPTION.value[1])

  221.             in_bytes = self.pull_p.stdout.read(self.wh)

  222.             self.current_frame += 1

  223.             if in_bytes is not None and len(in_bytes) > 0:

  224.                 # result = (np.frombuffer(in_bytes, np.uint8).reshape([int(self.height), int(self.width), 3]))

  225.                 try:

  226.                     img = (np.frombuffer(in_bytes, np.uint8)).reshape((self.h, self.w))

  227.                 except Exception as ei:

  228.                     logger.exception("视频格式异常：{}, requestId:{}", ei, self.requestId)

  229.                     raise ServiceException(ExceptionType.VIDEO_RESOLUTION_EXCEPTION.value[0],

  230.                                            ExceptionType.VIDEO_RESOLUTION_EXCEPTION.value[1])

  231.                 result = cv2.cvtColor(img, cv2.COLOR_YUV2BGR_NV12)

  232.                 if self.resize_status:

  233.                     if self.width > 1600:

  234.                         result = cv2.resize(result, (int(self.width / 2), int(self.height / 2)),

  235.                                             interpolation=cv2.INTER_LINEAR)

  236.         except ServiceException as s:

  237.             raise s

  238.         except Exception as e:

  239.             logger.exception("读流异常：{}, requestId:{}", e, self.requestId)

  240.         return result

  241. 

  242.     def close(self):

  243.         if self.pull_p:

  244.             if self.pull_p.stdout:

  245.                 self.pull_p.stdout.close()

  246.             self.pull_p.terminate()

  247.             self.pull_p.wait()

  248.             logger.info("关闭拉流管道完成, requestId:{}", self.requestId)

  249.         if self.p:

  250.             if self.p.stdin:

  251.                 self.p.stdin.close()

  252.             self.p.terminate()

  253.             self.p.wait()

  254.             # self.p.communicate()

  255.             # self.p.kill()

  256.             logger.info("关闭管道完成, requestId:{}", self.requestId)

  257.         if self.or_video_file:

  258.             self.or_video_file.release()

  259.             logger.info("关闭原视频写入流完成, requestId:{}", self.requestId)

  260.         if self.ai_video_file:

  261.             self.ai_video_file.release()

  262.             logger.info("关闭AI视频写入流完成, requestId:{}", self.requestId)

  263. 

  264.     # 构建 cv2

  265.     # def build_cv2(self):

  266.     #     try:

  267.     #         if self.cap is not None:

  268.     #             logger.info("重试, 关闭cap, requestId:{}", self.requestId)

  269.     #             self.cap.release()

  270.     #         if self.p is not None:

  271.     #             logger.info("重试, 关闭管道, requestId:{}", self.requestId)

  272.     #             self.p.stdin.close()

  273.     #             self.p.terminate()

  274.     #             self.p.wait()

  275.     #         if self.pullUrl is None:

  276.     #             logger.error("拉流地址不能为空, requestId:{}", self.requestId)

  277.     #             raise ServiceException(ExceptionType.PULL_STREAM_URL_EXCEPTION.value[0],

  278.     #                                    ExceptionType.PULL_STREAM_URL_EXCEPTION.value[1])

  279.     #         if self.pushUrl is None:

  280.     #             logger.error("推流地址不能为空, requestId:{}", self.requestId)

  281.     #             raise ServiceException(ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[0],

  282.     #                                    ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[1])

  283.     #         self.cap = cv2.VideoCapture(self.pullUrl)

  284.     #         if self.fps is None or self.fps == 0:

  285.     #             self.fps = int(self.cap.get(cv2.CAP_PROP_FPS))

  286.     #         if self.width is None or self.width == 0:

  287.     #             self.width = int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH))

  288.     #         if self.height is None or self.height == 0:

  289.     #             self.height = int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

  290.     #         command = ['/usr/bin/ffmpeg',

  291.     #                    # '-y',  # 不经过确认，输出时直接覆盖同名文件。

  292.     #                    '-f', 'rawvideo',

  293.     #                    '-vcodec', 'rawvideo',

  294.     #                    '-pix_fmt', 'bgr24',  # 显示可用的像素格式

  295.     #                    # '-s', "{}x{}".format(self.width * 2, self.height),

  296.     #                    '-s', "{}x{}".format(int(self.width), int(self.height/2)),

  297.     #                    # '-r', str(15),

  298.     #                    '-i', '-',  # 指定输入文件

  299.     #                    '-g', '25',

  300.     #                    '-b:v', '3000k',

  301.     #                    '-tune', 'zerolatency',  # 加速编码速度

  302.     #                    '-c:v', 'libx264',  # 指定视频编码器

  303.     #                    '-sc_threshold', '0',

  304.     #                    '-pix_fmt', 'yuv420p',

  305.     #                    '-an',

  306.     #                    '-preset', 'ultrafast',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  307.     #                    # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。

  308.     #                    '-f', 'flv',

  309.     #                    self.pushUrl]

  310.     #         # 管道配置

  311.     #         logger.info("fps:{}|height:{}|width:{}|requestId:{}", self.fps, self.height, self.width, self.requestId)

  312.     #         self.p = sp.Popen(command, stdin=sp.PIPE)

  313.     #     except ServiceException as s:

  314.     #         logger.exception("构建cv2异常: {}, requestId:{}", s, self.requestId)

  315.     #         raise s

  316.     #     except Exception as e:

  317.     #         logger.exception("初始化cv2异常：{}, requestId:{}", e, self.requestId)

  318. 

  319.     # 构建 cv2

  320.     def build_p(self):

  321.         try:

  322.             if self.p:

  323.                 logger.info("重试, 关闭管道, requestId:{}", self.requestId)

  324.                 if self.p.stdin:

  325.                     self.p.stdin.close()

  326.                 self.p.terminate()

  327.                 self.p.wait()

  328.                 # self.p.communicate()

  329.                 # self.p.kill()

  330.             if self.pushUrl is None:

  331.                 logger.error("推流地址不能为空, requestId:{}", self.requestId)

  332.                 raise ServiceException(ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[0],

  333.                                        ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[1])

  334.             width = int(self.width)

  335.             if width <= 1600:

  336.                 width = 2 * int(self.width)

  337.             command = ['ffmpeg',

  338.                        # '-loglevel', 'debug',

  339.                        '-y',

  340.                        '-f', 'rawvideo',

  341.                        '-vcodec', 'rawvideo',

  342.                        '-pix_fmt', 'bgr24',

  343.                        '-thread_queue_size', '1024',

  344.                        # '-s', "{}x{}".format(self.width * 2, self.height),

  345.                        '-s', "{}x{}".format(width, int(self.hn)),

  346.                        '-r', str(self.fps),

  347.                        '-i', '-',  # 指定输入文件

  348.                        '-g', str(self.fps),

  349.                        '-maxrate', '8000k',

  350.                        # '-profile:v', 'high',

  351.                        '-b:v', '5000k',

  352.                        # '-crf', '18',

  353.                        # '-rc:v', 'vbr',

  354.                        # '-cq:v', '25',

  355.                        # '-qmin', '25',

  356.                        # '-qmax', '25',

  357.                        '-c:v', 'h264_nvenc',  #

  358.                        '-bufsize', '5000k',

  359.                        # '-c:v', 'libx264',  # 指定视频编码器

  360.                        # '-tune', 'zerolatency',  # 加速编码速度

  361.                        # '-sc_threshold', '0',

  362.                        '-pix_fmt', 'yuv420p',

  363.                        "-an",

  364.                        # '-flvflags', 'no_duration_filesize',

  365.                        # '-preset', 'fast',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  366.                        '-preset', 'p6',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  367.                        '-tune', 'll',

  368.                        '-f', 'flv',

  369.                        self.pushUrl]

  370.             # command = 'ffmpeg -loglevel debug -y -f rawvideo -vcodec rawvideo -pix_fmt bgr24' +\

  371.             #            ' -s ' +  "{}x{}".format(int(self.width), int(self.height/2))\

  372.             #            + ' -i - ' + '-g ' + str(self.fps)+\

  373.             #            ' -b:v 6000k -tune zerolatency -c:v libx264 -pix_fmt yuv420p -preset ultrafast'+\

  374.             #            ' -f flv ' + self.pushUrl

  375. 

  376.             # kwargs = {'format': 'rawvideo',

  377.             #           # 'vcodec': 'rawvideo',

  378.             #           'pix_fmt': 'bgr24',

  379.             #           's': '{}x{}'.format(int(self.width), int(self.height/2))}

  380.             # out = {

  381.             #     'r': str(self.fps),

  382.             #     'g': str(self.fps),

  383.             #     'b:v': '5500k', # 恒定码率

  384.             #     # 'maxrate': '15000k',

  385.             #     # 'crf': '18',

  386.             #     'bufsize': '5500k',

  387.             #     'tune': 'zerolatency', # 加速编码速度

  388.             #     'c:v': 'libx264',  # 指定视频编码器

  389.             #     'sc_threshold': '0',

  390.             #     'pix_fmt': 'yuv420p',

  391.             #     # 'flvflags': 'no_duration_filesize',

  392.             #     'preset': 'medium',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  393.             #     # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。

  394.             #     'format': 'flv'}

  395.             # 管道配置

  396.             # process2 = (

  397.             #     ffmpeg

  398.             #         .input('pipe:', **kwargs)

  399.             #         .output(self.pushUrl, **out)

  400.             #         .global_args('-y', '-an')

  401.             #         .overwrite_output()

  402.             #         .run_async(pipe_stdin=True)

  403.             # )

  404.             logger.info("fps:{}|height:{}|width:{}|requestId:{}", self.fps, self.height, self.width, self.requestId)

  405.             self.p = sp.Popen(command, stdin=sp.PIPE, shell=False)

  406.             # self.p = process2

  407.         except ServiceException as s:

  408.             logger.exception("构建p管道异常: {}, requestId:{}", s, self.requestId)

  409.             raise s

  410.         except Exception as e:

  411.             logger.exception("初始化p管道异常：{}, requestId:{}", e, self.requestId)

  412. 

  413.     async def push_stream_write(self, frame):

  414.         self.p.stdin.write(frame.tostring())

  415. 

  416.     async def push_stream(self, frame):

  417.         if self.p is None:

  418.             self.build_p()

  419.         try:

  420.             await self.push_stream_write(frame)

  421.             return True

  422.         except Exception as ex:

  423.             logger.exception("推流进管道异常：{}, requestId: {}", ex, self.requestId)

  424.             current_retry_num = 0

  425.             while True:

  426.                 try:

  427.                     time.sleep(1)

  428.                     self.p_push_retry_num += 1

  429.                     current_retry_num += 1

  430.                     if current_retry_num > 3 or self.p_push_retry_num > 600:

  431.                         return False

  432.                     self.build_p()

  433.                     await self.push_stream_write(frame)

  434.                     logger.info("构建p管道重试成功, 当前重试次数: {}, requestId: {}", current_retry_num,

  435.                                 self.requestId)

  436.                     return True

  437.                 except Exception as e:

  438.                     logger.exception("构建p管道异常：{}, 开始重试, 当前重试次数：{}, requestId: {}", e,

  439.                                      current_retry_num, self.requestId)

  440.         return False

  441. 

  442.     async def video_frame_write(self, or_frame, ai_frame):

  443.         if or_frame is not None:

  444.             self.or_video_file.write(or_frame)

  445.         if ai_frame is not None:

  446.             self.ai_video_file.write(ai_frame)

  447. 

  448.     async def video_write(self, or_frame, ai_frame):

  449.         try:

  450.             self.build_write()

  451.             if or_frame is not None and len(or_frame) > 0:

  452.                 await self.video_frame_write(or_frame, None)

  453.             if ai_frame is not None and len(ai_frame) > 0:

  454.                 await self.video_frame_write(None, ai_frame)

  455.             return True

  456.         except Exception as ex:

  457.             ai_retry_num = 0

  458.             while True:

  459.                 try:

  460.                     ai_retry_num += 1

  461.                     if ai_retry_num > 3:

  462.                         logger.exception("重新写入离线分析后视频到本地,重试失败：{}, requestId: {}", e, self.requestId)

  463.                         return False

  464.                     if or_frame is not None and len(or_frame) > 0:

  465.                         await self.or_video_file.write(or_frame)

  466.                     if ai_frame is not None and len(ai_frame) > 0:

  467.                         await self.ai_video_file.write(ai_frame)

  468.                     logger.info("重新写入离线分析后视频到本地, 当前重试次数: {}, requestId: {}", ai_retry_num,

  469.                                 self.requestId)

  470.                     return True

  471.                 except Exception as e:

  472.                     logger.exception("重新写入离线分析后视频到本地：{}, 开始重试, 当前重试次数：{}, requestId: {}", e,

  473.                                      ai_retry_num, self.requestId)

  474. 

  475.     def build_write(self):

  476.         try:

  477.             if self.fps is None or self.width is None or self.height is None:

  478.                 logger.error("fps、 width、 height为空, requestId:{}", self.requestId)

  479.                 raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],

  480.                                        ExceptionType.SERVICE_INNER_EXCEPTION.value[1])

  481.             if self.orFilePath is not None and self.or_video_file is None:

  482.                 self.or_video_file = cv2.VideoWriter(self.orFilePath, cv2.VideoWriter_fourcc(*'mp4v'), self.fps,

  483.                                                      (int(self.wn), int(self.hn)))

  484.                 if self.or_video_file is None:

  485.                     logger.error("or_video_file为空, requestId:{}", self.requestId)

  486.                     raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],

  487.                                            ExceptionType.SERVICE_INNER_EXCEPTION.value[1])

  488.             if self.aiFilePath is not None and self.ai_video_file is None:

  489.                 self.ai_video_file = cv2.VideoWriter(self.aiFilePath, cv2.VideoWriter_fourcc(*'mp4v'), self.fps,

  490.                                                      (int(self.wn * 2), int(self.hn)))

  491.                 if self.ai_video_file is None:

  492.                     logger.error("ai_video_file为空, requestId:{}", self.requestId)

  493.                     raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],

  494.                                            ExceptionType.SERVICE_INNER_EXCEPTION.value[1])

  495.         except ServiceException as s:

  496.             logger.exception("构建文件写对象异常: {}, requestId:{}", s, self.requestId)

  497.             raise s

  498.         except Exception as e:

  499.             logger.exception("构建文件写对象异常: {}, requestId:{}", e, self.requestId)

  500.             raise e

  501. 

  502.     def video_merge(self, frame1, frame2):

  503.         # frameLeft = cv2.resize(frame1, (int(self.width / 2), int(self.height / 2)), interpolation=cv2.INTER_LINEAR)

  504.         # frameRight = cv2.resize(frame2, (int(self.width / 2), int(self.height / 2)), interpolation=cv2.INTER_LINEAR)

  505.         # frame_merge = np.hstack((frameLeft, frameRight))

  506.         frame_merge = np.hstack((frame1, frame2))

  507.         return frame_merge

  508. 

  509.     def getP(self):

  510.         if self.p is None:

  511.             logger.error("获取管道为空, requestId:{}", self.requestId)

  512.             raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],

  513.                                    ExceptionType.SERVICE_INNER_EXCEPTION.value[1])

  514.         return self.p

  515. 

  516.     def getOrVideoFile(self):

  517.         if self.or_video_file is None:

  518.             logger.error("获取原视频写入对象为空, requestId:{}", self.requestId)

  519.             raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],

  520.                                    ExceptionType.SERVICE_INNER_EXCEPTION.value[1])

  521.         return self.or_video_file

  522. 

  523.     def getAiVideoFile(self):

  524.         if self.ai_video_file is None:

  525.             logger.error("获取AI视频写入对象为空, requestId:{}", self.requestId)

  526.             raise ServiceException(ExceptionType.SERVICE_INNER_EXCEPTION.value[0],

  527.                                    ExceptionType.SERVICE_INNER_EXCEPTION.value[1])

  528.         return self.ai_video_file