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