VQ-VAE：离散表征学习的图像压缩利器

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.json.JSONArray;
import org.json.JSONObject;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.URLEncoder;
import java.util.Base64;

public class VQVAEExample {
    private static final String MODEL_URL = "http://localhost:8501/v1/models/vq_vae:predict";
    
    public static void main(String[] args) {
        try {
            // 加载图像
            BufferedImage image = ImageIO.read(new File("input.jpg"));
            
            // 图像预处理
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageIO.write(image, "jpg", baos);
            byte[] imageBytes = baos.toByteArray();
            String base64Image = Base64.getEncoder().encodeToString(imageBytes);
            
            // 构建请求JSON
            JSONObject request = new JSONObject();
            JSONObject inputs = new JSONObject();
            inputs.put("image", base64Image);
            request.put("instances", new JSONArray().put(inputs));
            
            // 发送请求
            HttpClient httpClient = HttpClients.createDefault();
            HttpPost httpPost = new HttpPost(MODEL_URL);
            httpPost.setHeader("Content-Type", "application/json");
            httpPost.setEntity(new StringEntity(request.toString()));
            
            HttpResponse response = httpClient.execute(httpPost);
            HttpEntity entity = response.getEntity();
            String responseString = EntityUtils.toString(entity);
            
            // 解析响应
            JSONObject responseJson = new JSONObject(responseString);
            JSONArray predictions = responseJson.getJSONArray("predictions");
            String compressedData = predictions.getJSONObject(0).getString("compressed");
            
            System.out.println("图像压缩完成！压缩后数据大小: " + compressedData.length() + " 字节");
            
            // 可以将压缩数据保存到文件
            saveCompressedData(compressedData, "compressed.bin");
            
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    private static void saveCompressedData(String compressedData, String filePath) throws IOException {
        try (FileOutputStream fos = new FileOutputStream(filePath)) {
            // 实际应用中需要将Base64编码的压缩数据转换回二进制
            byte[] data = Base64.getDecoder().decode(compressedData);
            fos.write(data);
        }
    }
}