mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-15 14:33:03 +00:00
a8e7ce6c4e
- 【人脸识别】新增Seetaface6轻量模型 - 【目标检测】支持视频流目标检测（rtsp、视频文件等） - 【目标检测】支持tensorflow2目标检测模型 - 【目标检测】新增行人检测模型(yolo-person) - 【通用视觉】新增4个动作识别模型 - 【通用视觉】新增语义分割模型 - 【通用视觉】新增5个实例分割模型(含yolov8-seg、yolov11-seg) - 【通用视觉】新增yolo-obb11旋转框检测(含yolov11-obb) - 【通用视觉】新增5个姿态估计模型(含yolov8-pose、yolov11-pose)
43 lines
1.1 KiB
Java
43 lines
1.1 KiB
Java
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;
|
|
}
|
|
|
|
}
|