ObjectAugmentationParameters.h

#pragma once
 
#include "DefinitionsAI.h"
 
#ifdef _M_X64

 
#pragma pack(push, 1)

namespace FLImaging
{
    namespace AI
    {
        class FL_EXPORT CObjectAugmentationParameters : public Base::CFLBase
        {
        public:
            
            enum EObjectAugmentation_OperationSource
            {
                EObjectAugmentation_OperationSource_Self = 0,
                EObjectAugmentation_OperationSource_LearningDataset,
                EObjectAugmentation_OperationSource_UserDefined,
            };
 
        public:

            CObjectAugmentationParameters();
            CObjectAugmentationParameters(const CObjectAugmentationParameters& ap);
            CObjectAugmentationParameters(const CObjectAugmentationParameters* pAp);

            virtual const CResult Assign(const CObjectAugmentationParameters& ap);
            virtual const CResult Assign(const CObjectAugmentationParameters* pAp);

            const CObjectAugmentationParameters& operator=(const CObjectAugmentationParameters& ap);
            virtual const CResult Clear();
 
            virtual const CResult SetCommonActivationRate(double f64ActivationRate = 1.);
            virtual const CResult GetCommonActivationRate(double& f64ActivationRate) const;
 
            virtual const CResult SetCommonIoUThreshold(double f64IoUThreshold);

            virtual const CResult GetCommonIoUThreshold(double& f64IoUThreshold) const;
 
            virtual const CResult EnableAugmentation(bool bEnable);
            virtual bool IsAugmentationEnabled() const;

            virtual const CResult SetObjectAugmentationOperationSourceMethod(EObjectAugmentation_OperationSource eObjectAugmentationOperationSourceMethod = EObjectAugmentation_OperationSource_LearningDataset);
            virtual const CResult GetObjectAugmentationOperationSourceMethod(EObjectAugmentation_OperationSource& eObjectAugmentationOperationSourceMethod) const;

            virtual const CResult SetObjectAugmentationUserDefinedImage(const Base::CFLImage& fliObjectAugmentationUserDefinedImage);
            virtual const CResult SetObjectAugmentationUserDefinedImage(const Base::CFLImage* pFliObjectAugmentationUserDefinedImage);
            virtual const Base::CFLImage* GetObjectAugmentationUserDefinedImage() const;

            virtual const CResult SetObjectCountParam(int64_t i64MinCount = 1, int64_t i64MaxCount = 2);
            virtual const CResult GetObjectCountParam(int64_t& i64MinCount, int64_t& i64MaxCount) const;

            virtual const CResult SetExcludedClassNumbers(const wchar_t* pWcharExludedClassNumbers);

            virtual const wchar_t* GetExcludedClassNumbers() const;

            virtual const CResult EnableTranslation(bool bTranslation);
            virtual bool IsTranslationEnabled() const;
            virtual const CResult SetTranslationParam(double f64MinTranslationRatioX = -0.1, double f64MaxTranslationRatioX = 0.1, double f64MinTranslationRatioY = -0.1, double f64MaxTranslationRatioY = 0.1, double f64TranslationActivationRate = 1.);
            virtual const CResult GetTranslationParam(double& f64MinTranslationRatioX, double& f64MaxTranslationRatioX, double& f64MinTranslationRatioY, double& f64MaxTranslationRatioY, double& f64TranslationActivationRate) const;

            virtual const CResult EnableHorizontalFlip(bool bFlip);
            virtual bool IsHorizontalFlipEnabled() const;
            virtual const CResult SetHorizontalFlipParam(double f64HorizontalFlipActivationRate = 1.);
            virtual const CResult GetHorizontalFlipParam(double& f64HorizontalFlipActivationRate) const;
            virtual const CResult EnableVerticalFlip(bool bFlip);
            virtual bool IsVerticalFlipEnabled() const;
            virtual const CResult SetVerticalFlipParam(double f64VerticalFlipActivationRate = 1.);
            virtual const CResult GetVerticalFlipParam(double& f64VerticalFlipActivationRate) const;

            virtual const CResult EnableAdjustment(bool bAdjustment);
            virtual bool IsAdjustmentEnabled() const;
            virtual const CResult SetAdjustmentParam(double f64MinAdjustmentBrightness = .8, double f64MaxAdjustmentBrightness = 1.25, double f64MinAdjustmentContrast = .8, double f64MaxAdjustmentContrast = 1.25, double f64AdjustmentActivationRate = 1.);
            virtual const CResult GetAdjustmentParam(double& f64MinAdjustmentBrightness, double& f64MaxAdjustmentBrightness, double& f64MinAdjustmentContrast, double& f64MaxAdjustmentContrast, double& f64AdjustmentActivationRate) const;

            virtual const CResult EnableIntensity(bool bIntensity);
            virtual bool IsIntensityEnabled() const;
            virtual const CResult SetIntensityParam(double f64MinSaturation = .8, double f64MaxSaturation = 1.25, double f64MinExposure = .8, double f64MaxExposure = 1.25, double f64MinHue = -0.03, double f64MaxHue = .03, double f64IntensityActivationRate = 1.);
            virtual const CResult GetIntensityParam(double& f64MinSaturation, double& f64MaxSaturation, double& f64MinExposure, double& f64MaxExposure, double& f64MinHue, double& f64MaxHue, double& f64IntensityActivationRate) const;

