- 【人脸检测】新增6个模型(MTCNN、YOLOV5、RetinaFace小尺寸版),大幅提升性能

- 【人脸识别】新增Seetaface6轻量模型
- 【目标检测】支持视频流目标检测(rtsp、视频文件等)
- 【目标检测】支持tensorflow2目标检测模型
- 【目标检测】新增行人检测模型(yolo-person)
- 【通用视觉】新增4个动作识别模型
- 【通用视觉】新增语义分割模型
- 【通用视觉】新增5个实例分割模型(含yolov8-seg、yolov11-seg)
- 【通用视觉】新增yolo-obb11旋转框检测(含yolov11-obb)
- 【通用视觉】新增5个姿态估计模型(含yolov8-pose、yolov11-pose)
This commit is contained in:
dengwenjie
2025-09-07 17:19:19 +08:00
parent 2b044fda29
commit a8e7ce6c4e
102 changed files with 4473 additions and 1368 deletions

View File

@@ -37,8 +37,6 @@ public class Yolo5PlateDetectTranslator implements Translator<Image, DetectedObj
private int topK;
private LetterBoxUtils.ResizeResult letterBoxResult;
public Yolo5PlateDetectTranslator(Map<String, ?> arguments) {
confThreshold =
arguments.containsKey("confThreshold")
@@ -62,7 +60,8 @@ public class Yolo5PlateDetectTranslator implements Translator<Image, DetectedObj
imageWidth = (int) array.getShape().get(1);
imageHeight = (int) array.getShape().get(0);
//Letter box resize 640x640 with padding (保持比例,补边缘)
letterBoxResult = LetterBoxUtils.letterbox(manager, array, inputSize, inputSize, 114f, LetterBoxUtils.PaddingPosition.CENTER);
LetterBoxUtils.ResizeResult letterBoxResult = LetterBoxUtils.letterbox(manager, array, inputSize, inputSize, 114f, LetterBoxUtils.PaddingPosition.CENTER);
ctx.setAttachment("letterBoxResult", letterBoxResult);
array = letterBoxResult.image;
// 转为 float32 且归一化到 0~1
array = array.toType(DataType.FLOAT32, false).div(255f); // HWC
@@ -74,6 +73,7 @@ public class Yolo5PlateDetectTranslator implements Translator<Image, DetectedObj
@Override
public DetectedObjects processOutput(TranslatorContext ctx, NDList list) {
NDManager manager = ctx.getNDManager();
LetterBoxUtils.ResizeResult letterBoxResult = (LetterBoxUtils.ResizeResult)ctx.getAttachment("letterBoxResult");
//[x_center, y_center, w, h, obj_conf, 8个关键点, class1_conf, class2_conf]
//目标置信度 obj_conf 5:13 关键点 [13:15]分类得分:单层车牌 / 双层车牌
NDArray dets = list.singletonOrThrow();