C++ FFmpeg 将视频转换成图片

// 函数:将视频转换为图片序列 ,只对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);
}
  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值