ArgMaxParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using System.ComponentModel;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class ArgMaxParameter : LayerParameterBase
    {
        bool m_bOutMaxVal = false;
        uint m_nTopK = 1;
        int? m_nAxis = null;
        COMPARE_OPERATOR m_op = COMPARE_OPERATOR.MAX;
        bool m_bEnableCudaImplementation = false;
 
        public enum COMPARE_OPERATOR
        {
            MAX,
            MIN
        }
 
        public ArgMaxParameter()
        {
        }
 
        public bool enable_cuda_impl
        {
            get { return m_bEnableCudaImplementation; }
            set { m_bEnableCudaImplementation = value; }
        }
 
        [Description("Specifies the operation to use (default = MAX).")]
        public COMPARE_OPERATOR operation
        {
            get { return m_op; }
            set { m_op = value; }
        }
 
        [Description("If true, produce pairs (argmax, maxval -or- argmin, minval when using 'op' = MIN).")]
        public bool out_max_val
        {
            get { return m_bOutMaxVal; }
            set { m_bOutMaxVal = value; }
        }
 
        [Description("When computing accuracy, count as correct by comparing the true label to the 'top_k' scoring classes.  By default, only compare the top scoring classes (i.e. argmax).")]
        public uint top_k
        {
            get { return m_nTopK; }
            set { m_nTopK = value; }
        }
 
        [Description("Specifies the axis along which to maximize -- may be negative to index from end (e.g., -1 for the last axis). By default the ArgMaxLayer maximizes over the flattened trailing dimensions for each index of the first / num dimension.")]
        public int? axis
        {
            get { return m_nAxis; }
            set { m_nAxis = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            ArgMaxParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            ArgMaxParameter p = (ArgMaxParameter)src;
            m_bOutMaxVal = p.m_bOutMaxVal;
            m_nTopK = p.m_nTopK;
            m_nAxis = p.m_nAxis;
            m_op = p.m_op;
            m_bEnableCudaImplementation = p.enable_cuda_impl;
        }
 
        public override LayerParameterBase Clone()
        {
            ArgMaxParameter p = new ArgMaxParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            if (out_max_val != false)
                rgChildren.Add("out_max_val", out_max_val.ToString());
 
            if (top_k != 1)
                rgChildren.Add("top_k", top_k.ToString());
 
            if (axis.HasValue)
                rgChildren.Add("axis", axis.Value.ToString());
 
            if (operation != COMPARE_OPERATOR.MAX)
                rgChildren.Add("operation", operation.ToString());
 
            rgChildren.Add("enable_cuda_impl", m_bEnableCudaImplementation.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static ArgMaxParameter FromProto(RawProto rp)
        {
            string strVal;
            ArgMaxParameter p = new ArgMaxParameter();
 
            if ((strVal = rp.FindValue("out_max_val")) != null)
                p.out_max_val = bool.Parse(strVal);
 
            if ((strVal = rp.FindValue("top_k")) != null)
                p.top_k = uint.Parse(strVal);
 
            if ((strVal = rp.FindValue("axis")) != null)
                p.axis = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("operation")) != null)
            {
                if (strVal == COMPARE_OPERATOR.MIN.ToString())
                    p.operation = COMPARE_OPERATOR.MIN;
                else
                    p.operation = COMPARE_OPERATOR.MAX;
            }
 
            if ((strVal = rp.FindValue("enable_cuda_impl")) != null)
                p.m_bEnableCudaImplementation = bool.Parse(strVal);
 
            return p;
        }
    }
}