利用JavaCV截取FLV图片

最近公司有个AI需求，需要从摄像头的flv流中截取几张图片(前端很简单，调用API就可以了。由于是后台任务，只能通过JavaCV来调用了)。

• 1.pom依赖(这个依赖打包生成有点大)

  <dependency>
    <groupId>org.bytedeco</groupId>
    <artifactId>javacv-platform</artifactId>
    <version>1.5.3</version>
 </dependency>

• 2.java代码

 /**
     * 获取指定帧数的封面图片
     *
     * @param filePath flv地址(eg:http://ip:port/xxx.flv)
     */
    private static BufferedImage getBufferedImageByFrame(String filePath) throws IOException {
        FFmpegFrameGrabber grabber = FFmpegFrameGrabber.createDefault(filePath);
        return getBufferedImageByFrame(grabber);
    }

    private static BufferedImage getBufferedImageByFrame(FFmpegFrameGrabber grabber)
            throws FrameGrabber.Exception {
        grabber.start();
        Frame frame;
        frame = grabber.grabImage();
        Java2DFrameConverter converter = new Java2DFrameConverter();
        BufferedImage buffer = converter.getBufferedImage(frame);
        grabber.stop();
        return buffer;
    }

    public static byte[] getFlvImg(String path) throws Exception {
        return bufferedImageToByteArray(getBufferedImageByFrame(path));
    }


    /**
     * 将BufferedImage转换为byte[]
     *
     * @param image
     * @return
     */
    public static byte[] bufferedImageToByteArray(BufferedImage image) throws IOException {
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        ImageIO.write(image, "jpg", os);
        return os.toByteArray();
    }

    /**
     * 上传文件
     *
     * @param data     文件数据
     * @param url      上传地址
     * @param fileName 文件名称
     * @return
     */
    public static JSONObject postFile(byte[] data, String url, String fileName) {
        RestTemplate restTemplate = new RestTemplate();
        MultiValueMap<String, Object> paramMap = new LinkedMultiValueMap<>();
        ByteArrayResource contentsAsResource = new ByteArrayResource(data) {
            @Override
            public String getFilename() {
                return fileName;
            }
        };
        paramMap.add("file", contentsAsResource);
        return restTemplate.postForObject(url, paramMap, JSONObject.class);
    }