ValidatorForSuperResolution.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "ValidatorBase.h"
#include "Tensor.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CValidatorForSuperResolution : public CValidatorBase<T>
        {
 
        public:
            CValidatorForSuperResolution();
            CValidatorForSuperResolution(const CValidatorForSuperResolution<T>& validator);
            CValidatorForSuperResolution(const CValidatorForSuperResolution<T>* pValidator);
            virtual ~CValidatorForSuperResolution();
 
            const CValidatorForSuperResolution<T>& operator=(const CValidatorForSuperResolution<T>& validator);
 
            virtual const CResult Assign(const CValidatorForSuperResolution<T>& validator);
            virtual const CResult Assign(const CValidatorForSuperResolution<T>* pValidator);
 
            virtual T Validate(int32_t* pI32IterationFeedback = nullptr) override;
            
            const CTensor<T>& GetPSNRTensor() const;
            const CTensor<T>& GetSSIMTensor() const;
 
            DeclareGetClassType();
            SupportToDuplicateObject(CValidatorForSuperResolution<T>, *this);
        protected:
            CTensor<T> m_tsrPSNR;
            CTensor<T> m_tsrSSIM;
 
        private:
            template<typename T>
            friend class COptimizer;
        };
 
        typedef CValidatorForSuperResolution<float> CValidatorForSuperResolutionF;
        typedef CValidatorForSuperResolution<double> CValidatorForSuperResolutionD;
    }
}
 
#endif