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

TASK:模块化；

Browse Source
This commit is contained in:
shahaibo
2025-03-28 11:05:45 +08:00
parent bb00d9a714
commit 63513df053
4 changed files with 38 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java
View File

@@ -163,7 +163,9 @@ public class LibraryServiceImpl extends ServiceImpl<LibraryMapper, Library> impl
QueryWrapper<BrandRelLibrary> qw = new QueryWrapper<>();
qw.lambda().eq(BrandRelLibrary::getBrandId, query.getBrandId());
Set<Long> collect = brandRelLibraryMapper.selectList(qw).stream().map(BrandRelLibrary::getLibraryId).collect(Collectors.toSet());
queryWrapper.in("id", collect);
if (!CollectionUtils.isEmpty(collect)) {
queryWrapper.in("id", collect);
}
}
// 新增分类过滤

17
src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
View File

@@ -1169,14 +1169,15 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
// 提取sketch
JSONObject jsonObject = pythonService.segProduct(url);
String clothCategory = pythonService.getClothCategory(url, gender);
JSONObject attributeRecognition = pythonService.getAttributeRecognition(url, clothCategory, gender);
JSONObject data = attributeRecognition.getJSONObject("data");
JSONObject attrDict = ((JSONObject) data.getJSONArray("list").get(0)).getJSONObject("attr_dict");
AttributeRecognitionJSON attrDictJSON = attrDict.toJavaObject(AttributeRecognitionJSON.class);
ProductImageAttribute productImageAttribute = toAttrDict(attrDictJSON);
productImageAttributeMapper.insert(productImageAttribute);
JSONArray data = jsonObject.getJSONArray("data");
for (int i = 0; i < data.size(); i++) {
JSONObject jsonObject1 = data.getJSONObject(i);
JSONObject attribute = jsonObject1.getJSONObject("attribute");
AttributeRecognitionJSON attrDictJSON = attribute.toJavaObject(AttributeRecognitionJSON.class);
ProductImageAttribute productImageAttribute = toAttrDict(attrDictJSON);
productImageAttribute.setLibraryId(library.getId());
productImageAttributeMapper.insert(productImageAttribute);
}
}
return null;
}