ClassEqualizerBase.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class CTensor;
 
        template <typename T>
        class FL_EXPORT CClassEqualizerBase : public Base::CFLBase
        {
        public:
            CClassEqualizerBase();
            CClassEqualizerBase(const CClassEqualizerBase& lrs);
            CClassEqualizerBase(const CClassEqualizerBase* pLrs);
            virtual ~CClassEqualizerBase();
 
            virtual const CResult Assign(const CClassEqualizerBase& lrs);
            virtual const CResult Assign(const CClassEqualizerBase* pLrs);
 
            virtual const CResult Clear();
            virtual const CResult Initialize();
 
            virtual const CResult SetBatchTensor(CTensor<T>* pTsrBatchTensor);
            virtual const CTensor<T>* GetBatchTensor() const;
 
            virtual const CResult EnableClassEqaulization(bool bEnable);
            virtual bool IsClassEqaulizationEnabled() const;
 
            virtual const CResult ArrangeBatch(int64_t i64BatchSize, int64_t i64SubMiniBatchSize, std::vector<int64_t>& vctResultBatchIndices, int64_t i64TotalIteration = 0);
 
            virtual const std::vector<int64_t>& GetOriginOrders() const;
 
 
            DeclareGetClassType();
            SupportToDuplicateObject(CClassEqualizerBase, *this);
        protected:
            virtual void Throw(const CResult& res, const wchar_t* pWcsExtraMessage = nullptr) const override;
 
        protected:
            CTensor<T>* m_pTsrBatchTensor;
            std::vector<int64_t>& m_vctLearnOrders;
            bool m_bClassEqualization;
 
        private:
            template<typename T>
            friend class COptimizer;
        };
 
 
        typedef CClassEqualizerBase<float> CClassEqualizerBaseF;
        typedef CClassEqualizerBase<double> CClassEqualizerBaseD;
 
        typedef CClassEqualizerBase<float> CCEBaseF;
        typedef CClassEqualizerBase<double> CCEBaseD;
 
        template <typename T>
        using CCEBase = CClassEqualizerBase<T>;
    }
}
 
#endif