ComputationalGraphAugmentationMaskOperationForSegmentation.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "Tensor.h"
#include "BackendLabel.h"
#include "ComputationalGraphAugmentationBase.h"
 
namespace FLImaging
{
    namespace AI
    {
 
        class CAugmentationParameters;
        template<typename T>
        class CInternalAugmentationMaskOperationForSegmentation;
 
        template <typename T>
        class FL_EXPORT CComputationalGraphAugmentationMaskOperationForSegmentation : public CComputationalGraphAugmentationBase<T>
        {
        private:
            CComputationalGraphAugmentationMaskOperationForSegmentation();
 
        protected:
            CComputationalGraphAugmentationMaskOperationForSegmentation(const CComputationalGraphAugmentationMaskOperationForSegmentation<T>& cg);
 
        public:
 
            CComputationalGraphAugmentationMaskOperationForSegmentation(const CComputationalBase<T>& cbOperand, const CAugmentationParameters& apParams);
            CComputationalGraphAugmentationMaskOperationForSegmentation(const CComputationalBase<T>& cbOperand, const CAugmentationParameters* pApParams);
            virtual ~CComputationalGraphAugmentationMaskOperationForSegmentation();
 
            virtual CComputationalBase<T>* Clone() const override;
            virtual const CTensor<T>& GetMaskTensor() const;
            virtual CTensor<T>& Forward() override;
 
            virtual int64_t GetRequiredDedicatedMemory(bool bTraining = false, bool bRecursively = true, int64_t i64BatchSize = 1) const override;
 
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphAugmentationMaskOperationForSegmentation, *this);
 
        protected:
            CTensor<T> m_tsrMaskTmpBuffer;
            CTensor<T> m_tsrMask;
            CBackendLabel<T> m_backendLabel;
        public:
            DeclareGetSignletonObject(CComputationalGraphAugmentationMaskOperationForSegmentation);
 
 
        private:
            template <typename T> friend class CInternalAugmentationMaskOperationForSegmentation;
        };
 
        #define CCGFAugmentationMaskOPSS(...) (*(new CComputationalGraphAugmentationMaskOperationForSegmentation<float>(__VA_ARGS__)))
        #define CCGDAugmentationMaskOPSS(...) (*(new CComputationalGraphAugmentationMaskOperationForSegmentation<double>(__VA_ARGS__)))
 
        #define CCGTAugmentationMaskOPSS(T, ...) (*(new CComputationalGraphAugmentationMaskOperationForSegmentation<T>(__VA_ARGS__)))
    }
}
 
#endif