TripletLossParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.beta
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class TripletLossParameter : LayerParameterBase 
    {
        double m_dfAlpha = 1.1;
        int m_nPreGenerateTargtStart = 0;
 
        public TripletLossParameter()
        {
        }
 
        public double alpha
        {
            get { return m_dfAlpha; }
            set { m_dfAlpha = value; }
        }
 
        [Description("Specifies the starting label for pre-generated targets, only used when 'colBottom.Count' = 5, which contains centroids.")]
        public int pregen_label_start
        {
            get { return m_nPreGenerateTargtStart; }
            set { m_nPreGenerateTargtStart = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            TripletLossParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            TripletLossParameter p = (TripletLossParameter)src;
            m_dfAlpha = p.m_dfAlpha;
            m_nPreGenerateTargtStart = p.m_nPreGenerateTargtStart;
        }
 
        public override LayerParameterBase Clone()
        {
            TripletLossParameter p = new TripletLossParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("alpha", alpha.ToString());
            rgChildren.Add("pregen_label_start", pregen_label_start.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static TripletLossParameter FromProto(RawProto rp)
        {
            string strVal;
            TripletLossParameter p = new TripletLossParameter();
 
            if ((strVal = rp.FindValue("alpha")) != null)
                p.alpha = ParseDouble(strVal);
 
            if ((strVal = rp.FindValue("pregen_label_start")) != null)
                p.pregen_label_start = int.Parse(strVal);
 
            return p;
        }
    }
}