faceFeature

liujun001
1 parent e7875eeb
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 {
             return null;
         }
  
-        return null;
+        ImageInfo sourceImage = new ImageInfo();
+
+        byte[] sourceImageData = Base64.decode(sourceDriver.getImage());
+        sourceImage.setImageData(sourceImageData);
+        FaceFeature sourceFaceFeature = generateFaceFeature(sourceDriver, sourceImage);
+        if (Objects.isNull(sourceFaceFeature)) {
+            return null;
+        }
+
+
+        Optional<NewDriver> optional = drivers.stream().map(dr -> {
+            float imageScore = checkFaceScore(dr, sourceFaceFeature);
+            dr.setImageScore(imageScore);
+
+            return dr;
+        }).max(Comparator.comparing(NewDriver::getImageScore));
+        return optional.isPresent() ? optional.get() : null;
     }
  
     private float checkFaceScore(NewDriver dr, FaceFeature sourceFaceFeature) {