AnnotatedDataParameter.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.ssd
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class AnnotatedDataParameter : LayerParameterBase
    {
        SimpleDatum.ANNOTATION_TYPE m_type = SimpleDatum.ANNOTATION_TYPE.NONE;
        List<BatchSampler> m_rgBatchSampler = new List<BatchSampler>();
        string m_strLabelFile;
 
 
        public AnnotatedDataParameter()
        {
        }
 
        [Description("Get/set the annotation type.")]
        public SimpleDatum.ANNOTATION_TYPE anno_type
        {
            get { return m_type; }
            set { m_type = value; }
        }
 
        [Description("Get/set the batch sampler.")]
        public List<BatchSampler> batch_sampler
        {
            get { return m_rgBatchSampler; }
            set { m_rgBatchSampler = value; }
        }
 
        [Description("Get/set the label map file.")]
        public string label_map_file
        {
            get { return m_strLabelFile; }
            set { m_strLabelFile = value; }
        }
 
        public override object Load(BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            AnnotatedDataParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            if (src is AnnotatedDataParameter)
            {
                AnnotatedDataParameter p = (AnnotatedDataParameter)src;
                m_type = p.m_type;
                m_strLabelFile = p.m_strLabelFile;
 
                m_rgBatchSampler = new List<BatchSampler>();
                foreach (BatchSampler bs in p.m_rgBatchSampler)
                {
                    m_rgBatchSampler.Add(bs.Clone());
                }
            }
        }
 
        public override LayerParameterBase Clone()
        {
            AnnotatedDataParameter p = new param.ssd.AnnotatedDataParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(new RawProto("anno_type", ((int)m_type).ToString()));
            rgChildren.Add(new RawProto("label_map_file", m_strLabelFile));
 
            foreach (BatchSampler bs in m_rgBatchSampler)
            {
                rgChildren.Add(bs.ToProto("batch_sampler"));
            }
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static AnnotatedDataParameter FromProto(RawProto rp)
        {
            AnnotatedDataParameter p = new AnnotatedDataParameter();
            string strVal;
 
            if ((strVal = rp.FindValue("anno_type")) != null)
                p.m_type = (SimpleDatum.ANNOTATION_TYPE)int.Parse(strVal);
 
            if ((strVal = rp.FindValue("label_map_file")) != null)
                p.m_strLabelFile = strVal;
 
            RawProtoCollection col = rp.FindChildren("batch_sampler");
            foreach (RawProto rp1 in col)
            {
                p.m_rgBatchSampler.Add(BatchSampler.FromProto(rp1));
            }
 
            return p;
        }
    }
}