ClassEqualizerForTensorLabels.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "ClassEqualizerForTensorImages.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class CTensor;
 
        template <typename T>
        class FL_EXPORT CClassEqualizerForTensorLabels : public CClassEqualizerForTensorImages<T>
        {
        public:
            CClassEqualizerForTensorLabels();
            CClassEqualizerForTensorLabels(const CClassEqualizerForTensorLabels& lrs);
            CClassEqualizerForTensorLabels(const CClassEqualizerForTensorLabels* pLrs);
            virtual ~CClassEqualizerForTensorLabels();
 
            virtual const CResult Assign(const CClassEqualizerForTensorLabels& lrs);
            virtual const CResult Assign(const CClassEqualizerForTensorLabels* pLrs);
 
            virtual const CResult Initialize() override;
 
            virtual const CResult ArrangeBatch(int64_t i64BatchSize, int64_t i64SubMiniBatchSize, std::vector<int64_t>& vctResultBatchIndices, int64_t i64TotalIteration = 0) override;
 
            DeclareGetClassType();
            SupportToDuplicateObject(CClassEqualizerForTensorLabels, *this);
 
        protected:
            Base::CFLArray<int64_t> m_flaClassCount;
            Base::CFLArray<double> m_flaClassCountRatio;
            Base::CFLArray<int64_t> m_flaAccumulateClassCount;
 
        private:
            template<typename T>
            friend class COptimizer;
        };
 
 
        typedef CClassEqualizerForTensorLabels<float> CClassEqualizerForTensorLabelsF;
        typedef CClassEqualizerForTensorLabels<double> CClassEqualizerForTensorLabelsD;
 
        typedef CClassEqualizerForTensorLabels<float> CCELabelsF;
        typedef CClassEqualizerForTensorLabels<double> CCELabelsD;
 
        template <typename T>
        using CCELabels = CClassEqualizerForTensorLabels<T>;
    }
}
 
#endif