InputParameter.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class InputParameter : LayerParameterBase
    {
        List<BlobShape> m_rgShape = new List<BlobShape>();
 
        public InputParameter()
        {
        }
 
        [Description("Define N shapes to set a shape for each top; Define 1 shape to set the same shape for every top; Define no shapes to defer to manual reshaping.")]
        public List<BlobShape> shape
        {
            get { return m_rgShape; }
            set { m_rgShape = value; }
        }
 
        public override object Load(BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            InputParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            InputParameter p = (InputParameter)src;
 
            m_rgShape = new List<param.BlobShape>();
 
            foreach (BlobShape b in p.shape)
            {
                m_rgShape.Add(b.Clone());
            }
        }
 
        public override LayerParameterBase Clone()
        {
            InputParameter p = new param.InputParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            foreach (BlobShape b in m_rgShape)
            {
                rgChildren.Add(b.ToProto("shape"));
            }
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static InputParameter FromProto(RawProto rp)
        {
            InputParameter p = new InputParameter();
 
            RawProtoCollection col = rp.FindChildren("shape");
            foreach (RawProto rp1 in col)
            {
                BlobShape b = BlobShape.FromProto(rp1);
                p.m_rgShape.Add(b);
            }
 
            return p;
        }
    }
}