#ifndef MASKELEMENT_H
#define MASKELEMENT_H

#include "layer.h"
#include "net.h"

#if defined(USE_NCNN_SIMPLEOCV)
#include "simpleocv.h"
#else
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#endif
#include <float.h>
#include <stdio.h>
#include <vector>

#include <iak/akelement.h>
#include <akvideopacket.h>

class MaskElementPrivate;

class MaskElement : public AkElement
{
    Q_OBJECT
    public:
        MaskElement();
        ~MaskElement();

        AkPacket iVideoStream(const AkVideoPacket &packet) override;

    private:
        MaskElementPrivate *d;
};

#endif // MASKELEMENT_H