PermuteParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.ssd
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class PermuteParameter : LayerParameterBase 
    {
        List<int> m_rgOrder = new List<int>();
 
        public PermuteParameter()
        {
        }
 
        [Description("Specifies the new orders of the axes of data.  Should be within the same range as the input data starting with 0 and no repeated orders.")]
        public List<int> order
        {
            get { return m_rgOrder; }
            set { m_rgOrder = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            PermuteParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            PermuteParameter p = (PermuteParameter)src;
            m_rgOrder = Utility.Clone<int>(p.order);
        }
 
        public override LayerParameterBase Clone()
        {
            PermuteParameter p = new PermuteParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            foreach (int nOrder in m_rgOrder)
            {
                rgChildren.Add("order", nOrder.ToString());
            }
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static PermuteParameter FromProto(RawProto rp)
        {
            PermuteParameter p = new PermuteParameter();
 
            RawProtoCollection rgChildren = rp.FindChildren("order");
            foreach (RawProto rp1 in rgChildren)
            {
                p.order.Add(int.Parse(rp1.Value));
            }
 
            return p;
        }
 
        public static List<int> Reshape(List<int> rgOrder, List<int> rgShape, int nNumAxes)
        {
            List<int> rgTopShape = new List<int>();
 
            for (int i = 0; i < nNumAxes; i++)
            {
                int nOrder = rgOrder[i];
                int nShape = rgShape[nOrder];
                rgTopShape.Add(nShape);
            }
 
            return rgTopShape;
        }
    }
}