SettingsCaffe.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Runtime.Serialization;
 
namespace MyCaffe.basecode
{
    [Serializable]
    public class SettingsCaffe : ISerializable
    {
        DB_VERSION m_imgDbVersion = DB_VERSION.DEFAULT;
        bool m_bEnableLabelBalancing = false;
        bool m_bEnableLabelBoosting = false;
        bool m_bEnableRandomInputSelection = true;
        bool m_bEnablePairInputSelection = false;
        bool m_bUseTrainingSourceForTesting = false;
        bool m_bVerifyDatasetOnLoad = false;
        double m_dfSuperBoostProbability = 0.0;
        int m_nMaximumIterationOverride = -1;
        int m_nTestingIterationOverride = -1;
        string m_strDefaultModelGroup = "";
        string m_strGpuIds = "0";
        DB_LOAD_METHOD m_dbLoadMethod = DB_LOAD_METHOD.LOAD_ON_DEMAND;
        int m_nDbLoadLimit = 0;
        int m_nAutoRefreshScheduledUpdateInMs = 0;
        double m_dfAutoRefreshScheduledReplacementPct = 0;
        bool m_bItemDbLoadDataCriteria = false;
        bool m_bItemDbLoadDebugData = false;
        SNAPSHOT_WEIGHT_UPDATE_METHOD m_snapshotWeightUpdateMethod = SNAPSHOT_WEIGHT_UPDATE_METHOD.FAVOR_ACCURACY;
        SNAPSHOT_LOAD_METHOD m_snapshotLoadMethod = SNAPSHOT_LOAD_METHOD.STATE_BEST_ACCURACY;
        bool m_bSkipMeanCheck = false;
 
        public SettingsCaffe()
        {
        }
 
        public SettingsCaffe(SettingsCaffe s)
        {
            m_imgDbVersion = s.m_imgDbVersion;
            m_bEnableLabelBalancing = s.m_bEnableLabelBalancing;
            m_bEnableLabelBoosting = s.m_bEnableLabelBoosting;
            m_bEnableRandomInputSelection = s.m_bEnableRandomInputSelection;
            m_bEnablePairInputSelection = s.m_bEnablePairInputSelection;
            m_bUseTrainingSourceForTesting = s.m_bUseTrainingSourceForTesting;
            m_bVerifyDatasetOnLoad = s.m_bVerifyDatasetOnLoad;
            m_dfSuperBoostProbability = s.m_dfSuperBoostProbability;
            m_nMaximumIterationOverride = s.m_nMaximumIterationOverride;
            m_nTestingIterationOverride = s.m_nTestingIterationOverride;
            m_strDefaultModelGroup = s.m_strDefaultModelGroup;
            m_strGpuIds = s.m_strGpuIds;
            m_dbLoadMethod = s.m_dbLoadMethod;
            m_nDbLoadLimit = s.m_nDbLoadLimit;
            m_nAutoRefreshScheduledUpdateInMs = s.m_nAutoRefreshScheduledUpdateInMs;
            m_dfAutoRefreshScheduledReplacementPct = s.m_dfAutoRefreshScheduledReplacementPct;
            m_bItemDbLoadDataCriteria = s.m_bItemDbLoadDataCriteria;
            m_bItemDbLoadDebugData = s.m_bItemDbLoadDebugData;
            m_snapshotWeightUpdateMethod = s.m_snapshotWeightUpdateMethod;
            m_snapshotLoadMethod = s.m_snapshotLoadMethod;
            m_bSkipMeanCheck = s.m_bSkipMeanCheck;
        }
 
        public SettingsCaffe(SerializationInfo info, StreamingContext context)
        {
            m_imgDbVersion = (DB_VERSION)getInt(info, "ImageDbVersion", (int)m_imgDbVersion);
            m_bEnableLabelBalancing = getBool(info, "bEnableLabelBalancing", m_bEnableLabelBalancing);
            m_bEnableLabelBoosting = getBool(info, "bEnableLabelBoosting", m_bEnableLabelBoosting);
            m_bEnableRandomInputSelection = getBool(info, "bEnableRandomInputSelection", m_bEnableRandomInputSelection);
            m_bEnablePairInputSelection = getBool(info, "bEnablePairInputSelection", m_bEnablePairInputSelection);
            m_bUseTrainingSourceForTesting = getBool(info, "bUseTrainingSourceForTesting", m_bUseTrainingSourceForTesting);
            m_bVerifyDatasetOnLoad = getBool(info, "bVerifyDatasetOnLoad", m_bVerifyDatasetOnLoad);
            m_dfSuperBoostProbability = getDouble(info, "dfSuperBoostProbability", m_dfSuperBoostProbability);
            m_nMaximumIterationOverride = getInt(info, "nMaximumIterationOverride", m_nMaximumIterationOverride);
            m_nTestingIterationOverride = getInt(info, "nTestingIterationOverride", m_nTestingIterationOverride);
            m_strDefaultModelGroup = info.GetString("strDefaultModelGroup");
            m_strGpuIds = getString(info, "strGpuIds", m_strGpuIds);
            m_dbLoadMethod = (DB_LOAD_METHOD)getInt(info, "ImageDbLoadMethod", (int)m_dbLoadMethod);
            m_nDbLoadLimit = getInt(info, "ImageDbLoadLimit", m_nDbLoadLimit);
            m_nAutoRefreshScheduledUpdateInMs = getInt(info, "ImageDbAutoRefreshScheduledUpdateInMs", m_nAutoRefreshScheduledUpdateInMs);
            m_dfAutoRefreshScheduledReplacementPct = getDouble(info, "ImageDbAutoRefreshReplacementPct", m_dfAutoRefreshScheduledReplacementPct);
            m_bItemDbLoadDataCriteria = getBool(info, "ImageDbLoadDataCriteria", m_bItemDbLoadDataCriteria);
            m_bItemDbLoadDebugData = getBool(info, "ImageDbLoadDebugData", m_bItemDbLoadDebugData);
            m_snapshotWeightUpdateMethod = (SNAPSHOT_WEIGHT_UPDATE_METHOD)getInt(info, "SnapshotWeightUpdateMethod", (int)m_snapshotWeightUpdateMethod);
            m_snapshotLoadMethod = (SNAPSHOT_LOAD_METHOD)getInt(info, "SnapshotLoadMethod", (int)m_snapshotLoadMethod);
            m_bSkipMeanCheck = getBool(info, "SkipMeanCheck", m_bSkipMeanCheck);
        }
 
