javacv心得,每隔n帧抓一张图
javacv新手,从网上找的大神写好的代码,但是没有自己想要的效果,对代码进行小修改之后实现了想要的功能。
package com.soft;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import javax.imageio.ImageIO;
import org.bytedeco.javacpp.opencv_core;
import org.bytedeco.javacpp.opencv_core.IplImage;
import org.bytedeco.javacpp.presets.opencv_highgui;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.FrameGrabber.Exception;
import org.bytedeco.javacv.Java2DFrameConverter;
import org.bytedeco.javacv.OpenCVFrameConverter;
/**
* javacv
*
* @author 春晓
*
*/
// D:\javatest\testvideo D:\javatest\c
public class FrameGrabberKit {
/*
* static int num=0; static String name="test"+num;
*/
public static void main(String[] args) throws Exception {
System.out.println(System.getProperty("java.library.path"));
try {
fetchFrame("D:/javatest/testvideo/test01.mp4");
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 获取指定视频的帧并保存为图片至指定目录
*
* @param videofile
* 源视频文件路径
* @param framefile
* 截取帧的图片存放路径
* @throws Exception
* @throws IOException
*/
public static void fetchFrame(String videofile) throws Exception, IOException {
FFmpegFrameGrabber ff = new FFmpegFrameGrabber(videofile);
ff.start();
ff.getAudioChannels();
String rotate = ff.getVideoMetadata("rotate");// 视频的旋转角度
int lenght = ff.getLengthInFrames();
System.out.println("lenght:" + lenght);
int i = 0;
int j = 1;
Frame f = null;
while (i < lenght) {
// 过滤前5帧,避免出现全黑的图片,依自己情况而定
//每一千帧生成一张图片保留,若显示帧不为关键帧,则寻找相邻的下一帧,
//因为是空循环,所以把总长度lenght减一
f = ff.grabFrame();
if ((i % 1000) == 0 && i > 5) {
while (f.image == null) {
f = ff.grabFrame();
lenght = lenght - 1;
}
if (f.image != null) {
IplImage src = null;
if (null != rotate && rotate.length() > 1) {
OpenCVFrameConverter.ToIplImage converter = new OpenCVFrameConverter.ToIplImage();
src = converter.convert(f);
f = converter.convert(rotate(src, Integer.valueOf(rotate)));
}
doExecuteFrame(f, "D:/javatest/testpic/test" + j + ".jpg");
j++;
}
}
i++;
}
}
public static IplImage rotate(IplImage src, int angle) {
IplImage img = IplImage.create(src.height(), src.width(), src.depth(), src.nChannels());
opencv_core.cvTranspose(src, img);
opencv_core.cvFlip(img, img, angle);
return img;
}
public static void doExecuteFrame(Frame f, String targetFileName) {
if (null == f || null == f.image) {
return;
}
Java2DFrameConverter converter = new Java2DFrameConverter();
BufferedImage bi = converter.getBufferedImage(f);
File output = new File(targetFileName);
try {
ImageIO.write(bi, "jpg", output);
} catch (IOException e) {
e.printStackTrace();
}
}
}