SPPParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class SPPParameter : EngineParameter 
    {
        PoolingParameter.PoolingMethod m_method = PoolingParameter.PoolingMethod.MAX;
        uint m_nPyramidHeight;
 
        public SPPParameter()
            : base()
        {
        }
 
        [Description("Specifies the pooling method to use.")]
        public PoolingParameter.PoolingMethod pool
        {
            get { return m_method; }
            set { m_method = value; }
        }
 
        [Description("Specifies the pyramid height.")]
        public uint pyramid_height
        {
            get { return m_nPyramidHeight; }
            set { m_nPyramidHeight = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            SPPParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            base.Copy(src);
 
            if (src is SPPParameter)
            {
                SPPParameter p = (SPPParameter)src;
                m_method = p.m_method;
                m_nPyramidHeight = p.m_nPyramidHeight;
            }
        }
 
        public override LayerParameterBase Clone()
        {
            SPPParameter p = new SPPParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProto rpBase = base.ToProto("engine");
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(rpBase.Children);
            rgChildren.Add("method", pool.ToString());
            rgChildren.Add("pyramid_height", pyramid_height.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static new SPPParameter FromProto(RawProto rp)
        {
            string strVal;
            SPPParameter p = new SPPParameter();
 
            p.Copy(EngineParameter.FromProto(rp));
 
            if ((strVal = rp.FindValue("method")) != null)
            {
                switch (strVal)
                {
                    case "MAX":
                        p.pool = PoolingParameter.PoolingMethod.MAX;
                        break;
 
                    case "AVE":
                        p.pool = PoolingParameter.PoolingMethod.AVE;
                        break;
 
                    case "STOCHASTIC":
                        p.pool = PoolingParameter.PoolingMethod.STOCHASTIC;
                        break;
 
                    default:
                        throw new Exception("Unknown pooling 'method' value: " + strVal);
                }
            }
 
            if ((strVal = rp.FindValue("pyramid_height")) != null)
                p.pyramid_height = uint.Parse(strVal);
 
            return p;
        }
    }
}