当前位置:网站首页>yuv420sp转jpg
yuv420sp转jpg
2022-08-04 08:34:00 【鲁破特之泪】
代码功能:将yuv420sp(NV12)数据转换成jpg图片
需要注意:文件名需包含图片宽高信息,如 ‘00001_1920_1080.yuv420sp’
import os
import cv2
import numpy as np
from tqdm import tqdm
def yuv420sp2bgr(path, w, h):
yuv = np.zeros((h, w, 3), dtype=np.uint8)
u = np.zeros((int(h / 2), int(w / 2)), dtype=np.uint8)
v = np.zeros((int(h / 2), int(w / 2)), dtype=np.uint8)
with open(path, 'rb') as f:
for row in range(h):
for col in range(w):
yuv[row, col, 0] = int(f.read(1).hex(), 16)
for row in range(int(h / 2)):
for col in range(int(w / 2)):
u[row, col] = int(f.read(1).hex(), 16)
v[row, col] = int(f.read(1).hex(), 16)
yuv[:, :, 1] = cv2.resize(u, (w, h))
yuv[:, :, 2] = cv2.resize(v, (w, h))
bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV2BGR)
return bgr
# yuv420sp文件名需包含图片宽高信息,如 '00001_1920_1080.yuv420sp'
# yuv420sp文件是yuv420图像存储格式的数据位: yyyyyy...uvuv..
path_yuv420sp = r'/media/fang/3441-913C/tsr/img' # yuv420sp文件路径
path_output = path_yuv420sp # jpg图片输出路径,不要有中文
for root, dirs, files in os.walk(path_yuv420sp):
for file in tqdm(files):
if file.split(".")[-1] == 'yuv420sp':
w, h = int(file.split(".")[0].split('_')[-2]), int(file.split(".")[0].split('_')[-1])
bgr = yuv420sp2bgr(root + '/' + file, w, h)
cv2.imwrite(path_output + '/' + file.replace('yuv420sp', 'jpg'), bgr)
边栏推荐
猜你喜欢
随机推荐
通过GBase 8c Platform安装数据库集群时报错
How to write patents are more likely to pass?
字符流与字节流的区别
Typora颜色公式代码大全
推荐几种可以直接翻译PDF英文文献的方法
经典递归回溯问题之——解数独(LeetCode 37)
发现WRH几个表被锁了,怎么办?
Thread类的基本使用。
使用单调栈解决接雨水问题——LeetCode 42 接雨水+单调栈说明
【UE虚幻引擎】UE5实现动态导航样条线绘制
25.时间序列预测实战
使用requests post请求爬取申万一级行业指数行情
研究性学习专题 3_LL(1)语法分析设计原理与实现
ShuffleNet v2 network structure reproduction (Pytorch version)
金仓数据库KingbaseES客户端编程接口指南-JDBC(8. JDBC 元数据处理)
布局管理器
金仓数据库 KDTS 迁移工具使用指南 (5. SHELL版使用说明)
最近的一些杂感-20220731
spark算子讲解
【STM32】STM32F103系列名称与封装、内存