// 函数:将视频转换为图片序列 ,只对mp4做了测试,其他格式未测试。
void videoToImages(const char* videoPath, const char* outputFolder) {
avformat_network_init();
// 打开视频文件
AVFormatContext* formatContext = nullptr;
if (avformat_open_input(&formatContext, videoPath, nullptr, nullptr) != 0) {
// 处理打开视频失败的情况
return;
}
// 获取视频流信息
if (avformat_find_stream_info(formatContext, nullptr) < 0) {
// 处理获取视频流信息失败的情况
avformat_close_input(&formatContext);
return;
}
// 寻找视频流
int videoStreamIndex = -1;
for (unsigned int i = 0; i < formatContext->nb_streams; ++i) {
if (formatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
videoStreamIndex = i;
break;
}
}
if (videoStreamIndex == -1) {
// 处理未找到视频流的情况
avformat_close_input(&formatContext);
return;
}
// 获取视频流解码器参数
AVCodecParameters* codecParameters = formatContext->streams[videoStreamIndex]->codecpar;
const AVCodec* codec = avcodec_find_decoder(codecParameters->codec_id);
AVCodecContext* codecContext = avcodec_alloc_context3(codec);
avcodec_parameters_to_context(codecContext, codecParameters);
avcodec_open2(codecContext, codec, nullptr);
// 创建视频帧
AVFrame* frame = av_frame_alloc();
// 读取视频帧并保存为图片
int frameIndex = 1;
AVPacket packet;
while (av_read_frame(formatContext, &packet) >= 0) {
if (packet.stream_index == videoStreamIndex) {
avcodec_send_packet(codecContext, &packet);
avcodec_receive_frame(codecContext, frame);
// 保存图片
char filename[100];
sprintf(filename, "%s/%d.jpg", outputFolder, frameIndex);
const AVCodec* jpgCodec = avcodec_find_encoder(AV_CODEC_ID_MJPEG);
AVCodecContext* jpgCodecContext = avcodec_alloc_context3(jpgCodec);
jpgCodecContext->pix_fmt = AV_PIX_FMT_YUVJ420P;
jpgCodecContext->width = frame->width;
jpgCodecContext->height = frame->height;
jpgCodecContext->time_base = { 1, codecContext->framerate.num };
avcodec_open2(jpgCodecContext, jpgCodec, nullptr);
AVPacket *jpgPacket = av_packet_alloc();
/* av_init_packet(&jpgPacket);*/
jpgPacket->data = nullptr;
jpgPacket->size = 0;
int ret = avcodec_send_frame(jpgCodecContext,frame);
if (ret < 0) {
std::cout << "Error encoding video frame: " << ret << std::endl;
continue;
}
ret = avcodec_receive_packet(jpgCodecContext, jpgPacket);
//int ret = avcodec_encode_video2(jpgCodecContext, &jpgPacket, frame, nullptr);
if (ret >= 0) {
FILE* file = fopen(filename, "wb");
if (file != nullptr) {
fwrite(jpgPacket->data, 1, jpgPacket->size, file);
fclose(file);
}
}
av_packet_unref(jpgPacket);
frameIndex++;
}
av_packet_unref(&packet);
}
// 释放资源
av_frame_free(&frame);
avcodec_close(codecContext);
avformat_close_input(&formatContext);
}
C++ FFmpeg 将视频转换成图片
于 2023-05-31 15:30:23 首次发布