91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
|
import cv2
|
|||
|
|
import numpy as np
|
|||
|
|
import struct
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
import argparse
|
|||
|
|
|
|||
|
|
|
|||
|
|
def BGR888_2_RGB565(img):
|
|||
|
|
'''BGR888 转化为 RGB565
|
|||
|
|
|
|||
|
|
Arguments:
|
|||
|
|
- img -- 待转化的 numpy 数组
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
img_B = img[:, :, 0]
|
|||
|
|
img_G = img[:, :, 1]
|
|||
|
|
img_R = img[:, :, 2]
|
|||
|
|
img_RGB565 = ((img_B & 0xF8) >> 3) | (
|
|||
|
|
(img_G & 0xFC) << 3) | ((img_R & 0xF8) << 8)
|
|||
|
|
return img_RGB565.flatten()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description='生成供 STM32-Player (https://github.com/StopPointTeam/STM32-Player) 播放的视频')
|
|||
|
|
parser.add_argument('-i', '-input', help='输入视频路径')
|
|||
|
|
arg = vars(parser.parse_args())
|
|||
|
|
|
|||
|
|
if os.path.exists(arg['i']) == False:
|
|||
|
|
print('输入视频路径不存在，请检查路径！')
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
cap = cv2.VideoCapture(arg['i'])
|
|||
|
|
if cap.isOpened() == False:
|
|||
|
|
print('打开输入视频失败！')
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
fps = cap.get(cv2.CAP_PROP_FPS)
|
|||
|
|
frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)
|
|||
|
|
frame_width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
|
|||
|
|
frame_height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
|
|||
|
|
|
|||
|
|
print('输入视频分辨率:', frame_width, '*', frame_height)
|
|||
|
|
print('输入视频 FPS:', fps)
|
|||
|
|
print('输入视频总帧数:', round(frame_count))
|
|||
|
|
print('输入视频时长:', round(frame_count / fps, 2), 's')
|
|||
|
|
|
|||
|
|
if fps > 30:
|
|||
|
|
frame_interval = int(fps / (fps - 30))
|
|||
|
|
fps = 30
|
|||
|
|
print('输入视频 FPS 超过 30，将自动转换为 30')
|
|||
|
|
else:
|
|||
|
|
frame_interval = 0
|
|||
|
|
|
|||
|
|
frame_seq = 0 # 输入视频帧序号
|
|||
|
|
new_seq = 0 # 输出视频帧序号
|
|||
|
|
|
|||
|
|
# 保存到二进制文件中
|
|||
|
|
filename = os.path.basename(arg['i']) + '.32v'
|
|||
|
|
file = open(filename, 'wb')
|
|||
|
|
file.seek(153600, os.SEEK_SET)
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
ret, img = cap.read() # 捕获一帧
|
|||
|
|
if ret == True:
|
|||
|
|
frame_seq += 1
|
|||
|
|
if (frame_interval == 0 or frame_seq % frame_interval != 0):
|
|||
|
|
new_seq += 1
|
|||
|
|
|
|||
|
|
# 变换尺寸，并转化为 uint16 类型 numpy 数组
|
|||
|
|
img_resized = np.array(cv2.resize(
|
|||
|
|
img, (320, 240))).astype('uint16')
|
|||
|
|
|
|||
|
|
img_RGB565 = BGR888_2_RGB565(img_resized) # 颜色空间转换
|
|||
|
|
|
|||
|
|
fmt = '>' + len(img_RGB565) * 'H' # 格式化字符串，大端模式
|
|||
|
|
data = struct.pack(fmt, *img_RGB565) # 列表转换为二进制字节流
|
|||
|
|
|
|||
|
|
file.write(data)
|
|||
|
|
print('正在处理：第', frame_seq, '/', round(frame_count), '帧，进度：',
|
|||
|
|
round(frame_seq / frame_count * 100, 2), '%')
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
# 写入视频信息头
|
|||
|
|
file.seek(0, os.SEEK_SET)
|
|||
|
|
file.write(struct.pack('<dI', fps, int(new_seq))) # 小端模式
|
|||
|
|
print('处理完成，输出视频共', new_seq, '帧, FPS 为', fps)
|
|||
|
|
break
|