ValidatorForSemanticSegmentation.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "Tensor.h"
#include "ValidatorBase.h"
#include "ValidatorForObjectDetection.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CValidatorForSemanticSegmentation : public CValidatorBase<T>
        {
 
        public:
            CValidatorForSemanticSegmentation();
            CValidatorForSemanticSegmentation(const CValidatorForSemanticSegmentation<T>& validator);
            CValidatorForSemanticSegmentation(const CValidatorForSemanticSegmentation<T>* pValidator);
            virtual ~CValidatorForSemanticSegmentation();
 
            const CValidatorForSemanticSegmentation<T>& operator=(const CValidatorForSemanticSegmentation<T>& validator);
 
            virtual const CResult Assign(const CValidatorForSemanticSegmentation<T>& validator);
            virtual const CResult Assign(const CValidatorForSemanticSegmentation<T>* pValidator);
 
            virtual const CResult Clear() override;
 
            virtual T Validate(int32_t* pI32IterationFeedback = nullptr) override;
 
            static const CResult CalculateMeanIoU(const int64_t* pTConfusionMatrix, T& tResultMeanIoU, T& tResultMeanIoUZE, int64_t i64ClassCount);
 
            virtual T GetLastPa() const;
            virtual T GetLastMeanIoU() const;
            
            virtual T GetLastPaZE() const;
            virtual T GetLastMeanIoUZE() const;
            virtual T GetLastMeanAP() const;
            virtual const CResult SetIgnoreIndex(int64_t i64IgnoreIndex);
            virtual int64_t GetIgnoreIndex() const;
            virtual int64_t GetZETotalCount() const;
            
            const CTensor<double>& GetConfusionMatrix() const;
            
            virtual const std::vector< CDetectionInfo<T>>& GetDetectionInstances();
 
            virtual const CResult SetIoUThreshold(T tThreshold);
            virtual T GetIoUThreshold() const;
 
            virtual const CResult SetConfidenceScoreThreshold(T tThreshold);
            virtual T GetConfidenceScoreThreshold() const;
 
            virtual const std::vector<int64_t>& GetTruthClassesCount() const;
            virtual int64_t GetUniqueTruthCount() const;
 
            virtual const CResult EnableMeanAP(bool bEnable = false);
            virtual bool IsMeanAPEnabled() const;
 
            DeclareGetClassType();
            SupportToDuplicateObject(CValidatorForSemanticSegmentation<T>, *this);
 
        protected:
            T m_tLastPa;
            T m_tLastMeanIoU;
 
            T m_tLastPaZE;
            T m_tLastMeanIoUZE;
 
            T m_tLastMeanAP;
 
            int64_t m_i64IgnoreIndex;
            int64_t m_i64ZETotalCount;
 
            CTensor<double> m_tsrCM;
            std::vector<CDetectionInfo<T>>& m_vctDetectionInstances;
            std::vector<int64_t>& m_vctTruthClassesCount;
            int64_t m_i64UniqueTruthCount;
            T m_tIoUThrehsold;
            T m_tConfidenceScoreThreshold;
 
            bool m_bEnableMeanAP;
 
        private:
            template<typename T>
            friend class COptimizer;
 
            template<typename T>
            friend class COptimizerForSemanticSegmentation;
        };
 
 
 
        typedef CValidatorForSemanticSegmentation<float> CValidatorForSemanticSegmentationF;
        typedef CValidatorForSemanticSegmentation<double> CValidatorForSemanticSegmentationD;
    }
}
 
#endif