从手动标注到智能打标：AI 数据标注工具实战解析

3.2 引入依赖（pom.xml）

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.openpnp</groupId>
        <artifactId>opencv</artifactId>
        <version>4.9.0-0</version>
    </dependency>
    <dependency>
        <groupId>com.microsoft.onnxruntime</groupId>
        <artifactId>onnxruntime</artifactId>
        <version>1.16.3</version>
    </dependency>
    <dependency>
        <groupId>com.fasterxml.jackson.core</groupId>
        <artifactId>jackson-databind</artifactId>
    </dependency>
</dependencies>

注意：YOLOv5 ONNX 模型需提前导出，可从 Ultralytics 官方 GitHub 获取。

3.3 加载 ONNX 模型（OnnxModelRunner.java）

package com.example.smartlabeling.util;

import ai.onnxruntime.*;
import org.opencv.core.*;
import org.opencv.imgproc.Imgproc;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.nio.FloatBuffer;
import java.util.ArrayList;
import java.util.List;

@Component
public class OnnxModelRunner {
    private OrtEnvironment env;
    private OrtSession session;

    @PostConstruct
    public void init() throws Exception {
        env = OrtEnvironment.getEnvironment();
        String modelPath = "models/yolov5s.onnx";
        session = env.createSession(modelPath, new OrtSession.SessionOptions());
    }

    public List<BoundingBox> runInference(Mat image) {
        try {
            // 预处理：调整尺寸为 640x640，归一化
            Mat resized = new Mat();
            Imgproc.resize(image, resized, new Size(640, 640));
            resized.convertTo(resized, CvType.CV_32F, 1.0 / 255.0);

            // 转为 NCHW 格式 (1,3,640,640)
            float[][][][] inputArray = new float[1][3][640][640];
            for (int c = 0; c < 3; c++) {
                for (int i = 0; i < 640; i++) {
                    for (int j = 0; j < 640; j++) {
                        double[] pixel = new double[3];
                        resized.get(i, j, pixel);
                        inputArray[0][c][i][j] = (float) pixel[c];
                    }
                }
            }

            OnnxTensor tensor = OnnxTensor.createTensor(env, inputArray);
            OrtSession.Result result = session.run(Map.of("images", tensor));

            // 解析输出 (1, 25200, 85)
            OnnxTensor outputTensor = (OnnxTensor) result.get(0);
            float[][][] detections = (float[][][]) outputTensor.getValue();
            return parseDetections(detections, image.size());
        } catch (Exception e) {
            e.printStackTrace();
            return new ArrayList<>();
        }
    }

    private List<BoundingBox> parseDetections(float[][][] rawOutput, Size originalSize) {
        List<BoundingBox> boxes = new ArrayList<>();
        float confThreshold = 0.5f;
        for (float[] detection : rawOutput[0]) {
            float confidence = detection[4];
            if (confidence > confThreshold) {
                // YOLO 输出：[x_center, y_center, w, h, obj_conf, cls_probs...]
                float xCenter = detection[0] * originalSize.width / 640f;
                float yCenter = detection[1] * originalSize.height / 640f;
                float width = detection[2] * originalSize.width / 640f;
                float height = detection[3] * originalSize.height / 640f;
                float x1 = xCenter - width / 2;
                float y1 = yCenter - height / 2;
                float x2 = x1 + width;
                float y2 = y1 + height;
                int classId = argMax(detection, 5, detection.length);
                boxes.add(new BoundingBox(x1, y1, x2, y2, classId, confidence));
            }
        }
        return boxes;
    }

    private int argMax(float[] arr, int start, int end) {
        int maxIdx = start;
        for (int i = start + 1; i < end; i++) {
            if (arr[i] > arr[maxIdx]) maxIdx = i;
        }
        return maxIdx - 5; // 类别索引从 0 开始
    }
}

3.4 预标注服务（PreLabelService.java）

@Service
public class PreLabelService {
    @Autowired
    private OnnxModelRunner modelRunner;

    public List<BoundingBox> generatePreLabels(String imagePath) {
        Mat image = Imgcodecs.imread(imagePath);
        if (image.empty()) {
            throw new RuntimeException("无法加载图像：" + imagePath);
        }
        return modelRunner.runInference(image);
    }
}

3.5 主动学习策略（ActiveLearningService.java）

主动学习的核心是不确定性采样。我们以预测置信度最低的样本优先标注。

@Service
public class ActiveLearningService {
    // 模拟未标注数据池
    private List<String> unlabeledImages = new ArrayList<>();
    private Map<String, Float> uncertaintyScores = new HashMap<>();

    @Autowired
    private PreLabelService preLabelService;

    public void addUnlabeledImage(String imagePath) {
        unlabeledImages.add(imagePath);
    }

    public String getNextImageToLabel() {
        if (unlabeledImages.isEmpty()) return null;

        // 计算每张图的最大预测置信度（越低越不确定）
        for (String path : unlabeledImages) {
            List<BoundingBox> preds = preLabelService.generatePreLabels(path);
            float maxConf = preds.stream().mapToFloat(BoundingBox::getConfidence).max().orElse(0.0f);
            uncertaintyScores.put(path, 1.0f - maxConf); // 不确定性 = 1 - 最大置信度
        }

        // 返回不确定性最高的图像
        return unlabeledImages.stream().max(Comparator.comparing(uncertaintyScores::get)).orElse(null);
    }
}

