javaCV視頻處理之提取人像視頻
更新時間:2021年12月24日 11:03:07 作者:洛陽泰山
這篇文章主要介紹了利用JavaCV實現提取視頻中的人像并保存為視頻,文中的示例代碼講解詳細,對我們學習JavaCV有一定的幫助,需要的可以參考一下
效果圖對比

左側的為原視頻,右側為提取人像跳舞的視頻。
pom文件引入依賴
<!-- https://mvnrepository.com/artifact/com.baidu.aip/java-sdk --> <dependency> <groupId>com.baidu.aip</groupId> <artifactId>java-sdk</artifactId> <version>4.16.3</version> </dependency> <!-- https://mvnrepository.com/artifact/org.bytedeco/javacv-platform --> <dependency> <groupId>org.bytedeco</groupId> <artifactId>javacv-platform</artifactId> <version>1.5.5</version> </dependency>
?java核心實現代碼(完整)?
import com.baidu.aip.bodyanalysis.AipBodyAnalysis;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.FFmpegFrameRecorder;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.Java2DFrameConverter;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.HashMap;
import org.bytedeco.ffmpeg.global.avutil;
import org.bytedeco.ffmpeg.global.avcodec;
import org.json.JSONObject;
import sun.misc.BASE64Decoder;
public class VideoProcessor {
//設置APPID/AK/SK
public static final String APP_ID = "25393592";
public static final String API_KEY = "OkRDD6FQwm5hTKGSMIEL9RN4";
public static final String SECRET_KEY = "ONAxohflnqL2HwBEQB2iGUCjmO5lgywp";
final static String videoFolderPath = "C:/Users/liuya/Desktop/video/";
final static String videoName = "demo.mp4";
final static String imageFolderPath = "C:/Users/liuya/Desktop/people/";
public static void main(String[] args) throws Exception {
videoProcess(videoFolderPath + videoName);
}
//視頻水印
public static void videoProcess(String filePath) {
//抓取視頻圖像資源
FFmpegFrameGrabber videoGrabber = new FFmpegFrameGrabber(filePath);
//抓取視頻圖像資源
FFmpegFrameGrabber audioGrabber = new FFmpegFrameGrabber(filePath);
try {
videoGrabber.start();
audioGrabber.start();
FFmpegFrameRecorder recorder = new FFmpegFrameRecorder(videoFolderPath + "new" + videoName, videoGrabber.getImageWidth(), videoGrabber.getImageHeight(), videoGrabber.getAudioChannels());
recorder.setPixelFormat(avutil.AV_PIX_FMT_YUV420P);
recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);
recorder.start();
//處理圖像
int videoSize = videoGrabber.getLengthInVideoFrames();
for (int i = 0; i < videoSize; i++) {
Frame videoFrame = videoGrabber.grabImage();
if (videoFrame != null && videoFrame.image != null) {
System.out.println("視頻共" + videoSize + "幀,正處理第" + (i + 1) + "幀圖片");
Java2DFrameConverter converter = new Java2DFrameConverter();
BufferedImage bi=converter.getBufferedImage(videoFrame);
BufferedImage bufferedImage = splitting(bi);
recorder.record(converter.convert(bufferedImage));
}
}
//處理音頻
for (int i = 0; i < audioGrabber.getLengthInAudioFrames(); i++) {
Frame audioFrame = audioGrabber.grabSamples();
if (audioFrame != null && audioFrame.samples != null) {
recorder.recordSamples(audioFrame.sampleRate, audioFrame.audioChannels, audioFrame.samples);
}
}
recorder.stop();
recorder.release();
videoGrabber.stop();
audioGrabber.stop();
} catch (Exception e) {
e.printStackTrace();
}
}
public static BufferedImage splitting(BufferedImage image){
ByteArrayOutputStream out=new ByteArrayOutputStream();
try {
ImageIO.write(image,"png",out);
} catch (IOException e) {
e.printStackTrace();
}
return splitting(out.toByteArray());
}
public static BufferedImage splitting(byte[] image){
// 初始化一個AipBodyAnalysis
AipBodyAnalysis client = new AipBodyAnalysis(APP_ID, API_KEY, SECRET_KEY);
// 可選:設置網絡連接參數
client.setConnectionTimeoutInMillis(2000);
client.setSocketTimeoutInMillis(60000);
// 傳入可選參數調用接口
HashMap<String, String> options = new HashMap<String, String>();
options.put("type", "foreground");
// 參數為本地路徑
JSONObject res = client.bodySeg(image, options);
return convert(res.get("foreground").toString());
}
public static BufferedImage convert(String labelmapBase64) {
try {
BASE64Decoder decoder = new BASE64Decoder();
byte[] bytes = decoder.decodeBuffer(labelmapBase64);
InputStream is = new ByteArrayInputStream(bytes);
BufferedImage image = ImageIO.read(is);
//失真處理
BufferedImage newBufferedImage = new BufferedImage(image.getWidth(), image.getHeight(), BufferedImage.TYPE_INT_RGB);
newBufferedImage.createGraphics().drawImage(image, 0, 0, Color.WHITE, null);
ByteArrayOutputStream out=new ByteArrayOutputStream();
ImageIO.write(newBufferedImage, "png", out);
ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
return ImageIO.read(in);
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
}
控制臺輸出

到此這篇關于javaCV視頻處理之提取人像視頻的文章就介紹到這了,更多相關javaCV提取人像視頻內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
完美解決Eclipse導入的項目上有個紅叉,但不影響項目運行的問題
這篇文章主要介紹了完美解決Eclipse導入的項目上有個紅叉,但不影響項目運行的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2021-01-01
spring cloud實現Eureka注冊中心的HA的方法
本篇文章主要介紹了spring cloud實現Eureka注冊中心的HA的方法,小編覺得挺不錯的,現在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2018-01-01

