ValidatorForInstanceSegmentation.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "ValidatorBase.h"
#include "ValidatorForObjectDetection.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CValidatorForInstanceSegmentation : public CValidatorBase<T>
        {
 
        public:
            CValidatorForInstanceSegmentation();
            CValidatorForInstanceSegmentation(const CValidatorForInstanceSegmentation<T>& validator);
            CValidatorForInstanceSegmentation(const CValidatorForInstanceSegmentation<T>* pValidator);
            virtual ~CValidatorForInstanceSegmentation();
 
            const CValidatorForInstanceSegmentation<T>& operator=(const CValidatorForInstanceSegmentation<T>& validator);
 
            virtual const CResult Assign(const CValidatorForInstanceSegmentation<T>& validator);
            virtual const CResult Assign(const CValidatorForInstanceSegmentation<T>* pValidator);
 
            virtual const CResult SetNumOfClasses(int64_t i64ClassCount);
            virtual int64_t GetNumOfClasses() const;
 
            virtual const CResult SetObjectnessThreshold(T tObjectnessThreshold = .005);
            virtual T GetObjectnessThreshold() const;
 
            virtual const CResult SetIoUThreshold(T tIoUThreshold = .5);
            virtual T GetIoUThreshold() const;
 
            virtual const CResult SetMaskThreshold(T tMaskThreshold);
            virtual T GetMaskThreshold() const;
 
            virtual const CResult SetNMSThreshold(T tNMSThreshold);
            virtual T GetNMSThreshold() const;
 
            virtual const CResult EnableIOUThresholdUpdate(bool bIOUThresholdUpdate = true);
            virtual bool IsIOUThresholdUpdateEnabled() const;
 
            virtual T GetLastValidation() const;
 
            virtual T Validate(int32_t* pI32IterationFeedback = nullptr) override;
 
            virtual const std::vector<T>& GetRecallPerClass() const;
            virtual const std::vector<T>& GetPrecisionPerClass() const;
            virtual const std::vector<T>& GetAPPerClass() const;
 
            virtual const int64_t* GetTruthClassesCount() const ;
            virtual const std::vector< CDetectionInfo<T>>& GetDetectionInstances();
            virtual int64_t GetUniqueTruthCount() const;
 
            DeclareGetClassType();
            SupportToDuplicateObject(CValidatorForInstanceSegmentation<T>, *this);
 
        protected:
            virtual const CResult ResetComputationPerClass();
 
            T m_tObjectnessThreshold;
            T m_tIoUThreshold;
            T m_tMaskThreshold;
            T m_tNMSThreshold;
            T m_tLastValidation;
 
            int64_t m_i64TruthLabelSize;
            T* m_pTTruthLabels;
 
            int64_t m_i64ClassCount;
            int64_t* m_pI64TruthClassesCount;
    
            int64_t m_i64UniqueTruthCount;
            int64_t* m_pI64TruthFlag;
 
            bool m_bIOUThresholdUpdate;
            
            std::vector<T>& m_vctPrecisionPerClass;
            std::vector<T>& m_vctRecallPerClass;
            std::vector<T>& m_vctAPPerClass;
            std::vector<CDetectionInfo<T>>& m_vctDetectionInstances;
        };
 
 
        typedef CValidatorForInstanceSegmentation<float> CValidatorForInstanceSegmentationF;
        typedef CValidatorForInstanceSegmentation<double> CValidatorForInstanceSegmentationD;
    }
}
 
#endif