TASK: 当选择童装时,sketch生成和to product的prompt需添加儿童限定词
This commit is contained in:
@@ -58,4 +58,7 @@ public class GenerateThroughImageTextDTO {
|
|||||||
|
|
||||||
@ApiModelProperty("种子 取值范围 0~500")
|
@ApiModelProperty("种子 取值范围 0~500")
|
||||||
private String seed;
|
private String seed;
|
||||||
|
|
||||||
|
@ApiModelProperty("年龄区间 Adult || Child")
|
||||||
|
private String ageGroup = "Adult";
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,4 +15,5 @@ public class ToProductImageDTO {
|
|||||||
private String direction;
|
private String direction;
|
||||||
private Double brightenValue;
|
private Double brightenValue;
|
||||||
private String modelName;
|
private String modelName;
|
||||||
|
private String ageGroup;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -173,7 +173,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
|
|||||||
Long elementId = generateThroughImageTextDTO.getCollectionElementId();
|
Long elementId = generateThroughImageTextDTO.getCollectionElementId();
|
||||||
// validateGeneraType(generate, text, elementId);
|
// validateGeneraType(generate, text, elementId);
|
||||||
if (!StringUtil.isNullOrEmpty(text)) {
|
if (!StringUtil.isNullOrEmpty(text)) {
|
||||||
text = modifyPrompt(text, generate, generateThroughImageTextDTO.getLevel1Type());
|
text = modifyPrompt(text, generate, generateThroughImageTextDTO.getLevel1Type(), generateThroughImageTextDTO.getAgeGroup());
|
||||||
}
|
}
|
||||||
|
|
||||||
// todo 这一步现在还是有必要的吗?
|
// todo 这一步现在还是有必要的吗?
|
||||||
@@ -352,7 +352,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private String modifyPrompt(String userInput, Generate generate, String level1Type) {
|
private String modifyPrompt(String userInput, Generate generate, String level1Type, String ageGroup) {
|
||||||
String text = "";
|
String text = "";
|
||||||
String prefix = "";
|
String prefix = "";
|
||||||
if (userInput.startsWith("Painting Style")
|
if (userInput.startsWith("Painting Style")
|
||||||
@@ -384,6 +384,9 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
|
|||||||
case "Sketchboard":
|
case "Sketchboard":
|
||||||
// text = "clear lines, simple outlines monochrome white vector image of " + translated + ", no background, sketch flat, front view display, best quality, ultra-high resolution 8k";
|
// text = "clear lines, simple outlines monochrome white vector image of " + translated + ", no background, sketch flat, front view display, best quality, ultra-high resolution 8k";
|
||||||
text = "a single item of sketch of " + translated + ", 4k, white background";
|
text = "a single item of sketch of " + translated + ", 4k, white background";
|
||||||
|
if (!StringUtil.isNullOrEmpty(ageGroup) && ageGroup.equals("Child")){
|
||||||
|
text = text + ", Children's clothing";
|
||||||
|
}
|
||||||
// generate.setText(text);
|
// generate.setText(text);
|
||||||
default:
|
default:
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -447,6 +447,10 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
|
|||||||
toProductImageResult.setResultType("ToProductImage");
|
toProductImageResult.setResultType("ToProductImage");
|
||||||
} else {
|
} else {
|
||||||
// 走模型
|
// 走模型
|
||||||
|
if (!StringUtil.isNullOrEmpty(toProductImageDTO.getAgeGroup())
|
||||||
|
&& toProductImageDTO.getAgeGroup().equals("Child")){
|
||||||
|
sb.append(", Children's face");
|
||||||
|
}
|
||||||
pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), productType);
|
pythonService.toProductImage(tDesignPythonOutfit.getDesignUrl(), taskId, sb.toString(), toProductImageDTO.getImageStrength(), productType);
|
||||||
toProductImageResult.setModelName("local");
|
toProductImageResult.setModelName("local");
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user