BackendLabel.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CTensorImagesWithMaskLabels;
 
        template <typename T>
        class FL_EXPORT CBackendLabel : public CBackendBase<T>
        {
        public:
            CBackendLabel();
            CBackendLabel(const CBackendLabel<T>& bl);
            virtual ~CBackendLabel();
 
            virtual const CResult LabelForSegmentation(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, bool bMaskingMode = false);
            virtual const CResult LabelForClassifier(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, int64_t i64ClassCount);
            virtual const CResult LabelForObjectDetection(const CTensor<T>* pTsrOperand, std::vector<CTensor<T>>& vctResultBoxes, std::vector<CTensor<T>>& vctResultLabels, bool bCuda, bool bRevertAngle = false);
            virtual const CResult LabelForInstanceSegmentationMask(const CTensor<T>* pTsrOperand, int64_t i64MaskWidth, int64_t i64MaskHeight, std::vector<CTensor<T>>& vctResultMasks);
            virtual const CResult LabelForInstanceSegmentation(const CTensor<T>* pTsrOperand, int64_t i64MaskWidth, int64_t i64MaskHeight, std::vector<CTensor<T>>& vctResultBoxesShared, std::vector<CTensor<T>>& vctResultLabelsShared, std::vector<CTensor<T>>& vctResultMasksShared, int64_t i64DeviceIndex, int64_t i64MemoryIndex);
            virtual const CResult LabelShapeForInstanceSegmentation(const CTensor<T>* pTsrOperand, int64_t i64MaskWidth, int64_t i64MaskHeight, std::vector<std::vector<int64_t>>& vctResultBoxes, std::vector<std::vector<int64_t>>& vctResultLabels, std::vector<std::vector<int64_t>>& vctResultMasks);
            virtual const CResult LabelForInstanceSegmentation(const CTensor<T>* pTsrOperand, int64_t i64MaskWidth, int64_t i64MaskHeight, std::vector<CTensor<T>>& vctResultBoxes, std::vector<CTensor<T>>& vctResultLabels, std::vector<CTensor<T>>& vctResultMasks);
            virtual const CResult LabelForStringBasedOCR(const CTensor<T>* pTsrOperand, int64_t i64LongestLabelLength, int64_t i64CharacterCount, const std::map<wchar_t, uint32_t>* pMpCharToClassNum, CTensor<T>* pTsrResult);
            
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendLabel<T>, *this);
 
 
        protected:
            virtual bool ValidLabel(const Foundation::CFLFigure* pFig);
            virtual const CResult LabelForSegmentationMaskImage(const CTensorImagesWithMaskLabels<T>* pTsrOperand, CTensor<T>* pTsrResult);
        };
    }
}
 
#endif