3.6 控制器接口（AnnotationController.java）

@RestController
@RequestMapping("/api/annotation")
public class AnnotationController {
    @Autowired
    private PreLabelService preLabelService;

    @Autowired
    private ActiveLearningService alService;

    @PostMapping("/prelabel")
    public ResponseEntity<List<BoundingBox>> getPreLabels(@RequestParam String imagePath) {
        try {
            List<BoundingBox> labels = preLabelService.generatePreLabels(imagePath);
            return ResponseEntity.ok(labels);
        } catch (Exception e) {
            return ResponseEntity.status(500).build();
        }
    }

    @PostMapping("/active-learning/next")
    public ResponseEntity<String> getNextImageForLabeling() {
        String nextImage = alService.getNextImageToLabel();
        if (nextImage == null) {
            return ResponseEntity.noContent().build();
        }
        return ResponseEntity.ok(nextImage);
    }

    @PostMapping("/submit")
    public ResponseEntity<Void> submitLabel(@RequestBody AnnotationSubmission submission) {
        // 保存标注结果到数据库（此处省略）
        System.out.println("Received label for: " + submission.getImagePath());
        // 触发模型增量训练（可选）
        return ResponseEntity.ok().build();
    }

    public static class AnnotationSubmission {
        private String imagePath;
        private List<BoundingBox> labels;
        // getters/setters
    }
}

3.7 启动类

@SpringBootApplication
public class SmartLabelingApplication {
    static {
        nu.pattern.OpenCV.loadShared(); // 加载 OpenCV native 库
    }

    public static void main(String[] args) {
        SpringApplication.run(SmartLabelingApplication.class, args);
    }
}

四、前端集成与用户体验

虽然本文聚焦后端，但好的标注工具离不开直观的前端。我们可以用 Vue/React 构建一个简单界面，调用上述 API：

• 显示图像
• 叠加预标注框（带置信度）
• 支持拖拽调整、删除、新增
• '接受/拒绝'按钮
• 批量操作

对于通用场景，也可直接集成 Label Studio 作为前端，通过其 ML Backend API 对接我们的 Java 服务。

五、进阶：构建闭环训练流水线

真正的智能打标不是一次性预标注，而是持续迭代：

1. 初始模型 → 预标注一批数据
2. 人工校验 → 提交高质量标签
3. 新标签加入训练集 → 微调模型
4. 更新预标注模型 → 进入下一轮

5.1 模型增量训练（伪代码）

# 在 submitLabel 接口后触发
@Async
public void triggerIncrementalTraining() {
    if (newLabelsCount > THRESHOLD) {
        // 调用 Python 训练脚本（或使用 DL4J）
        ProcessBuilder pb = new ProcessBuilder("python", "train_incremental.py");
        pb.start();
        // 训练完成后替换 ONNX 模型文件
        // 重启 OnnxModelRunner（或热加载）
    }
}

注意：Java 生态中深度学习框架较少，建议用 Python 负责训练，Java 负责服务部署，通过 gRPC 或 REST 通信。

六、评估与质量控制

如何判断智能打标是否有效？

6.1 关键指标

6.2 质量控制策略

• 交叉验证：随机抽取 5% 样本由第二人复核。
• 置信度过滤：低于阈值的预标注强制人工介入。
• 异常检测：检测标注框面积突变、类别跳跃等异常行为。

实际项目中，YOLO 等成熟模型在常见场景下预标注接受率可达 70%+。

七、挑战与未来方向

尽管智能打标前景广阔，仍面临挑战：

7.1 当前局限

• 领域迁移问题：COCO 预训练模型在医疗图像上表现差。
• 长尾类别：罕见类别缺乏预标注能力。
• 多模态标注：图文、音视频对齐标注复杂度高。

7.2 未来趋势

• 大模型驱动：利用 GPT-4V、Gemini 等多模态大模型进行零样本预标注。
• 自动化质检：用 AI 自动检测标注错误（如 CleanLab）。
• 联邦标注：在隐私保护前提下跨机构协作标注。

八、结语：迈向高效 AI 数据工厂

从手动标注到智能打标，不仅是工具的升级，更是数据生产范式的变革。通过人机协同，我们能以更低的成本、更快的速度、更高的质量构建训练数据，从而加速 AI 落地。

本文提供的 Java 示例虽简化，但展示了核心思想：将预训练模型嵌入标注流程，结合主动学习策略，形成闭环优化。你可以在此基础上扩展：

• 支持文本 NER 标注（集成 spaCy 或 Transformers）
• 添加用户权限与任务分配
• 集成 MinIO 存储海量图像
• 使用 Kafka 实现异步标注事件流

AI 的未来属于那些能高效驾驭数据的人。

延伸阅读

• Label Studio 官方文档
• Active Learning Literature Review (Settles, 2009)
• ONNX Runtime Java API Guide