            virtual const CResult EnablePerspective(bool bPerspective);
            virtual bool IsPerspectiveEnabled() const;
            virtual const CResult SetPerspectiveParam(double f64MinPerspectiveRatio = .0, double f64MaxPerspectiveRatio = .1, double f64PerspectiveActivationRate = 1.);
            virtual const CResult GetPerspectiveParam(double& f64MinPerspectiveRatio, double& f64MaxPerspectiveRatio, double& f64PerspectiveActivationRate) const;

            virtual const CResult EnableQuarterRotation(bool bQuarterRotation);
            virtual bool IsQuarterRotationEnabled() const;
            virtual const CResult SetQuarterRotationParam(bool bAngle0 = true, bool bAngle90 = true, bool bAngle180 = true, bool bAngle270 = true, double f64QuarterRotationActivationRate = 1.);
            virtual const CResult GetQuarterRotationParam(bool& bAngle0, bool& bAngle90, bool& bAngle180, bool& bAngle270, double& f64QuarterRotationActivationRate) const;

            virtual const CResult EnableRotation(bool bRotation);
            virtual bool IsRotationEnabled() const;
            virtual const CResult SetRotationParam(double f64MinAngle = -5., double f64MaxAngle = 5., double f64RotationActivationRate = 1.);
            virtual const CResult GetRotationParam(double& f64MinAngle, double& f64MaxAngle, double& f64RotationActivationRate) const;

            virtual const CResult EnableScale(bool bScale);
            virtual bool IsScaleEnabled() const;
            virtual const CResult SetScaleParam(double f64MinScaleRatioX = 0.8, double f64MaxScaleRatioX = 1.2, double f64MinScaleRatioY = 0.8, double f64MaxScaleRatioY = 1.2, bool bPreservingScaleRatio = true, double f64ScaleActivationRate = 1.);
            virtual const CResult GetScaleParam(double& f64MinScaleRatioX, double& f64MaxScaleRatioX, double& f64MinScaleRatioY, double& f64MaxScaleRatioY, bool& bPreservingScaleRatio, double& f64ScaleActivationRate) const;
 
            virtual const CResult GetBinaryData(Base::CFLData& fldBinary, int32_t i32Version = -1) const;
            virtual const CResult GetBinaryData(Base::CFLData* pFldBinary, int32_t i32Version = -1) const;
 
            virtual const CResult SetBinaryData(const Base::CFLData& fldBinary);
            virtual const CResult SetBinaryData(const Base::CFLData* pFldBinary);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CObjectAugmentationParameters, *this);
 
        protected:
            bool m_bAugmentation;
            double m_f64CommonActivationRate;
            double m_f64CommonIoUThreshold;
            EObjectAugmentation_OperationSource m_eObjectAugmentationOperationSourceMethod;

            Base::CFLImage* m_pFliObjectAugmentationUserDefinedImage;

            int64_t m_i64MinCount;
            int64_t m_i64MaxCount;
            Base::CFLString<wchar_t> m_flsExludedClassNumbers;
            bool m_bAugmentationHorizontalFlip;
            double m_f64HorizontalFlipActivationRate;
            bool m_bVerticalFlip;
            double m_f64VerticalFlipActivationRate;

            bool m_bAdjustment;
            double m_f64AdjustmentActivationRate;
            double m_f64MinAdjustmentBrightness;
            double m_f64MaxAdjustmentBrightness;
            double m_f64MinAdjustmentContrast;
            double m_f64MaxAdjustmentContrast;

            bool m_bIntensity;
            double m_f64IntensityActivationRate;
            double m_f64MinSaturation;
            double m_f64MaxSaturation;
            double m_f64MinExposure;
            double m_f64MaxExposure;
            double m_f64MinHue;
            double m_f64MaxHue;

            bool m_bPerspective;
            double m_f64PerspectiveActivationRate;
            double m_f64MinPerspectiveRatio;
            double m_f64MaxPerspectiveRatio;

            bool m_bQuarterRotation;
            double m_f64QuarterRotationActivationRate;
            bool m_bQuarterRotationAngle0;
            bool m_bQuarterRotationAngle90;
            bool m_bQuarterRotationAngle180;
            bool m_bQuarterRotationAngle270;

            bool m_bRotation;
            double m_f64RotationActivationRate;
            double m_f64MinAngle;
            double m_f64MaxAngle;
            
            bool m_bScale;
            double m_f64ScaleActivationRate;
            double m_f64MinScaleRatioX;
            double m_f64MaxScaleRatioX;
            double m_f64MinScaleRatioY;
            double m_f64MaxScaleRatioY;
            bool m_bPreservingScaleRatio;

            bool m_bTranslation;
            double m_f64TranslationActivationRate;
            double m_f64MinTranslationRatioX;
            double m_f64MaxTranslationRatioX;
            double m_f64MinTranslationRatioY;
            double m_f64MaxTranslationRatioY;
        };
    }
}
 
#pragma pack(pop)
#endif