RED-3813: image similarity hotifx #675
@ -34,7 +34,7 @@ public class ImageSimilaritySearchController implements ImageSimilaritySearchRes
|
||||
public ResponseEntity<ImageSimilaritySearchResponse> getSimilarImages(@RequestBody ImageSimilaritySearchRequest imageSimilaritySearchRequest) {
|
||||
|
||||
log.info("received similiar image search request {}", imageSimilaritySearchRequest);
|
||||
List<ImageDocument> similarImages = this.imageSimilarityService.findSimilarImages(imageSimilaritySearchRequest.getCentroId(),
|
||||
List<ImageDocument> similarImages = this.imageSimilarityService.findSimilarImages(imageSimilaritySearchRequest.getAnnotationId(),
|
||||
imageSimilaritySearchRequest.getDistance(),
|
||||
imageSimilaritySearchRequest.getScope());
|
||||
List<String> similarImagesIds = new ArrayList<>();
|
||||
|
||||
@ -58,10 +58,10 @@ public class ImageSimilarityService {
|
||||
}
|
||||
|
||||
|
||||
public List<ImageDocument> findSimilarImages(String centroId, double distance, Scope scope) throws Exception {
|
||||
public List<ImageDocument> findSimilarImages(String annotationId, double distance, Scope scope) throws Exception {
|
||||
|
||||
ImageDocument centroImage = this.imageMongoService.findById(centroId);
|
||||
log.info("image received with id {}: {}", centroId, centroImage);
|
||||
ImageDocument centroImage = this.imageMongoService.findById(annotationId);
|
||||
log.info("image received with id {}: {}", annotationId, centroImage);
|
||||
List<ImageDocument> similarImages = this.imageMongoService.findSimilarImages(centroImage, distance, scope);
|
||||
log.info("received similar images: {}", similarImages);
|
||||
return similarImages.stream()
|
||||
|
||||
@ -20,7 +20,7 @@ public class ImageSimilaritySearchRequest {
|
||||
|
||||
@NonNull
|
||||
@Schema(description = "id of the central image")
|
||||
private String centroId;
|
||||
private String annotationId;
|
||||
@NonNull
|
||||
@Schema(description = "the manhattan distance used in the similiarity search")
|
||||
private double distance;
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user