PoseTransformation-初版

src/main/java/com/ai/da/controller/GenerateController.java

@@ -16,7 +16,6 @@ import org.springframework.web.bind.annotation.*;
 import javax.annotation.Resource;
 import javax.validation.Valid;
 import java.util.List;
-
 /**
  * @author XP
  */
@@ -98,4 +97,30 @@ public class GenerateController {
         return Response.success(generateService.modifySketch(generateModifyDTO));
     }
 
+    @ApiOperation(value = "请求poseTransform，异步获取结果")
+    @PostMapping("/poseTransform")
+    public Response<String> poseTransform(@ApiParam("projectId") @RequestParam Long projectId,
+                                          @ApiParam("productImage") @RequestParam String productImage,
+                                          @ApiParam("poseId") @RequestParam int poseId) {
+        return Response.success(generateService.poseTransform(projectId, productImage, poseId));
+    }
+
+    @ApiOperation(value = "获取pose transformation生成结果")
+    @PostMapping("/poseTransformResult")
+    public Response<PoseTransformationVO> getPoseTransformationResults(@ApiParam("taskId") @RequestParam String taskId) {
+        PoseTransformationVO generateResult = generateService.getPoseTransformationResult(taskId);
+        return Response.success(generateResult);
+    }
+
+    public Response<String> modifyModelProportion(){
+        return null;
+    }
+
+    public Response<String> sketchReconstruction(){
+        return null;
+    }
+
+
+
+
 }