詳解如何用Java實現(xiàn)對m3u8直播流抽幀

更新時間：2023年03月06日 17:08:00 作者：全村最野的狗

抽幀（frame extraction）是指從視頻流中提取一些特定的幀，通常是關鍵幀或者隨機幀，以供后續(xù)處理。這篇文章主要為大家介紹了如何用Java實現(xiàn)對m3u8直播流抽幀，需要的可以參考一下

什么是抽幀

抽幀（frame extraction）是指從視頻流中提取一些特定的幀，通常是關鍵幀或者隨機幀，以供后續(xù)處理。對于m3u8直播流，可以使用Java中的FFmpeg庫來實現(xiàn)抽幀功能。

什么是 FFmpeg

FFmpeg是一套可以用來記錄、轉換數(shù)字音頻、視頻，并能將其轉化為流的開源計算機程序。采用LGPL或GPL許可證。它提供了錄制、轉換以及流化音視頻的完整解決方案。

什么是 JavaCV

JavaCV 是一款基于JavaCPP 調(diào)用方式，由多種開源計算機視覺庫組成的包裝庫，封裝了包含F(xiàn)Fmpeg、OpenCV、tensorflow、caffe、tesseract、libdc1394、OpenKinect、videoInput和ARToolKitPlus等在內(nèi)的計算機視覺領域的常用庫和實用程序類。

最簡單的抽幀

使用 Java 中的 FFmpeg 庫實現(xiàn)的最簡單的抽幀。

<dependency>
    <groupId>org.bytedeco</groupId>
    <artifactId>javacv</artifactId>
    <version>1.4.1</version>
</dependency>

import org.bytedeco.javacv.*;
import java.io.IOException;

public class FrameExtractor {
    public static void main(String[] args) throws IOException, FrameGrabber.Exception, FrameRecorder.Exception {
        FFmpegFrameGrabber grabber = new FFmpegFrameGrabber("直播流地址xxx.m3u8");
        grabber.setOption("rtsp_transport", "tcp");
        grabber.start();

        Frame frame;
        Java2DFrameConverter converter = new Java2DFrameConverter();
        int i = 0;

        // grabber.grabImage() 獲取幀圖片，不包含音頻
        // grabber.grab() 包含音頻
        while ((frame = grabber.grabImage()) != null) {
            // 在這里處理抽取到的幀
            // 例如，將幀保存為圖像文件
            converter.convert(frame).createGraphics().dispose();
            String outputFilename = "frame_" + i + ".jpg";
            File f = new File(outputFilename);
            if(!f.exists())f.mkdirs();
            ImageIO.write(converter.convert(frame), "jpg", f);
            i++;
        }
        grabber.stop();
    }
}

抽幀算法

什么是幀率：每秒刷新幾次就是幾幀。例如25幀就是每秒展示25張圖片。

指定每幾秒抽取幾幀。這里的核心思想是，平均數(shù)累加。

假設對幀率為25的視頻。要實現(xiàn)每3秒抽3幀。

設：幀率=fps；時間=t；t時間內(nèi)抽取總幀數(shù)=x；

avg=(fps.t)/x;

只需找出 1*avg,2*avg,...,x*avg分別對應的值就找到了需要抽取的幀。

最后只需要使用一個變量對幀率計數(shù)。在指定的幀率進行抽取操作就可以了。

代碼：

public class FrameExtractor {
    public static void main(String[] args) throws IOException, FrameGrabber.Exception, FrameRecorder.Exception {
        // 讀取流
        FFmpegFrameGrabber grabber = new FFmpegFrameGrabber("http:.m3u8");
        grabber.setOption("rtsp_transport", "tcp");
        grabber.start();

        Frame frame;
        Java2DFrameConverter converter = new Java2DFrameConverter();

        // 控制讀取幀數(shù)
        int frameRate = (int) grabber.getFrameRate();
        int targetFrameRateNum[] = {10,10}; //（每10秒讀10幀） 目標幀率數(shù)量，每幾秒鐘讀取幾幀
        SortedSet<Integer> targetFrameRate = new TreeSet<>(); // 需要抽取的目標幀率
        // 計算需要抽取的目標幀
        int calFrameRate = frameRate * targetFrameRateNum[0];
        int partSize = calFrameRate / targetFrameRateNum[1];
        int remainder = calFrameRate % targetFrameRateNum[1];
        int current = 1;
        for (int i = 0; i < targetFrameRateNum[1]; i++) {
            // 避免出現(xiàn)幀數(shù)多一個
            if (current <= calFrameRate){
                targetFrameRate.add(current);
            }
            current += partSize;
            if (i < remainder) {
                current++;
            }
        }

        // 計數(shù)器
        int frameCount = 0;
        int i = 0;
        while ((frame = grabber.grabImage()) != null) {
            // 每讀取一幀，增加計數(shù)器
            frameCount++;
            // 如果計數(shù)器達到目標幀率，則進行處理
            if (targetFrameRate.contains(frameCount)) {
                // 處理抽取到的幀

            }
            // 重置計數(shù)器
            if (calFrameRate == frameCount){
                frameCount = 0;
            }
        }

        grabber.stop();
    }
}

到此這篇關于詳解如何用Java實現(xiàn)對m3u8直播流抽幀的文章就介紹到這了,更多相關Java m3u8直播流抽幀內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: