FlattenParameter.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class FlattenParameter : LayerParameterBase
    {
        int m_nAxis = 1;
        int m_nEndAxis = -1;
 
        public FlattenParameter()
        {
        }
 
        [Description("Specifies the first axis to flatten: all preceding axes are retained in the output.")]
        public int axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        [Description("Specifies the last axis to flatten: all following axes are retained in the output.")]
        public int end_axis
        {
            get { return m_nEndAxis; }
            set { m_nEndAxis = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            FlattenParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            FlattenParameter p = (FlattenParameter)src;
            m_nAxis = p.m_nAxis;
            m_nEndAxis = p.m_nEndAxis;
        }
 
        public override LayerParameterBase Clone()
        {
            FlattenParameter p = new FlattenParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("axis", axis.ToString());
            rgChildren.Add("end_axis", end_axis.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static FlattenParameter FromProto(RawProto rp)
        {
            string strVal;
            FlattenParameter p = new FlattenParameter();
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("end_axis")) != null)
                p.end_axis = int.Parse(strVal);
 
            return p;
        }
 
        public static List<int> Reshape(int nParamAxis, int nParamEndAxis, List<int> rgShape, int nStartAxis = -1, int nEndAxis = -1)
        {
            if (nStartAxis < 0)
                nStartAxis = Utility.CanonicalAxisIndex(nParamAxis, rgShape.Count);
 
            if (nEndAxis < 0)
                nEndAxis = Utility.CanonicalAxisIndex(nParamEndAxis, rgShape.Count);
 
            List<int> rgTopShape = new List<int>();
            for (int i = 0; i < nStartAxis; i++)
            {
                rgTopShape.Add(rgShape[i]);
            }
 
            int nFlattenDim = Utility.Count(rgShape, nStartAxis, nEndAxis + 1);
            rgTopShape.Add(nFlattenDim);
 
            for (int i = nEndAxis + 1; i < rgShape.Count; i++)
            {
                rgTopShape.Add(rgShape[i]);
            }
 
            return rgTopShape;
        }
    }
}