CfcParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.lnn
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class CfcParameter : LayerParameterBase
    {
        int m_nInputFeatures = 0;
        int m_nHiddenSize = 0;
        int m_nOutputFeatures = 0;
        bool m_bReturnSequence = false;
        CELL_TYPE m_cellType = CELL_TYPE.CFC;
 
        public enum CELL_TYPE
        {
            CFC,
            LTC
        }
 
        public CfcParameter()
        {
        }
 
        [Description("Specifies the cell type to use.")]
        public CELL_TYPE cell_type
        {
            get { return m_cellType; }
            set { m_cellType = value; }
        }
 
        [Description("Specifies the number of input features.")]
        public int input_features
        {
            get { return m_nInputFeatures; }
            set { m_nInputFeatures = value; }
        }
 
        [Description("Specifies the hidden size used to size the backbone units and other internal layers.")]
        public int hidden_size
        {
            get { return m_nHiddenSize; }
            set { m_nHiddenSize = value; }
        }
 
        [Description("Specifies the number of output features.")]
        public int output_features
        {
            get { return m_nOutputFeatures; }
            set { m_nOutputFeatures = value; }
        }
 
        [Description("Specifies whether or not to return the sequence.")]
        public bool return_sequences
        {
            get { return m_bReturnSequence; }
            set { m_bReturnSequence = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            CfcParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            CfcParameter p = (CfcParameter)src;
 
            m_nInputFeatures = p.input_features;
            m_nHiddenSize = p.hidden_size;
            m_nOutputFeatures = p.output_features;
            m_bReturnSequence = p.m_bReturnSequence;
            m_cellType = p.cell_type;
        }
 
        public override LayerParameterBase Clone()
        {
            CfcParameter p = new CfcParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("input_features", input_features.ToString());
            rgChildren.Add("hidden_size", hidden_size.ToString());
            rgChildren.Add("output_features", output_features.ToString());
            rgChildren.Add("return_sequences", return_sequences.ToString());
            rgChildren.Add("cell_type", cell_type.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static CfcParameter FromProto(RawProto rp)
        {
            string strVal;
            CfcParameter p = new CfcParameter();
 
            if ((strVal = rp.FindValue("input_features")) != null)
                p.input_features = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("hidden_size")) != null)
                p.hidden_size = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("output_features")) != null)
                p.output_features = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("return_sequences")) != null)
                p.return_sequences = bool.Parse(strVal);
 
            if ((strVal = rp.FindValue("cell_type")) != null)
                p.cell_type = (CELL_TYPE)Enum.Parse(typeof(CELL_TYPE), strVal, true);
 
            return p;
        }
    }
}