临时提交

vision/src/main/java/cn/smartjavaai/action/enums/ActionRecModelEnum.java

@@ -0,0 +1,48 @@
+package cn.smartjavaai.action.enums;
+
+/**
+ * 人类动作识别模型枚举
+ * @author dwj
+ */
+public enum ActionRecModelEnum {
+
+    VIT_BASE_PATCH16_224("djl://ai.djl.pytorch/Human-Action-Recognition-VIT-Base-patch16-224"),
+
+    INCEPTIONV3_KINETICS400(""),
+
+    INCEPTIONV1_KINETICS400(""),
+
+    RESNET18_V1B_KINETICS400(""),
+
+    RESNET34_V1B_KINETICS400(""),
+
+    RESNET50_V1B_KINETICS400(""),
+
+    RESNET101_V1B_KINETICS400(""),
+
+    RESNET152_V1B_KINETICS400("");
+
+    /**
+     * 根据名称获取枚举 (忽略大小写和下划线变体)
+     */
+    public static ActionRecModelEnum fromName(String name) {
+        String formatted = name.trim().toUpperCase().replaceAll("[-_]", "");
+        for (ActionRecModelEnum model : values()) {
+            if (model.name().replaceAll("_", "").equals(formatted)) {
+                return model;
+            }
+        }
+        throw new IllegalArgumentException("未知模型名称: " + name);
+    }
+
+    private final String modelUri;
+
+    ActionRecModelEnum(String modelUri) {
+        this.modelUri = modelUri;
+    }
+
+    public String getModelUri() {
+        return modelUri;
+    }
+
+}