博客

最近在做一个东西，想要将图片压缩后存储到服务器上以节省空间。

直接使用JDK中的类进行压缩处理，但发现压缩后的图片大小跨度区间较大。

最终考虑在计算压缩比例时引入调整值，并对这个调整值进行训练，以尽量缩减压缩后大小与目标大小的差距。

使用不同大小的图片进行训练后就可以得到一组调整值，最终在代码中可以直接使用这组结果来处理。而不是每次都需要重新训练。

实现如下：

package com.irootech.fm.service.impl;

import com.irootech.fm.service.ImageCompressService;
import com.irootech.fm.service.SysConfigService;
import com.sun.image.codec.jpeg.JPEGCodec;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.Optional;

/**
 * 图片压缩工具类
 *
 * @author LiuQI 2018/12/11 19:05
 * @version V1.0
 **/
@Service
public class ImageCompressServiceImpl implements ImageCompressService {
   
    private static final Logger logger = LoggerFactory.getLogger(ImageCompressServiceImpl.class);
    private static final String FILE_TO_COMPRESS_MIN_SIZE = "fileToCompressMinSize";

    // 压缩经验值训练调整幅度，越小最终效果越好，但所需训练数据也越多
    private static final float TRAIN_EMPIRIC_COMPRESS_RATIO_ADJUST = 0.01f;

    // 压缩经验值保存区间范围；如10000表示10K训练一个调整值；100000表示100K训练一个调整值
    private static final int EMPIRIC_RATIO_SPAN_SIZE = 100000;

    private int fileToCompressMinSize;

    private EmpiricRatioList empiricRatioList = new EmpiricRatioList();

    @Resource
    private SysConfigService configService;

    @PostConstruct
    public void init() {
   
        this.fileToCompressMinSize = configService.find(FILE_TO_COMPRESS_MIN_SIZE)
                .map(config -> Integer.valueOf(config.getConfigValue())).orElse(512000);
    }

    /**
     * 对图片进行一定范围内的压缩
     */
    @Override
    public InputStream compress(InputStream fileInputStream, String fileName) {
   
        if (!fileName.endsWith(".jpg")