Commit 0b3b7f57ee036fe3595d42984e519bd5b1601e70
1 parent
e7875eeb
faceFeature
Showing
1 changed file
with
17 additions
and
1 deletions
Bsth-admin/src/main/java/com/ruoyi/service/impl/dss/FaceServiceImpl.java
| @@ -48,7 +48,23 @@ public class FaceServiceImpl implements FaceService { | @@ -48,7 +48,23 @@ public class FaceServiceImpl implements FaceService { | ||
| 48 | return null; | 48 | return null; |
| 49 | } | 49 | } |
| 50 | 50 | ||
| 51 | - return null; | 51 | + ImageInfo sourceImage = new ImageInfo(); |
| 52 | + | ||
| 53 | + byte[] sourceImageData = Base64.decode(sourceDriver.getImage()); | ||
| 54 | + sourceImage.setImageData(sourceImageData); | ||
| 55 | + FaceFeature sourceFaceFeature = generateFaceFeature(sourceDriver, sourceImage); | ||
| 56 | + if (Objects.isNull(sourceFaceFeature)) { | ||
| 57 | + return null; | ||
| 58 | + } | ||
| 59 | + | ||
| 60 | + | ||
| 61 | + Optional<NewDriver> optional = drivers.stream().map(dr -> { | ||
| 62 | + float imageScore = checkFaceScore(dr, sourceFaceFeature); | ||
| 63 | + dr.setImageScore(imageScore); | ||
| 64 | + | ||
| 65 | + return dr; | ||
| 66 | + }).max(Comparator.comparing(NewDriver::getImageScore)); | ||
| 67 | + return optional.isPresent() ? optional.get() : null; | ||
| 52 | } | 68 | } |
| 53 | 69 | ||
| 54 | private float checkFaceScore(NewDriver dr, FaceFeature sourceFaceFeature) { | 70 | private float checkFaceScore(NewDriver dr, FaceFeature sourceFaceFeature) { |