ExpansionParameter.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using MyCaffe.basecode;
 
namespace MyCaffe.param.ssd
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class ExpansionParameter : OptionalParameter
    {
        float m_fProb = 0;
        float m_fMaxExpandRatio = 1.0f;
 
        public ExpansionParameter(bool bActive) : base(bActive)
        {
        }
 
        [Description("Get/set probability of using this expansion policy.")]
        public float prob
        {
            get { return m_fProb; }
            set { m_fProb = value; }
        }
 
        [Description("Get/set the ratio to expand the image.")]
        public float max_expand_ratio
        {
            get { return m_fMaxExpandRatio; }
            set { m_fMaxExpandRatio = value; }
        }
 
        public override void Copy(OptionalParameter src)
        {
            base.Copy(src);
 
            if (src is ExpansionParameter)
            {
                ExpansionParameter p = (ExpansionParameter)src;
                m_fProb = p.prob;
                m_fMaxExpandRatio = p.max_expand_ratio;
            }
        }
 
        public ExpansionParameter Clone()
        {
            ExpansionParameter p = new param.ssd.ExpansionParameter(Active);
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProto rpBase = base.ToProto("option");
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(rpBase);
            rgChildren.Add(new RawProto("active", Active.ToString()));
            rgChildren.Add(new RawProto("prob", prob.ToString()));
            rgChildren.Add(new RawProto("max_expand_ratio", max_expand_ratio.ToString()));
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static new ExpansionParameter FromProto(RawProto rp)
        {
            ExpansionParameter p = new ExpansionParameter(false);
            string strVal;
 
            RawProto rpOption = rp.FindChild("option");
            if (rpOption != null)
                ((OptionalParameter)p).Copy(OptionalParameter.FromProto(rpOption));
 
            if ((strVal = rp.FindValue("prob")) != null)
                p.prob = BaseParameter.ParseFloat(strVal);
 
            if ((strVal = rp.FindValue("max_expand_ratio")) != null)
                p.max_expand_ratio = BaseParameter.ParseFloat(strVal);
 
            return p;
        }
    }
}