java/api/src/main/java/com/exadel/frs/core/trainservice/component/FaceClassifierPredictor.java
<reponame>rohankumardubey/CompreFace<gh_stars>0/*
 * Copyright (c) 2020 the original author or authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

package com.exadel.frs.core.trainservice.component;

import com.exadel.frs.core.trainservice.component.classifiers.Classifier;
import java.util.List;
import java.util.UUID;

import lombok.RequiredArgsConstructor;
import org.apache.commons.lang3.tuple.Pair;
import org.springframework.stereotype.Component;

@Component
@RequiredArgsConstructor
public class FaceClassifierPredictor {

    private final Classifier classifier;

    public List<Pair<Double, String>> predict(final String modelKey, final double[] input, final int resultCount) {
        return classifier.predict(input, modelKey, resultCount);
    }

    public Double verify(final String modelKey, final double[] input, final UUID embeddingId) {
        return classifier.verify(input, modelKey, embeddingId);
    }

    public double[] verify(final double[] sourceImageEmbedding, final double[][] targetImageEmbedding) {
        return classifier.verify(sourceImageEmbedding, targetImageEmbedding);
    }
}