From cd2de667b89e03015e55e50867f498c0599c336e Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Sun, 27 Apr 2025 15:52:29 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=86=E9=A2=91=E5=88=86=E7=B1=BB=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD=E5=BC=80=E5=8F=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/MachineLearnServiceImpl.java | 11 +++-
 .../vo/VideoClassificationParamVo.java        | 50 +++++++++++++++++++
 2 files changed, 60 insertions(+), 1 deletion(-)
 create mode 100644 ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/VideoClassificationParamVo.java

diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/MachineLearnServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/MachineLearnServiceImpl.java
index 32d7dd07..1ac75c1b 100644
--- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/MachineLearnServiceImpl.java
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/MachineLearnServiceImpl.java
@@ -12,6 +12,7 @@ import com.ruoyi.platform.utils.HttpUtils;
 import com.ruoyi.platform.utils.JsonUtils;
 import com.ruoyi.platform.vo.AutoMlParamVo;
 import com.ruoyi.platform.vo.TextClassificationParamVo;
+import com.ruoyi.platform.vo.VideoClassificationParamVo;
 import com.ruoyi.system.api.constant.Constant;
 import org.apache.commons.collections4.MapUtils;
 import org.apache.commons.lang3.StringUtils;
@@ -37,6 +38,8 @@ public class MachineLearnServiceImpl implements MachineLearnService {
     String convertAutoML;
     @Value("${argo.convertTextClassification}")
     String convertTextClassification;
+    @Value("${argo.convertVideoClassification}")
+    String convertVideoClassification;
     @Value("${argo.workflowRun}")
     private String argoWorkflowRun;
     @Value("${minio.endpointIp}")
@@ -143,7 +146,12 @@ public class MachineLearnServiceImpl implements MachineLearnService {
                     break;
                 }
                 case Constant.ML_VideoClassification: {
-                    // todo
+                    VideoClassificationParamVo paramVo = JsonUtils.jsonToObject(machineLearn.getParam(), VideoClassificationParamVo.class);
+                    computingResourceId = paramVo.getComputingResourceId();
+                    if (resourceOccupyService.haveResource(computingResourceId, 1)) {
+                        param = JsonUtils.getConvertParam(paramVo);
+                        convertRes = HttpUtils.sendPost(argoUrl + convertVideoClassification, param);
+                    }
                     break;
                 }
             }
@@ -205,6 +213,7 @@ public class MachineLearnServiceImpl implements MachineLearnService {
                     break;
                 }
                 case Constant.ML_VideoClassification: {
+                    machineLearnIns.setResultPath(outputPath);
                     break;
                 }
             }
diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/VideoClassificationParamVo.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/VideoClassificationParamVo.java
new file mode 100644
index 00000000..38d5c21e
--- /dev/null
+++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/vo/VideoClassificationParamVo.java
@@ -0,0 +1,50 @@
+package com.ruoyi.platform.vo;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.databind.PropertyNamingStrategy;
+import com.fasterxml.jackson.databind.annotation.JsonNaming;
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+import java.util.Map;
+
+@Data
+@JsonNaming(PropertyNamingStrategy.SnakeCaseStrategy.class)
+@JsonInclude(JsonInclude.Include.NON_NULL)
+@ApiModel(description = "视频分类参数")
+public class VideoClassificationParamVo {
+
+    @ApiModelProperty(value = "类别数量")
+    private Integer numClasses;
+
+    @ApiModelProperty(value = "数据集")
+    private Map<String, Object> dataset;
+
+    @ApiModelProperty(value = "epochs")
+    private Integer epochs;
+
+    @ApiModelProperty(value = "batch_size")
+    private Integer batchSize;
+
+    @ApiModelProperty(value = "学习率")
+    private Float lr;
+
+    @ApiModelProperty(value = "是否验证")
+    private Boolean isValidate;
+
+    @ApiModelProperty(value = "训练集路径")
+    private String trainDataPrefix;
+
+    @ApiModelProperty(value = "验证集路径")
+    private String validDataPrefix;
+
+    @ApiModelProperty(value = "训练集标注文件")
+    private String trainFilePath;
+
+    @ApiModelProperty(value = "验证集标注文件")
+    private String validFilePath;
+
+    private Integer computingResourceId;
+
+}