软件构造结课实验_2

package com.baidu.ai.aip;

import com.baidu.ai.aip.utils.Base64Util;
import com.baidu.ai.aip.utils.FileUtil;
import com.baidu.ai.aip.utils.HttpUtil;

import java.io.FileOutputStream;
import java.net.URLEncoder;
import java.util.Base64;

public class Image_definition_enhance {

    public static String image_definition_enhance(String filePath) {
        if (filePath == null || filePath.isEmpty()) {
            System.out.println("Error: 无效的图像路径!");
            return "";
        }

        // 请求url
        String url = "https://aip.baidubce.com/oauth/2.0/token";
        try {
            // 使用传入的文件路径
            byte[] imgData = FileUtil.readFileByBytes(filePath);
            String imgStr = Base64Util.encode(imgData);
            String imgParam = URLEncoder.encode(imgStr, "UTF-8");

            String param = "image=" + imgParam;

            // 获取 access_token(一般情况下可以缓存)
            String accessToken = "24.b83939e6d9bca2a235df588ce0d745b0.2592000.1736018850.282335-116551191";
            String result = HttpUtil.post(url, accessToken, param);
            System.out.println("API 返回结果: " + result);  // 打印返回结果用于调试

            // 提取 Base64 编码的图片数据
            String base64Image = extractBase64Image(result);

            // 调试:打印提取的 Base64 字符串
            System.out.println("提取的 Base64 字符串: " + base64Image);

            // 将 Base64 字符串解码为字节数组并保存为图片文件
            byte[] imageBytes = Base64.getDecoder().decode(base64Image);
            String outputPath = "E:\\newpage\\enhanced_output_image.jpg";  // 保存的图片路径

            // 打印保存路径,检查路径是否正确
            System.out.println("保存的路径: " + outputPath);

            try (FileOutputStream fos = new FileOutputStream(outputPath)) {
                fos.write(imageBytes);
            }
            System.out.println("图片已保存至: " + outputPath);
            return outputPath;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    // 提取返回结果中的 Base64 图片数据
    private static String extractBase64Image(String json) {
        // 使用正则或者 JSON 解析提取 base64 字符串(需要根据 API 返回的 JSON 格式来提取)
        String base64Image = json.substring(json.indexOf("\"image\":\"") + 8, json.lastIndexOf("\""));
        return base64Image;
    }

    public static void main(String[] args) {
        Image_definition_enhance.image_definition_enhance("E:\\图片\\1724212883022.jpg");
    }
}
package com.baidu.ai.aip;

import com.baidu.ai.aip.utils.Base64Util;
import com.baidu.ai.aip.utils.FileUtil;
import com.baidu.ai.aip.utils.HttpUtil;

import java.net.URLEncoder;

/**
 * 人像动漫化
 */
public class Selfie_anime {

    /**
     * 重要提示代码中所需工具类
     * FileUtil,Base64Util,HttpUtil,GsonUtils请从
     * https://ai.baidu.com/file/658A35ABAB2D404FBF903F64D47C1F72
     * https://ai.baidu.com/file/C8D81F3301E24D2892968F09AE1AD6E2
     * https://ai.baidu.com/file/544D677F5D4E4F17B4122FBD60DB82B3
     * https://ai.baidu.com/file/470B3ACCA3FE43788B5A963BF0B625F3
     * 下载
     */
    public static String selfie_anime(String filePath) {
        if (filePath == null || filePath.isEmpty()) {
            System.out.println("Error: 无效的图像路径!");
            return "";
        }

        // 请求url
        String url = "https://aip.baidubce.com/rest/2.0/image-process/v1/selfie_anime";
        try {
            // 使用传入的文件路径
            byte[] imgData = FileUtil.readFileByBytes(filePath);
            if (imgData == null || imgData.length == 0) {
                System.out.println("Error: 无法读取图像文件或文件为空!");
                return "";
            }

            String imgStr = Base64Util.encode(imgData);

            // 调试:打印Base64编码前的图像数据长度
            System.out.println("Image data length: " + imgData.length);
            System.out.println("Base64 Encoded string length: " + imgStr.length());

            // URLEncoder 编码
            String imgParam = URLEncoder.encode(imgStr, "UTF-8");

            String param = "image=" + imgParam;

            // 注意这里仅为了简化编码每一次请求都去获取access_token,线上环境access_token有过期时间, 客户端可自行缓存,过期后重新获取。
            String accessToken = "24.ab85520f41c69c5c49ce966473ac2be1.2592000.1736016942.282335-116551191";

            String result = HttpUtil.post(url, accessToken, param);
            System.out.println("API 返回结果: " + result);  // 打印返回结果用于调试
            return result;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "";
    }

    public static void main(String[] args) {
        Selfie_anime.selfie_anime("E:\\图片\\1724212883022.jpg");
    }
}

 

posted @ 2024-12-11 09:20  KuanDong24  阅读(0)  评论(0编辑  收藏  举报