HDF5DataParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class HDF5DataParameter : LayerParameterBase
    {
        string m_strSource = null;
        uint m_nBatchSize;
        bool m_bShuffle;
        List<BlobShape> m_rgExpectedTopShapes = new List<BlobShape>();
 
        public event EventHandler<VerifyBatchSizeArgs> OnVerifyBatchSize;
 
        public HDF5DataParameter()
        {
        }
 
        [Description("Specifies the data source.")]
        public string source
        {
            get { return m_strSource; }
            set { m_strSource = value; }
        }
 
        [Description("Specifies the batch size of images to collect and train on each iteration of the network.  NOTE: Setting the training netorks batch size >= to the testing net batch size will conserve memory by allowing the training net to share its gpu memory with the testing net.")]
        public virtual uint batch_size
        {
            get { return m_nBatchSize; }
            set
            {
                if (OnVerifyBatchSize != null)
                {
                    VerifyBatchSizeArgs args = new VerifyBatchSizeArgs(value);
                    OnVerifyBatchSize(this, args);
                    if (args.Error != null)
                        throw args.Error;
                }
 
                m_nBatchSize = value;
            }
        }
 
        [Description("Specifies the expected top shapes.  Only used to verify input shapes, if the shape is -1 or does not exist the verification test is ignored for the shape.")]
        public List<BlobShape> expected_top_shapes
        {
            get { return m_rgExpectedTopShapes; }
            set { m_rgExpectedTopShapes = value; }
        }
 
        [Description("Specifies whether to shuffle the data or now.")]
        public bool shuffle
        {
            get { return m_bShuffle; }
            set { m_bShuffle = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            HDF5DataParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            HDF5DataParameter p = (HDF5DataParameter)src;
            m_strSource = p.m_strSource;
            m_nBatchSize = p.m_nBatchSize;
            m_bShuffle = p.m_bShuffle;
 
            m_rgExpectedTopShapes = new List<BlobShape>();
            foreach (BlobShape shape in p.m_rgExpectedTopShapes)
            {
                m_rgExpectedTopShapes.Add(shape.Clone());
            }
        }
 
        public override LayerParameterBase Clone()
        {
            HDF5DataParameter p = new HDF5DataParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("source", "\"" + source + "\"");
            rgChildren.Add("batch_size", batch_size.ToString());
            rgChildren.Add("shuffle", shuffle.ToString());
 
            for (int i = 0; i < m_rgExpectedTopShapes.Count; i++)
            {
                rgChildren.Add(m_rgExpectedTopShapes[i].ToProto("expected_top_shape"));
            }
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static HDF5DataParameter FromProto(RawProto rp, HDF5DataParameter p = null)
        {
            string strVal;
 
            if (p == null)
                p = new HDF5DataParameter();
 
            if ((strVal = rp.FindValue("source")) != null)
                p.source = strVal.Trim('\"');
 
            if ((strVal = rp.FindValue("batch_size")) != null)
                p.batch_size = uint.Parse(strVal);
 
            if ((strVal = rp.FindValue("shuffle")) != null)
                p.shuffle = bool.Parse(strVal);
 
            p.m_rgExpectedTopShapes = new List<BlobShape>();
            RawProtoCollection colExpectedTopShapes = rp.FindChildren("expected_top_shape");
            foreach (RawProto rp1 in colExpectedTopShapes)
            {
                p.m_rgExpectedTopShapes.Add(BlobShape.FromProto(rp1));
            }
 
            return p;
        }
    }
}