1. FFMpegSDK下载
下载FFMpegSDK,我在官网上找了好久,现在应该在官网上下载不了,于是就在CSDN上使用积分下载了一下,下载链接:FFMpegSDK
下载解压后:
把shared里面的bin文件复制到dev文件中
bin文件中的.dll文件
复制后的dev文件
2. 建立vs工程
建立头文件
配置属性页>>VC++目录>>包含目录
配置属性页>>VC++目录>>库目录
把刚才的dev中include路径和lib路径分别放入包含目录和库目录中。
配置属性页>>链接器>>输入>>附加依赖项
把lib文件夹里面的lib文件添加进附加依赖项中
avcodec.lib
avdevice.lib
avfilter.lib
avformat.lib
avutil.lib
postproc.lib
swresample.lib
swscale.lib
把bin中的文件复制到项目文件中,把测试视频也放入
环境配置完成
注意
属性页中的两点:
字符集一定改成多字节的
SOL检查改成否
3. 程序
获取视频流程序:
int getStream(AVFormatContext **fmt_ctx, AVCodecContext **dec_ctx, const char* file_name) {
int video_stream_index = -1;
int ret;
bool decoder_init = false;
//打开视频文件
if ((ret = avformat_open_input(fmt_ctx, file_name, NULL, NULL)) < 0) {
av_log(NULL, AV_LOG_ERROR, "fail to open input file.\n");
return ret;
}
//获取流信息
if ((ret = avformat_find_stream_info(*fmt_ctx, NULL)) < 0) {
av_log(NULL, AV_LOG_ERROR, "fail to find stream information.\n");
avformat_close_input(fmt_ctx);
return ret;
}
//从多个流中寻找视频流
for (int i = 0; i < (*fmt_ctx)->nb_streams; i++) {
if ((*fmt_ctx)->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
video_stream_index = i;
//进行解码器的初始化
if (!decoder_init) {
*dec_ctx = (*fmt_ctx)->streams[i]->codec;
AVCodec *cod = avcodec_find_decoder((*dec_ctx)->codec_id);
if (!cod) {
av_log(NULL, AV_LOG_ERROR, "fail to find decoder.\n");
avformat_close_input(fmt_ctx);
return 1;
}
if (avcodec_open2(*dec_ctx, cod, NULL) != 0) {
av_log(NULL, AV_LOG_ERROR, "fail to open codecer.\n");
avformat_close_input(fmt_ctx);
return 2;
}
decoder_init = true;
}
}
return 0;
}
return video_stream_index;
}
保存图片的程序:
void saveImage(const char *filename, unsigned char *data, int width, int height) {
CImage image;
image.Create(width, height, 32, 0);
//依次写入每个像素点的RGB值
for (int i = 0; i < width; i++)
for (int j = 0; j < height; j++) {
image.SetPixelRGB(i, j,
data[j * width * 3 + i * 3 + 0],
data[j * width * 3 + i * 3 + 1],
data[j * width * 3 + i * 3 + 2]);
}
image.Save(filename);
}
视频编码解码的程序:
int decoder(AVFormatContext** fmt_ctx, AVCodecContext** dec_ctx, int video_stream_index, AVFrame *pFrame, AVFrame *pFrameColor, int* width, int* height)
{
AVPacket packet;
int i = 0;
int frameFinished;
uint8_t *buffer;
int numBytes;
char filename[32];
//计算图像所需空间大小并分配空间
numBytes = avpicture_get_size(AV_PIX_FMT_RGB24, (*dec_ctx)->width, (*dec_ctx)->height);
buffer = (uint8_t*)av_malloc(numBytes);
avpicture_fill((AVPicture *)pFrameColor, buffer, AV_PIX_FMT_RGB24, (*dec_ctx)->width, (*dec_ctx)->height);
//读取视频帧并解码得到图像
while (av_read_frame(*fmt_ctx, &packet) >= 0) {
if (packet.stream_index == video_stream_index) {
avcodec_decode_video2(*dec_ctx, pFrame, &frameFinished, &packet);
//得到已解码的图像
if (frameFinished)
{
if (pFrame->key_frame == 1) //判断是否为关键帧
{
struct SwsContext *img_convert_ctx = NULL;
img_convert_ctx = sws_getCachedContext(img_convert_ctx, (*dec_ctx)->width,
(*dec_ctx)->height, (*dec_ctx)->pix_fmt,
(*dec_ctx)->width, (*dec_ctx)->height,
AV_PIX_FMT_RGB24, SWS_BICUBIC,
NULL, NULL, NULL);
if (!img_convert_ctx) {
fprintf(stderr, "Cannot initialize sws conversion context\n");
exit(1);
}
//将解码后的图像格式转化为RGB24
sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data,
pFrame->linesize, 0, (*dec_ctx)->height, pFrameColor->data,
pFrameColor->linesize);
//将图像写入jpg文件
sprintf(filename, "result%d.jpg", ++i);
saveImage(filename, pFrameColor->data[0], (*dec_ctx)->width, (*dec_ctx)->height);
printf("keyframe %d get\n", i);
}
}
}
av_free_packet(&packet);
}
printf("finished");
//释放信息
av_free(buffer);
av_free(pFrameColor);
av_free(pFrame);
avcodec_close(*dec_ctx);
avformat_close_input(fmt_ctx);
}
主程序:
int main()
{
//初始化并分配帧
AVFormatContext *fmt_ctx = NULL;
AVCodecContext *dec_ctx = NULL;
AVFrame *pf = av_frame_alloc();
AVFrame *pfc = av_frame_alloc();
int video_stream_index;
int width, height;
av_register_all();
//读取视频
video_stream_index = getStream(&fmt_ctx, &dec_ctx, "video1.mp4");
//解码并保存关键帧数据为jpg图片输出
decoder(&fmt_ctx, &dec_ctx, video_stream_index, pf, pfc, &width, &height);
}
完整项目链接:FFMpeg关键帧提取