NonMaximumSuppressionParameter.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 NonMaximumSuppressionParameter : OptionalParameter
    {
        float m_fNmsThreshold = 0.3f;
        int? m_nTopK = null;
        float m_fEta = 1.0f;
 
        public NonMaximumSuppressionParameter(bool bActive) : base(bActive)
        {
        }
 
        [Description("Get/set the threshold to be used in nms.")]
        public float nms_threshold
        {
            get { return m_fNmsThreshold; }
            set { m_fNmsThreshold = value; }
        }
 
        [Description("Get/set the maximum number of results kept.")]
        public int? top_k
        {
            get { return m_nTopK; }
            set { m_nTopK = value; }
        }
 
        [Description("Get/set the parameter for adaptive nms.")]
        public float eta
        {
            get { return m_fEta; }
            set { m_fEta = value; }
        }
 
        public override void Copy(OptionalParameter src)
        {
            base.Copy(src);
 
            if (src is NonMaximumSuppressionParameter)
            {
                NonMaximumSuppressionParameter p = (NonMaximumSuppressionParameter)src;
                m_fNmsThreshold = p.nms_threshold;
                m_nTopK = p.top_k;
                m_fEta = p.eta;
            }
        }
 
        public NonMaximumSuppressionParameter Clone()
        {
            NonMaximumSuppressionParameter p = new param.ssd.NonMaximumSuppressionParameter(Active);
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProto rpBase = base.ToProto("option");
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(rpBase);
            rgChildren.Add(new RawProto("nms_threshold", nms_threshold.ToString()));
 
            if (top_k.HasValue)
                rgChildren.Add(new RawProto("top_k", top_k.Value.ToString()));
 
            rgChildren.Add(new RawProto("eta", eta.ToString()));
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static new NonMaximumSuppressionParameter FromProto(RawProto rp)
        {
            NonMaximumSuppressionParameter p = new NonMaximumSuppressionParameter(false);
            string strVal;
 
            RawProto rpOption = rp.FindChild("option");
            if (rpOption != null)
                ((OptionalParameter)p).Copy(OptionalParameter.FromProto(rpOption));
 
            if ((strVal = rp.FindValue("nms_threshold")) != null)
                p.nms_threshold = BaseParameter.ParseFloat(strVal);
 
            if ((strVal = rp.FindValue("top_k")) != null)
                p.top_k = int.Parse(strVal);
 
            if ((strVal = rp.FindValue("eta")) != null)
                p.eta = BaseParameter.ParseFloat(strVal);
 
            return p;
        }
    }
}