        private bool getBool(SerializationInfo info, string str, bool bDefault)
        {
            try
            {
                return info.GetBoolean(str);
            }
            catch
            {
                return bDefault;
            }
        }
 
        private double getDouble(SerializationInfo info, string str, double dfDefault)
        {
            try
            {
                return info.GetDouble(str);
            }
            catch
            {
                return dfDefault;
            }
        }
 
        private string getString(SerializationInfo info, string str, string strDefault)
        {
            try
            {
                return info.GetString(str);
            }
            catch
            {
                return strDefault;
            }
        }
 
        private int getInt(SerializationInfo info, string str, int nDefault)
        {
            try
            {
                return info.GetInt32(str);
            }
            catch
            {
                return nDefault;
            }
        }
 
        public void GetObjectData(SerializationInfo info, StreamingContext context)
        {
            info.AddValue("ImageDbVersion", (int)m_imgDbVersion);
            info.AddValue("bEnableLabelBalancing", m_bEnableLabelBalancing);
            info.AddValue("bEnableLabelBoosting", m_bEnableLabelBoosting);
            info.AddValue("bEnableRandomInputSelection", m_bEnableRandomInputSelection);
            info.AddValue("bEnablePairInputSelection", m_bEnablePairInputSelection);
            info.AddValue("bUseTrainingSourceForTesting", m_bUseTrainingSourceForTesting);
            info.AddValue("bVerifyDatasetOnLoad", m_bVerifyDatasetOnLoad);
            info.AddValue("dfSuperBoostProbability", m_dfSuperBoostProbability);
            info.AddValue("nMaximumIterationOverride", m_nMaximumIterationOverride);
            info.AddValue("nTestingIterationOverride", m_nTestingIterationOverride);
            info.AddValue("strDefaultModelGroup", m_strDefaultModelGroup);
            info.AddValue("strGpuIds", m_strGpuIds);
            info.AddValue("ImageDbLoadMethod", (int)m_dbLoadMethod);
            info.AddValue("ImageDbLoadLimit", m_nDbLoadLimit);
            info.AddValue("ImageDbAutoRefreshScheduledUpdateInMs", m_nAutoRefreshScheduledUpdateInMs);
            info.AddValue("ImageDbAutoRefreshReplacementPct", m_dfAutoRefreshScheduledReplacementPct);
            info.AddValue("ImageDbLoadDataCriteria", m_bItemDbLoadDataCriteria);
            info.AddValue("ImageDbLoadDebugData", m_bItemDbLoadDebugData);
            info.AddValue("SnapshotWeightUpdateMethod", (int)m_snapshotWeightUpdateMethod);
            info.AddValue("SnapshotLoadMethod", (int)m_snapshotLoadMethod);
            info.AddValue("SkipMeanCheck", m_bSkipMeanCheck);
        }
 
