gitee/face-search
1
0
Fork 0
mirror of https://gitee.com/open-visual/face-search.git synced 2025-06-10 19:19:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

update:修复bug

Browse Source
This commit is contained in:
divenswu 2022-03-22 15:41:30 +08:00
parent 6d20773815
commit be0ef664d0
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
face-search-server/src/main/java/com/visual/face/search/server/service/impl/FaceSearchServiceImpl.java
View File

@ -86,9 +86,23 @@ public class FaceSearchServiceImpl extends BaseService implements FaceSearchServ
} }
//结果和人数是否一致 //结果和人数是否一致
List<SearchResult> result = searchResponse.getResult(); List<SearchResult> result = searchResponse.getResult();
if(result.size() != faceInfos.size()){ if(result.size() > 0 && result.size() != faceInfos.size()){
throw new RuntimeException("search result error"); throw new RuntimeException("search result error");
} }
//如数据库中没有任何样本的情况下，会出现异常，这里进行单独处理
if(result.size() == 0 && faceInfos.size() > 0){
List<FaceSearchRepVo> vos = new ArrayList<>();
for(int i=0; i<faceInfos.size(); i++) {
FaceInfo.FaceBox box = faceInfos.get(i).box;
FaceSearchRepVo vo = FaceSearchRepVo.build();
vo.setLocation(FaceLocation.build(box.leftTop.x, box.leftTop.y, box.width(), box.height()));
vo.setFaceScore((float) Math.floor(faceInfos.get(i).score * 1000000) / 10000);
List<SampleFaceVo> match = new ArrayList<>();
vo.setMatch(match);
vos.add(vo);
}
return vos;
}
//获取关联数据ID //获取关联数据ID
boolean needFixFaceId = false; boolean needFixFaceId = false;
Set<Long> faceIds = new HashSet<>(); Set<Long> faceIds = new HashSet<>();