ReshapeTemporalParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.tft
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class ReshapeTemporalParameter : LayerParameterBase
    {
        MODE m_mode = MODE.BEFORE;
        bool m_bEnableClipOutput = false;
        bool m_bEnableWtsOutput = false;
        int m_nForcedRepeatCount = -1;
 
        public enum MODE
        {
            BEFORE,
            AFTER
        }
 
        public ReshapeTemporalParameter()
        {
        }
 
        [Description("Specifies the forced repeat steps bottom(1).  A value of -1 specifies to use the temporal axis as the repeat count (default), otherwise the forced count is used to compy the entire blob that number of times.")]
        public int forced_repeat_count
        {
            get { return m_nForcedRepeatCount; }
            set { m_nForcedRepeatCount = value; }
        }
 
        [Description("Specifies to output the clip for the data ouptut in the AFTER mode.")]
        public bool enable_clip_output
        {
            get { return m_bEnableClipOutput; }
            set { m_bEnableClipOutput = value; }
        }
 
        [Description("Specifies to output the weights for the data ouptut in the AFTER mode.")]
        public bool enable_weight_output
        {
            get { return m_bEnableWtsOutput; }
            set { m_bEnableWtsOutput = value; }
        }
 
        [Description("Specifies the mode of operation.")]
        public MODE mode
        {
            get { return m_mode; }
            set { m_mode = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            ReshapeTemporalParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            ReshapeTemporalParameter p = (ReshapeTemporalParameter)src;
 
            m_mode = p.mode;
            m_bEnableClipOutput = p.enable_clip_output;
            m_bEnableWtsOutput = p.enable_weight_output;
            m_nForcedRepeatCount = p.forced_repeat_count;
        }
 
        public override LayerParameterBase Clone()
        {
            ReshapeTemporalParameter p = new ReshapeTemporalParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("mode", mode.ToString());
            rgChildren.Add("enable_clip_output", enable_clip_output.ToString());
            rgChildren.Add("enable_weight_output", enable_weight_output.ToString());
            rgChildren.Add("forced_repeat_count", forced_repeat_count.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static ReshapeTemporalParameter FromProto(RawProto rp)
        {
            string strVal;
            ReshapeTemporalParameter p = new ReshapeTemporalParameter();
 
            if ((strVal = rp.FindValue("mode")) != null)
            {
                if (strVal == MODE.BEFORE.ToString())
                    p.mode = MODE.BEFORE;
                else if (strVal == MODE.AFTER.ToString())
                    p.mode = MODE.AFTER;
                else
                    throw new Exception("Invalid mode '" + strVal + "'.");
            }
 
            if ((strVal = rp.FindValue("enable_clip_output")) != null)
                p.enable_clip_output = bool.Parse(strVal);
 
            if ((strVal = rp.FindValue("enable_weight_output")) != null)
                p.enable_weight_output = bool.Parse(strVal);
 
            if ((strVal = rp.FindValue("forced_repeat_count")) != null)
                p.forced_repeat_count = int.Parse(strVal);
 
            return p;
        }
    }
}