IDeepFakeEstimator.h

#pragma once

#include <fsdk/IObject.h>
#include <fsdk/FSDKError.h>
#include <fsdk/Optional.h>
#include <fsdk/Types.h>

namespace fsdk {
    namespace experimental {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IDeepFakeEstimator);
#endif

        enum class DeepFakeMode {
            Default,  // Specified in config file.
            M1,       // M1.
            M2        // M2.
        };

        struct DeepFakeEstimation {
            enum class State {
                Real = 0,    
                Fake         
            };

            float score;    
            State state;    
        };

        struct IDeepFakeEstimator : IRefCounted {
            virtual Result<FSDKError> estimate(
                const Image& image,
                const Detection& detection,
                DeepFakeEstimation& estimation) const noexcept = 0;

            virtual Result<FSDKError> estimate(
                Span<const Image> images,
                Span<const Detection> detections,
                Span<DeepFakeEstimation> estimations) const noexcept = 0;

            virtual Result<FSDKError>
            validate(
                Span<const Image> images,
                Span<const Detection> detections,
                Span<Result<FSDKError>> errors) const noexcept = 0;
        };
    } // namespace experimental
} // namespace fsdk