ClassEqualizerForTensorImages.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "ClassEqualizerBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class CTensor;
 
        template <typename T>
        class FL_EXPORT CClassEqualizerForTensorImages : public CClassEqualizerBase<T>
        {
        public:
            CClassEqualizerForTensorImages();
            CClassEqualizerForTensorImages(const CClassEqualizerForTensorImages& lrs);
            CClassEqualizerForTensorImages(const CClassEqualizerForTensorImages* pLrs);
            virtual ~CClassEqualizerForTensorImages();
 
            virtual const CResult Assign(const CClassEqualizerForTensorImages& lrs);
            virtual const CResult Assign(const CClassEqualizerForTensorImages* 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(CClassEqualizerForTensorImages, *this);
 
        protected:
            std::vector<std::vector<std::pair<int64_t, int64_t>>>& m_vctLabelDistributionForTiling;
 
        private:
            template<typename T>
            friend class COptimizer;
        };
 
 
        typedef CClassEqualizerForTensorImages<float> CClassEqualizerForTensorImagesF;
        typedef CClassEqualizerForTensorImages<double> CClassEqualizerForTensorImagesD;
 
        typedef CClassEqualizerForTensorImages<float> CCEImagesF;
        typedef CClassEqualizerForTensorImages<double> CCEImagesD;
 
        template <typename T>
        using CCEImages = CClassEqualizerForTensorImages<T>;
    }
}
 
#endif