mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-17 23:52:45 +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:
@@ -0,0 +1,42 @@
|
||||
package cn.smartjavaai.pose.enums;
|
||||
|
||||
/**
|
||||
* 姿态估计模型枚举
|
||||
* @author dwj
|
||||
*/
|
||||
public enum PoseModelEnum {
|
||||
|
||||
YOLO11N_POSE_PT("djl://ai.djl.pytorch/yolo11n-pose"),
|
||||
|
||||
YOLOV8N_POSE_PT("djl://ai.djl.pytorch/yolov8n-pose"),
|
||||
YOLO11N_POSE_ONNX("djl://ai.djl.onnxruntime/yolo11n-pose"),
|
||||
|
||||
YOLOV8N_POSE_ONNX("djl://ai.djl.onnxruntime/yolov8n-pose");
|
||||
|
||||
// SIMPLE_POSE_MXNET("djl://ai.djl.mxnet/simple_pose");
|
||||
|
||||
|
||||
/**
|
||||
* 根据名称获取枚举 (忽略大小写和下划线变体)
|
||||
*/
|
||||
public static PoseModelEnum fromName(String name) {
|
||||
String formatted = name.trim().toUpperCase().replaceAll("[-_]", "");
|
||||
for (PoseModelEnum model : values()) {
|
||||
if (model.name().replaceAll("_", "").equals(formatted)) {
|
||||
return model;
|
||||
}
|
||||
}
|
||||
throw new IllegalArgumentException("未知模型名称: " + name);
|
||||
}
|
||||
|
||||
private final String modelUri;
|
||||
|
||||
PoseModelEnum(String modelUri) {
|
||||
this.modelUri = modelUri;
|
||||
}
|
||||
|
||||
public String getModelUri() {
|
||||
return modelUri;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user