# 目标检测

<dependency>
 <groupId>cn.smartjavaai</groupId>
 <artifactId>vision</artifactId>
 <version>1.0.27</version>
</dependency>

DetectorModelConfig config = new DetectorModelConfig();
config.setModelEnum(DetectorModelEnum.YOLOV12_OFFICIAL_ONNX);
config.setModelPath("/Users/xxx/Documents/develop/model/vision/object/yolov12/yolov12n.onnx");
DetectorModel detectorModel = ObjectDetectionModelFactory.getInstance().getModel(config);

//推荐使用
DetectionResponse detect(Image image);
//以下接口将在后续版本中移除
DetectionResponse detect(String imagePath);
DetectionResponse detect(BufferedImage image);
DetectionResponse detect(byte[] imageData);

Image类说明:

所有接口中我们使用了一个Image类,用来表示图像,具体使用请查看Image类

{
 "detectionInfoList": [ //检测信息列表
 {
 "detectionRectangle": { //矩形框
 "height": 174, // 矩形高度
 "width": 147, // 矩形宽度
 "x": 275, // 左上角横坐标
 "y": 143 // 左上角纵坐标
 },
 "objectDetInfo": { //目标检测信息
 "className": "person" //类别
 },
 "score": 0.8118719 //检测结果置信度分数
 }
 ]
}

void detectAndDraw(String imagePath, String outputPath);
Image detectAndDraw(Image image);

注意事项:

1、模型目录下必须包含 synset.txt(类别文件),否则程序无法运行。网盘中已包含此文件。

DetectorModelConfig config = new DetectorModelConfig();
config.setModelEnum(DetectorModelEnum.YOLOV12_CUSTOM);//自定义YOLOV12模型
// 指定模型路径,需要更改为自己的模型路径
config.setModelPath("/Users/xxx/Documents/develop/fire_model/best.onnx");
DetectorModel detectorModel = ObjectDetectionModelFactory.getInstance().getModel(config);
//指定图片路径,需要更改为自己的图片路径
detectorModel.detectAndDraw("/Users/xxx/Downloads/test.jpg","output/test_detected.jpg");

fire
smoke

注意事项:

1、训练方法详见模型训练教程。

2、支持与其他检测方法通用调用,示例仅使用 detectAndDraw 方法。

StreamDetector detector = new StreamDetector.Builder()
 // 视频源类型:支持 STREAM(视频流)、CAMERA(摄像头)、FILE(视频文件)
 .sourceType(VideoSourceType.STREAM)
 // 视频流地址,支持 rtsp、rtmp、http
 .streamUrl("rtsp://username:password@ip:port/Streaming/Channels/101")
 // 每隔多少帧进行一次检测(需根据模型推理速度调整)
 .frameDetectionInterval(10)
 // 指定目标检测模型
 .detectorModel(getModel())
 // 回调监听器:检测到目标时触发
 .listener(new StreamDetectionListener() {
 @Override
 public void onObjectDetected(List<DetectionInfo> detectionInfoList, Image image) {
 // 建议将耗时操作放到新线程执行
 log.info("检测到目标数量: {}", detectionInfoList.size());
 }
 @Override
 public void onStreamEnded() {
 log.info("视频流检测结束");
 }
 @Override
 public void onStreamDisconnected() {
 log.info("视频流断开连接");
 }
 })
 // 重复检测同一目标的间隔(秒)
 .repeatGap(5)
 .build();

detector.startDetection();

detector.stopDetection();

detector.close();

public interface StreamDetectionListener {
 /**
 * 检测到目标时触发
 * @param detectionInfoList 目标检测信息
 * @param image 当前帧图像
 */
 void onObjectDetected(List<DetectionInfo> detectionInfoList, Image image);
 /** 视频流正常结束 */
 void onStreamEnded();
 /** 视频流断开连接(异常) */
 void onStreamDisconnected();
}