【图片水印识别分类】利用图片水印文字信息实现图片分类与文件夹转移，基于Java和阿里云的解决方案-CFANZ编程社区

大型广告设计公司每天都会产生大量的设计图片，这些图片包含各种不同的项目信息和版权标识（即水印文字）。为了方便后续的查找、使用和管理，需要对图片进行分类整理。

具体应用：

分类规则设定：根据水印文字中的项目名称、客户名称、设计风格等关键信息，设定不同的分类规则。例如，水印文字中包含“XX品牌春季广告”的图片归类到“XX品牌 - 春季广告”文件夹；水印文字中体现“简约风格”的图片归类到“简约风格设计”文件夹。自动化分类与移动：使用专门的图片管理软件或脚本程序，自动识别图片中的水印文字，并按照预设的分类规则将图片移动到相应的文件夹中。这样，设计师在需要查找特定项目的图片时，只需进入对应的文件夹即可，大大提高了工作效率。以下是一个基于 Java 和阿里云 OCR 服务，利用图片水印文字信息实现图片分类与文件夹转移的解决方案，包含详细步骤、代码示例和相关解释。

准备工作注册阿里云账号：访问阿里云官网进行注册并完成实名认证。开通 OCR 服务：在阿里云控制台中开通 OCR 服务，并创建一个 AccessKey，用于后续的 API 调用身份验证。添加依赖：在你的 Java 项目中添加阿里云 OCR SDK 的依赖。如果你使用 Maven 项目，可以在 pom.xml 中添加以下依赖： xml

import java.io.*;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;
import java.util.Base64;
public class ImageClassificationByWatermark {


// 阿里云 AccessKey 信息
private static final String ACCESS_KEY_ID = "your_access_key_id";
private static final String ACCESS_KEY_SECRET = "your_access_key_secret";
private static final String REGION_ID = "cn-hangzhou";

public static void main(String[] args) {
    String inputDirectory = "path/to/input/directory";
    String outputDirectory = "path/to/output/directory";

    // 初始化 AcsClient
    DefaultProfile profile = DefaultProfile.getProfile(REGION_ID, ACCESS_KEY_ID, ACCESS_KEY_SECRET);
    IAcsClient client = new DefaultAcsClient(profile);

    File inputDir = new File(inputDirectory);
    if (inputDir.exists() && inputDir.isDirectory()) {
        File[] files = inputDir.listFiles();
        if (files != null) {
            for (File file : files) {
                if (file.isFile()) {
                    try {
                        // 调用 OCR 识别图片水印文字
                        String watermarkText = recognizeWatermark(client, file);
                        if (watermarkText != null && !watermarkText.isEmpty()) {
                            // 创建分类文件夹
                            File categoryDir = new File(outputDirectory + File.separator + watermarkText);
                            if (!categoryDir.exists()) {
                                categoryDir.mkdirs();
                            }
                            // 转移图片到分类文件夹
                            Files.move(file.toPath(), new File(categoryDir, file.getName()).toPath(), StandardCopyOption.REPLACE_EXISTING);
                        }
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            }
        }
    }
}

private static String recognizeWatermark(IAcsClient client, File imageFile) throws Exception {
    // 将图片转换为 Base64 编码
    byte[] imageBytes = Files.readAllBytes(imageFile.toPath());
    String imageBase64 = Base64.getEncoder().encodeToString(imageBytes);

    // 创建 OCR 请求
    RecognizeGeneralRequest request = new RecognizeGeneralRequest();
    request.setAcceptFormat("json");
    request.setImageContent(imageBase64);

    // 发送请求并获取响应
    RecognizeGeneralResponse response = client.getAcsResponse(request);
    String responseJson = response.getBody();

    // 解析响应结果，提取文字信息
    JSONObject jsonObject = new JSONObject(responseJson);
    JSONArray wordInfoList = jsonObject.getJSONObject("Data").getJSONArray("PrismWordInfoList");
    StringBuilder watermarkText = new StringBuilder();
    for (int i = 0; i < wordInfoList.length(); i++) {
        watermarkText.append(wordInfoList.getJSONObject(i).getString("text")).append(" ");
    }
    return watermarkText.toString().trim();
}}

代码说明初始化 AcsClient：使用阿里云的 DefaultProfile 和 DefaultAcsClient 类初始化一个客户端，用于后续的 OCR 请求。遍历输入文件夹：遍历指定的输入文件夹中的所有图片文件。调用 OCR 服务：将图片转换为 Base64 编码，创建 RecognizeGeneralRequest 请求对象，发送请求并获取响应。解析 OCR 结果：解析响应的 JSON 数据，提取其中的文字信息作为水印文字。创建分类文件夹并转移图片：根据水印文字创建相应的分类文件夹，并将图片移动到该文件夹中。
注意事项请将 your_access_key_id 和 your_access_key_secret 替换为你自己的阿里云 AccessKey ID 和 Secret。请将 path/to/input/directory 和 path/to/output/directory 替换为实际的输入文件夹和输出文件夹路径。确保你的 Java 项目具有足够的权限来读取输入文件夹中的图片文件，并在输出文件夹中创建新的文件夹和移动文件。
扩展建议错误处理：在实际应用中，可以添加更完善的错误处理机制，例如处理 OCR 请求失败、文件读写错误等情况。性能优化：如果需要处理大量图片，可以考虑使用多线程来提高处理效率。文字处理：可以对识别出的水印文字进行进一步的处理，例如去除特殊字符、转换为小写等，以提高分类的准确性。