mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-10 03:28:49 +00:00
- 【人脸检测】新增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:
@@ -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();
|
||||
|
||||
Reference in New Issue
Block a user