如何将AVFrame转换为PNG？转换后图像呈灰色问题求助

核心问题分析 #

你遇到的灰色图像问题，本质是AVFrame的像素格式与libpng要求的格式不匹配。MP4视频的帧通常是YUV格式（比如最常见的YUV420P），而你直接把Y分量（frame->data[0]）当成RGB数据写入PNG，自然只会显示灰度。

具体解决步骤 #

1. 确认当前AVFrame的像素格式

先在代码中添加日志，打印帧的像素格式：

printf("Frame pixel format: %s\n", av_get_pix_fmt_name(frame->format));

如果输出是yuv420p这类YUV格式，就必须做格式转换。

2. 使用Libav的sws_scale做像素格式转换

需要先初始化一个SwsContext，将YUV帧转换为RGB24（PNG支持的格式）：

// 初始化转换上下文
struct SwsContext *sws_ctx = sws_getContext(
    frame->width, frame->height, frame->format,
    frame->width, frame->height, AV_PIX_FMT_RGB24,
    SWS_BILINEAR, NULL, NULL, NULL
);

// 创建转换后的RGB帧
AVFrame *rgb_frame = av_frame_alloc();
rgb_frame->format = AV_PIX_FMT_RGB24;
rgb_frame->width = frame->width;
rgb_frame->height = frame->height;
av_frame_get_buffer(rgb_frame, 0);

// 执行格式转换
sws_scale(sws_ctx, frame->data, frame->linesize, 0, frame->height,
          rgb_frame->data, rgb_frame->linesize);

3. 修正libpng的写入逻辑

转换后的RGB帧，data[0]是连续的RGB字节数据，同时要确保libpng的颜色类型设置正确：

// 设置PNG颜色类型为RGB
png_set_color_type(png_ptr, PNG_COLOR_TYPE_RGB);
png_set_bit_depth(png_ptr, 8);

// 修正row_pointers，使用转换后的rgb_frame
png_bytep *row_pointers = (png_bytep *)malloc(sizeof(png_bytep) * rgb_frame->height);
for (int y = 0; y < rgb_frame->height; y++) {
    row_pointers[y] = rgb_frame->data[0] + y * rgb_frame->linesize[0];
}

4. 资源释放

记得在使用完后释放转换上下文和RGB帧：

sws_freeContext(sws_ctx);
av_frame_free(&rgb_frame);
free(row_pointers);

额外检查点 #

• 确认libpng初始化时的颜色类型没有错误，比如不要设置成PNG_COLOR_TYPE_GRAY。
• 检查linesize的对齐问题：Libav的linesize可能会因为内存对齐比实际宽度的字节数大，直接使用rgb_frame->linesize[0]是正确的，不要手动计算宽度×3。

内容的提问来源于stack exchange，提问作者xtingray