python将YUV420P文件转PNG图片格式的两种方法

方法一：

import os
import cv2 as cv
import numpy as np


# 读取yuv420p的一帧文件，并转化为png图片
if __name__ == '__main__':
  filepath = 'one_frame_of_highway.yuv'
  binfile = open(filepath, 'rb')
  size = os.path.getsize(filepath)
  image_width = 352
  image_hight = 288
  image_y = [[0] * image_width for i in range(image_hight)]
  image_u = [[0] * image_width for i in range(image_hight)]
  image_v = [[0] * image_width for i in range(image_hight)]
  for r in range(image_hight):
    for c in range(image_width):
      image_y[r][c] = binfile.read(1)[0]
  image_y = np.array(image_y)

  for r in range(int(image_hight / 2)):
    for c in range(int(image_width / 2)):
      pixel = binfile.read(1)[0]
      image_u[2 * r + 0][2 * c + 0] = pixel
      image_u[2 * r + 1][2 * c + 0] = pixel
      image_u[2 * r + 0][2 * c + 1] = pixel
      image_u[2 * r + 1][2 * c + 1] = pixel
  image_u = np.array(image_u)

  for r in range(int(image_hight / 2)):
    for c in range(int(image_width / 2)):
      pixel = binfile.read(1)[0]
      image_v[2 * r + 0][2 * c + 0] = pixel
      image_v[2 * r + 0][2 * c + 1] = pixel
      image_v[2 * r + 1][2 * c + 0] = pixel
      image_v[2 * r + 1][2 * c + 1] = pixel
  image_v = np.array(image_v)
  binfile.close()
  compose = np.array([image_y, image_v, image_u]).transpose([1, 2, 0]).astype(np.uint8)
  image = cv.cvtcolor(compose, cv.color_yuv2rgb)
  cv.imwrite("one_frame_of_highway.yuv.png", image)

方法二：

 ffmpeg -s 352x288 -i one_frame_of_highway.yuv one_frame_of_highway.png

highway视频网址：

附录：

将yuv文件转化为一帧帧yuv文件

#include <stdio.h>
#include <fcntl.h>
#include <zconf.h>
#include <stdint.h>
#include <strings.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
int file_size(int fd) {
  struct stat st;
  fstat(fd, &st);
  return st.st_size;
}

int frame_size_of_cif() {
  int width = 352;
  int heigh = 288;
  int y_size = width * heigh;
  int u_size = y_size / 4;
  int v_size = y_size / 4;
  int frame_size = y_size + u_size + v_size;
  return frame_size;
}

int frames_of_cif_file(int fd) {
  if (fd < 0) {
    printf("invalid fd!");
    return -1;
  }
  int frame_size = frame_size_of_cif();
  int fd_size = file_size(fd);
  return fd_size / frame_size;
}

void abstract_frame_from_cif_file(int fd,char *path_and_prefix_img,int len) {
  int frame_size = frame_size_of_cif();
  char file[128];
  memset(file,0,128);
  memcpy(file,path_and_prefix_img,len);
  uint8_t buf[frame_size];
  int ret = -1;
  int frames = 0;
  while ((ret = read(fd, buf, frame_size))) {
    frames += 1;
    uint64_t len = strlen(file);
    sprintf(file + len, "%d", frames);
    len = strlen(file);
    sprintf(file + len, "%s", ".yuv");
    int fdw = open(file, o_rdwr | o_creat, 0777);
    write(fdw, buf, ret);
    memset(file,0,128);
    memcpy(file,path_and_prefix_img,len);
    close(fdw);
  }
  printf("abstract %d frames!\n", frames);
}


int main() {

  int fd = open("./yuv420p_352x288.yuv", o_rdonly);
  abstract_frame_from_cif_file(fd,"/home/liu/frames/frames_",strlen("/home/liu/frames/frames_"));
  close(fd);
  return 0;
}

以上就是python将yuv420p文件转png图片格式的两种方法的详细内容，更多关于python将yuv420p文件转png的资料请关注其它相关文章！