写一段python实现把文件夹1内的yuv转换成jpg并输出到另一个文件夹2
时间: 2024-04-29 16:26:03 浏览: 146
可以使用Python中的OpenCV库来进行yuv到jpg的转换。以下是一个基本的实现:
```python
import cv2
import os
# 设置文件夹路径
folder_1 = "path/to/folder1/"
folder_2 = "path/to/folder2/"
# 获取文件夹中所有的yuv文件
files = os.listdir(folder_1)
yuv_files = [f for f in files if f.endswith('.yuv')]
# 循环遍历每个yuv文件
for f in yuv_files:
# 读取yuv文件
width = 1920 # 输入yuv的宽度
height = 1080 # 输入yuv的高度
yuv_file = folder_1 + f
with open(yuv_file, 'rb') as f:
yuv = f.read()
# 将yuv数据转换成图像
img = cv2.cvtColor(np.fromstring(yuv, dtype=np.uint8).reshape((height*3//2, width)), cv2.COLOR_YUV2BGR_NV12)
# 将图像保存成jpg文件
jpg_file = folder_2 + f.replace('.yuv', '.jpg')
cv2.imwrite(jpg_file, img)
```
注意,请将以上代码中的“path/to/folder1/”和“path/to/folder2/”替换为实际的文件夹路径!并且,此代码仅能转换YUV4:2:0格式的YUV文件。如果您的YUV文件格式与此不同,需要按照实际情况自己更改代码。
阅读全文