66 lines
1.4 KiB
C
66 lines
1.4 KiB
C
|
//
|
||
|
|
// 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
|