aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:mannequin；

Browse Source
This commit is contained in:
shahaibo
2025-05-28 11:12:26 +08:00
parent 74540610c6
commit 718e1c93f6
2 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/com/ai/da/model/vo/ValidateElementVO.java
View File

@@ -61,4 +61,5 @@ public class ValidateElementVO {
private Double brandScale; private Double brandScale;
private List<DesignLibraryModelPointVO> mannequins; private List<DesignLibraryModelPointVO> mannequins;
private DesignLibraryModelPointVO designLibraryModelPointVO;
} }

21
src/main/java/com/ai/da/python/PythonService.java
View File

@@ -348,7 +348,7 @@ public class PythonService {
private DesignPythonObject createDesignPythonObject(ValidateElementVO elementVO, CurrentDesignPictureTypeEnum designPictureType, BigDecimal systemScale, String singleOverall, String switchCategory, int i) { private DesignPythonObject createDesignPythonObject(ValidateElementVO elementVO, CurrentDesignPictureTypeEnum designPictureType, BigDecimal systemScale, String singleOverall, String switchCategory, int i) {
DesignPythonObject pythonObject = new DesignPythonObject(); DesignPythonObject pythonObject = new DesignPythonObject();
pythonObject.setItems(coverToDesignPythonItemNew(elementVO, designPictureType, systemScale)); pythonObject.setItems(coverToDesignPythonItemNew(elementVO, designPictureType, systemScale));
pythonObject.setBasic(coverToBasic(pythonObject.getItems().get(0), singleOverall, switchCategory, elementVO.getDesignLibraryModelPoint(), elementVO.getMannequins())); pythonObject.setBasic(coverToBasic(pythonObject.getItems().get(0), singleOverall, switchCategory, elementVO.getDesignLibraryModelPoint(), elementVO.getDesignLibraryModelPointVO()));
if (CollectionUtil.isNotEmpty(elementVO.getRequestIdList())) { if (CollectionUtil.isNotEmpty(elementVO.getRequestIdList())) {
pythonObject.setObjectSign(elementVO.getRequestIdList().get(i)); pythonObject.setObjectSign(elementVO.getRequestIdList().get(i));
} }
@@ -1696,12 +1696,20 @@ public class PythonService {
//类型 //类型
pythonItem.setType(type); pythonItem.setType(type);
if (SysFileLevel2TypeEnum.BODY.getRealName().equals(type)) { if (SysFileLevel2TypeEnum.BODY.getRealName().equals(type)) {
if (Objects.nonNull(designLibraryModelPoint)) { if (!CollectionUtils.isEmpty(elementVO.getMannequins())) {
pythonItem.setBody_path(designLibraryModelPoint.getTemplateUrl()); DesignLibraryModelPointVO vo = getRandomDesignLibraryModelPoint(elementVO.getMannequins());
pythonItem.setBody_path(vo.getTemplateUrl());
elementVO.setDesignLibraryModelPointVO(vo);
} else { } else {
// pythonItem.setBody_path("/workspace/python_code/Multi-layer-Virtual-Try-on/dataset_for_test/Img_model.png"); // pythonItem.setBody_path("/workspace/python_code/Multi-layer-Virtual-Try-on/dataset_for_test/Img_model.png");
pythonItem.setBody_path("aida-mannequins/model_1693218345.2714432.png"); pythonItem.setBody_path("aida-mannequins/model_1693218345.2714432.png");
} }
// if (Objects.nonNull(designLibraryModelPoint)) {
// pythonItem.setBody_path(designLibraryModelPoint.getTemplateUrl());
// } else {
//// pythonItem.setBody_path("/workspace/python_code/Multi-layer-Virtual-Try-on/dataset_for_test/Img_model.png");
// pythonItem.setBody_path("aida-mannequins/model_1693218345.2714432.png");
// }
} else { } else {
SysFileVO sysFileVO = getRandomSysFileByLevel2Type(type, Lists.newArrayList(), elementVO.getModelSex()); SysFileVO sysFileVO = getRandomSysFileByLevel2Type(type, Lists.newArrayList(), elementVO.getModelSex());
pythonItem.setPath(sysFileVO.getUrl()); pythonItem.setPath(sysFileVO.getUrl());
@@ -2167,7 +2175,7 @@ public class PythonService {
} }
private DesignPythonBasic coverToBasic(DesignPythonItem designPythonItem, String singleOverall, private DesignPythonBasic coverToBasic(DesignPythonItem designPythonItem, String singleOverall,
String switchCategory, DesignLibraryModelPointVO designLibraryModelPoint, List<DesignLibraryModelPointVO> mannequins) { String switchCategory, DesignLibraryModelPointVO designLibraryModelPoint, DesignLibraryModelPointVO designLibraryModelPointVO) {
DesignPythonBasic basic = new DesignPythonBasic(); DesignPythonBasic basic = new DesignPythonBasic();
basic.setSingle_overall(singleOverall); basic.setSingle_overall(singleOverall);
basic.setSwitch_category(switchCategory); basic.setSwitch_category(switchCategory);
@@ -2180,9 +2188,8 @@ public class PythonService {
basic.setSelf_template(Boolean.TRUE); basic.setSelf_template(Boolean.TRUE);
} }
basic.setScale_earrings(0.16); basic.setScale_earrings(0.16);
if (!CollectionUtils.isEmpty(mannequins)) { if (Objects.nonNull(designLibraryModelPointVO)) {
DesignLibraryModelPointVO vo = getRandomDesignLibraryModelPoint(mannequins); basic.setBody_point_test(getMap(designLibraryModelPointVO));
basic.setBody_point_test(getMap(vo));
}else { }else {
basic.setBody_point_test(getMap(designLibraryModelPoint)); basic.setBody_point_test(getMap(designLibraryModelPoint));
} }