springboot（JavaCV ）实现视频截取第N帧并保存图片

- 日理万妓 2023-10-15 06:24 60阅读 0赞

现在视频网站展示列表都是用img标签展示的，动图用的是gif，但是我们上传视频时并没有视屏封面，就这需要上传到服务器时自动生成封面并保存

本博客使用jar包的方式实现上传视频文件并且截取视频第一帧，保存到阿里云的OSS（也可以保存到本地获取其他任何地方）。

JavaCV 是一款开源的视觉处理库，基于GPLv2协议，对各种常用计算机视觉库封装后的一组jar包，

封装了OpenCV、libdc1394、OpenKinect、videoInput和ARToolKitPlus等计算机视觉编程人员常用库的接口。

此方法的好处是不需要再服务器上安装插件，直接代码中就可以实现视频截取。

我们需要截取视频第一帧，主要用到了ffmpeg和opencv。

一 , 引入jar包  
我用到的maven的目前最新javacv版本，1.4.3，它应该支持jdk1.7及以上，我项目用的还是jdk1.8.

不过需要注意的是在使用的过程当中 , maven引入jar的时候 会引入所有平台的版本

全部引入大小在五百兆左右(不建议使用)

<dependency>
                <groupId>org.bytedeco</groupId>
                <artifactId>javacv</artifactId>
                <version>1.4.3</version>
            </dependency>
            <dependency>
                <groupId>org.bytedeco.javacpp-presets</groupId>
                <artifactId>ffmpeg-platform</artifactId>
                <version>4.0.2-1.4.3</version>
            </dependency>
            <dependency>
                <groupId>org.springframework</groupId>
                <artifactId>spring-test</artifactId>
                <version>RELEASE</version>
            </dependency>

二 , java 代码实现

public class ImgTools {
        
        //util调用application.properties
        private final static ResourceBundle RESOURCE_BUNDLE = ResourceBundle.getBundle("application");
        private final static String aliyuVideonImg = RESOURCE_BUNDLE.getString("aliyun.video.img");
     
    //    public static void main(String[] args) throws Exception {
        
    //        ImgTools imgTools = new ImgTools();
    //        System.out.println(imgTools.randomGrabberFFmpegVideoImage
    //                ("视频地址，可以是网络视频，也可以是本地视频"));
    //    }
     
        /**
         * 获取视频缩略图
         *
         * @param filePath：视频路径
         * @throws Exception
         */
        public String randomGrabberFFmpegVideoImage(String filePath) throws Exception {
        
            String targetFilePath = "";
            FFmpegFrameGrabber ff = FFmpegFrameGrabber.createDefault(filePath);
            ff.start();
            //判断是否是竖屏小视频
            String rotate = ff.getVideoMetadata("rotate");
            int ffLength = ff.getLengthInFrames();
            Frame f;
            int i = 0;
            int index = 3;//截取图片第几帧
            while (i < ffLength) {
        
                f = ff.grabImage();
                if (i == index) {
        
                    if (null != rotate && rotate.length() > 1) {
        
                        targetFilePath = doExecuteFrame(f, true);   //获取缩略图
                    } else {
        
                        targetFilePath = doExecuteFrame(f, false);   //获取缩略图
                    }
                    break;
                }
                i++;
            }
            ff.stop();
            return targetFilePath;  //返回的是视频第N帧
        }
     
        /**
         * 截取缩略图，存入阿里云OSS（按自己的上传类型自定义转换文件格式）
         *
         * @param f
         * @return
         * @throws Exception
         */
        public String doExecuteFrame(Frame f, boolean bool) throws Exception {
        
            if (null == f || null == f.image) {
        
                return "";
            }
            Java2DFrameConverter converter = new Java2DFrameConverter();
            BufferedImage bi = converter.getBufferedImage(f);
            if (bool == true) {
        
                Image image = (Image) bi;
                bi = rotate(image, 90);//图片旋转90度
            }
            ByteArrayOutputStream os = new ByteArrayOutputStream();
            ImageIO.write(bi, "png", os);
            byte[] sdf = os.toByteArray();
            InputStream input = new ByteArrayInputStream(os.toByteArray());
            MultipartFile multipartFile = new MockMultipartFile("temp.jpg", "temp.jpg", "temp.jpg", input);
            Aliyunoss aliyunoss = new Aliyunoss();
            //如需了解阿里云OSS，请详读我的另一篇博客（"https://blog.csdn.net/weixin_44401989/article/details/105732856"）
            String url = aliyunoss.uploadAli(multipartFile, aliyuVideonImg);
            return url;
        }
     
        /**
         * 图片旋转角度
         *
         * @param src   源图片
         * @param angel 角度
         * @return 目标图片
         */
        public static BufferedImage rotate(Image src, int angel) {
        
            int src_width = src.getWidth(null);
            int src_height = src.getHeight(null);
            // calculate the new image size
            Rectangle rect_des = CalcRotatedSize(new Rectangle(new Dimension(
                    src_width, src_height)), angel);
     
            BufferedImage res = null;
            res = new BufferedImage(rect_des.width, rect_des.height,
                    BufferedImage.TYPE_INT_RGB);
            Graphics2D g2 = res.createGraphics();
            // transform(这里先平移、再旋转比较方便处理；绘图时会采用这些变化，绘图默认从画布的左上顶点开始绘画，源图片的左上顶点与画布左上顶点对齐，然后开始绘画，修改坐标原点后，绘画对应的画布起始点改变，起到平移的效果；然后旋转图片即可)
     
            //平移（原理修改坐标系原点，绘图起点变了，起到了平移的效果，如果作用于旋转，则为旋转中心点）
            g2.translate((rect_des.width - src_width) / 2, (rect_des.height - src_height) / 2);
     
     
            //旋转（原理transalte(dx,dy)->rotate(radians)->transalte(-dx,-dy);修改坐标系原点后，旋转90度，然后再还原坐标系原点为(0,0),但是整个坐标系已经旋转了相应的度数 ）
            g2.rotate(Math.toRadians(angel), src_width / 2, src_height / 2);
     
    //        //先旋转（以目标区域中心点为旋转中心点，源图片左上顶点对准目标区域中心点，然后旋转）
    //        g2.translate(rect_des.width/2,rect_des.height/ 2);
    //        g2.rotate(Math.toRadians(angel));
    //        //再平移（原点恢复到源图的左上顶点处（现在的右上顶点处），否则只能画出1/4）
    //        g2.translate(-src_width/2,-src_height/2);
     
     
            g2.drawImage(src, null, null);
            return res;
        }
     
        /**
         * 计算转换后目标矩形的宽高
         *
         * @param src   源矩形
         * @param angel 角度
         * @return 目标矩形
         */
        private static Rectangle CalcRotatedSize(Rectangle src, int angel) {
        
            double cos = Math.abs(Math.cos(Math.toRadians(angel)));
            double sin = Math.abs(Math.sin(Math.toRadians(angel)));
            int des_width = (int) (src.width * cos) + (int) (src.height * sin);
            int des_height = (int) (src.height * cos) + (int) (src.width * sin);
            return new java.awt.Rectangle(new Dimension(des_width, des_height));
        }
    }