//
// Created by DefTruth on 2021/8/8.
//

#ifndef LITE_AI_PIPELINE_H
#define LITE_AI_PIPELINE_H

#include "models.h"


// Pipeline namespace
namespace lite
{
  namespace cv
  {
    namespace pipeline
    {
      // VideoPipeline: video demos for general models, such Face Detection
      // + Face landmarks Detection | Face Attributes Analysis.
      class LITE_EXPORTS VideoPipeline;

      // FaceTracker: Face tracking using face detection and IoU.
      class LITE_EXPORTS FaceTracker;

      // ObjectTracker: Face tracking using object detection and IoU | DeepSort.
      class LITE_EXPORTS ObjectTracker;

      // FaceRecognizer: Face ID recognition using face detection.
      // face alignment and recognition.
      class LITE_EXPORTS FaceRecognizer;

      // FaceAttributesAnalyser: Face Attributes Analysis using face detection,
      // Face Attributes estimation.
      class LITE_EXPORTS FaceAttributesAnalyser;

      // VideoStyleTransfer: Style transfer for video demo.
      class LITE_EXPORTS VideoStyleTransfer;

      // VideoColorizer: colorization for video demo.
      class LITE_EXPORTS VideoColorizer;
    }
  }
}

namespace lite
{
  namespace cv
  {
    namespace pipeline
    {
      enum MODEL
      {
        FSANet = 0,
        PFLD = 1,
        UltraFace = 2,
        AgeGoogleNet = 3,
        GenderGoogleNet = 4
      };

    }
  }
}


#endif //LITE_AI_PIPELINE_H