        public SettingsCaffe Clone()
        {
            SettingsCaffe s = new SettingsCaffe();
 
            s.m_imgDbVersion = m_imgDbVersion;
            s.m_bEnableLabelBalancing = m_bEnableLabelBalancing;
            s.m_bEnableLabelBoosting = m_bEnableLabelBoosting;
            s.m_bEnableRandomInputSelection = m_bEnableRandomInputSelection;
            s.m_bEnablePairInputSelection = m_bEnablePairInputSelection;
            s.m_bUseTrainingSourceForTesting = m_bUseTrainingSourceForTesting;
            s.m_bVerifyDatasetOnLoad = m_bVerifyDatasetOnLoad;
            s.m_dfSuperBoostProbability = m_dfSuperBoostProbability;
            s.m_nMaximumIterationOverride = m_nMaximumIterationOverride;
            s.m_nTestingIterationOverride = m_nTestingIterationOverride;
            s.m_strDefaultModelGroup = m_strDefaultModelGroup;
            s.m_strGpuIds = m_strGpuIds;
            s.m_dbLoadMethod = m_dbLoadMethod;
            s.m_nDbLoadLimit = m_nDbLoadLimit;
            s.m_nAutoRefreshScheduledUpdateInMs = m_nAutoRefreshScheduledUpdateInMs;
            s.m_dfAutoRefreshScheduledReplacementPct = m_dfAutoRefreshScheduledReplacementPct;
            s.m_bItemDbLoadDataCriteria = m_bItemDbLoadDataCriteria;
            s.m_bItemDbLoadDebugData = m_bItemDbLoadDebugData;
            s.m_snapshotWeightUpdateMethod = m_snapshotWeightUpdateMethod;
            s.m_snapshotLoadMethod = m_snapshotLoadMethod;
            s.m_bSkipMeanCheck = m_bSkipMeanCheck;
 
            return s;
        }
 
        public bool VeriyDatasetOnLoad
        {
            get { return m_bVerifyDatasetOnLoad; }
            set { m_bVerifyDatasetOnLoad = value; }
        }
 
        public DB_VERSION DbVersion
        {
            get { return m_imgDbVersion; }
            set { m_imgDbVersion = value; }
        }
 
        public bool EnableLabelBalancing
        {
            get { return m_bEnableLabelBalancing; }
            set { m_bEnableLabelBalancing = value; }
        }
 
        public bool EnableLabelBoosting
        {
            get { return m_bEnableLabelBoosting; }
            set { m_bEnableLabelBoosting = value; }
        }
 
        public bool EnableRandomInputSelection
        {
            get { return m_bEnableRandomInputSelection; }
            set { m_bEnableRandomInputSelection = value; }
        }
 
        public bool EnablePairInputSelection
        {
            get { return m_bEnablePairInputSelection; }
            set { m_bEnablePairInputSelection = value; }
        }
 
        public bool UseTrainingSourceForTesting
        {
            get { return m_bUseTrainingSourceForTesting; }
            set { m_bUseTrainingSourceForTesting = value; }
        }
 
        public double SuperBoostProbability
        {
            get { return m_dfSuperBoostProbability; }
            set { m_dfSuperBoostProbability = value; }
        }
 
        public int MaximumIterationOverride
        {
            get { return m_nMaximumIterationOverride; }
            set { m_nMaximumIterationOverride = value; }
        }
 
        public int TestingIterationOverride
        {
            get { return m_nTestingIterationOverride; }
            set { m_nTestingIterationOverride = value; }
        }
 
        public string DefaultModelGroup
        {
            get { return m_strDefaultModelGroup; }
            set { m_strDefaultModelGroup = value; }
        }
 
        public string GpuIds
        {
            get { return m_strGpuIds; }
            set { m_strGpuIds = value; }
        }
 
        public DB_LOAD_METHOD DbLoadMethod
        {
            get { return m_dbLoadMethod; }
            set { m_dbLoadMethod = value; }
        }
 
        public int DbLoadLimit
        {
            get { return m_nDbLoadLimit; }
            set
            {
                if (value == 0 && m_nDbLoadLimit != 0)
                {
                    m_nAutoRefreshScheduledUpdateInMs = 0;
                    m_dfAutoRefreshScheduledReplacementPct = 0;
                }
                else if (value != 0 && m_nDbLoadLimit == 0)
                {
                    m_nAutoRefreshScheduledUpdateInMs = 10000;
                    m_dfAutoRefreshScheduledReplacementPct = 0.3;
                }
 
                m_nDbLoadLimit = value;
            }
        }
 
        public int DbAutoRefreshScheduledUpdateInMs
        {
            get { return m_nAutoRefreshScheduledUpdateInMs; }
            set { m_nAutoRefreshScheduledUpdateInMs = value; }
        }
 
        public double DbAutoRefreshScheduledReplacementPercent
        {
            get { return m_dfAutoRefreshScheduledReplacementPct; }
            set { m_dfAutoRefreshScheduledReplacementPct = value; }
        }
 
        public bool ItemDbLoadDataCriteria
        {
            get { return m_bItemDbLoadDataCriteria; }
            set { m_bItemDbLoadDataCriteria = value; }
        }
 
        public bool ItemDbLoadDebugData
        {
            get { return m_bItemDbLoadDebugData; }
            set { m_bItemDbLoadDebugData = value; }
        }
 
        public SNAPSHOT_WEIGHT_UPDATE_METHOD SnapshotWeightUpdateMethod
        {
            get { return m_snapshotWeightUpdateMethod; }
            set { m_snapshotWeightUpdateMethod = value; }
        }
 
        public SNAPSHOT_LOAD_METHOD SnapshotLoadMethod
        {
            get { return m_snapshotLoadMethod; }
            set { m_snapshotLoadMethod = value; }
        }
 
        public bool SkipMeanCheck
        {
            get { return m_bSkipMeanCheck; }
            set { m_bSkipMeanCheck = value; }
        }
    }
}