/// <summary> /// Parses the parameter from a RawProto. /// </summary> /// <param name="rp">Specifies the RawProto to parse.</param> /// <returns>A new instance of the parameter is returned.</returns> public static new PoolingParameter FromProto(RawProto rp) { string strVal; PoolingParameter p = new PoolingParameter(); ((KernelParameter)p).Copy(KernelParameter.FromProto(rp)); if ((strVal = rp.FindValue("pool")) != null) { switch (strVal) { case "MAX": p.pool = PoolingMethod.MAX; break; case "AVE": p.pool = PoolingMethod.AVE; break; case "STOCHASTIC": p.pool = PoolingMethod.STOCHASTIC; break; default: throw new Exception("Unknown pooling 'method' value: " + strVal); } } if ((strVal = rp.FindValue("global_pooling")) != null) { p.global_pooling = bool.Parse(strVal); } return(p); }
/// <summary> /// Convert the parameter into a RawProto. /// </summary> /// <param name="strName">Specifies the base name for the raw proto.</param> /// <returns>The RawProto is returned.</returns> public override RawProto ToProto(string strName) { RawProto rpBase = base.ToProto("engine"); RawProtoCollection rgChildren = new RawProtoCollection(); KernelParameter p = new KernelParameter(); rgChildren.Add(rpBase.Children); rgChildren.Add <uint>("kernel_size", m_rgKernelSize); rgChildren.Add <uint>("stride", m_rgStride); rgChildren.Add <uint>("pad", m_rgPad); if (m_rgDilation.Count > 0) { rgChildren.Add <uint>("dilation", m_rgDilation); } rgChildren.Add("kernel_h", m_nKernelH); rgChildren.Add("kernel_w", m_nKernelW); rgChildren.Add("stride_h", m_nStrideH); rgChildren.Add("stride_w", m_nStrideW); rgChildren.Add("pad_h", m_nPadH); rgChildren.Add("pad_w", m_nPadW); return(new RawProto(strName, "", rgChildren)); }
/** @copydoc EngineParameter::Clone */ public override LayerParameterBase Clone() { KernelParameter p = new KernelParameter(); p.Copy(this); return(p); }
/** @copydoc KernelParameter::FromProto */ public static new ConvolutionParameter FromProto(RawProto rp) { string strVal; ConvolutionParameter p = new ConvolutionParameter(); ((KernelParameter)p).Copy(KernelParameter.FromProto(rp)); if ((strVal = rp.FindValue("num_output")) != null) { p.num_output = uint.Parse(strVal); } if ((strVal = rp.FindValue("bias_term")) != null) { p.bias_term = bool.Parse(strVal); } if ((strVal = rp.FindValue("group")) != null) { p.group = uint.Parse(strVal); } RawProto rpWeightFiller = rp.FindChild("weight_filler"); if (rpWeightFiller != null) { p.weight_filler = FillerParameter.FromProto(rpWeightFiller); } RawProto rpBiasFiller = rp.FindChild("bias_filler"); if (rpBiasFiller != null) { p.bias_filler = FillerParameter.FromProto(rpBiasFiller); } if ((strVal = rp.FindValue("axis")) != null) { p.axis = int.Parse(strVal); } if ((strVal = rp.FindValue("force_nd_im2col")) != null) { p.force_nd_im2col = bool.Parse(strVal); } if ((strVal = rp.FindValue("cudnn_workspace_limit")) != null) { p.cudnn_workspace_limit = int.Parse(strVal); } if ((strVal = rp.FindValue("cudnn_worspace_allow_on_groups")) != null) { p.cudnn_workspace_allow_on_groups = bool.Parse(strVal); } return(p); }
/** @copydoc EngineParameter::Load */ public override object Load(System.IO.BinaryReader br, bool bNewInstance = true) { RawProto proto = RawProto.Parse(br.ReadString()); KernelParameter p = FromProto(proto); if (!bNewInstance) { Copy(p); } return(p); }
/// <summary> /// Parse a RawProto into a new instance of the parameter. /// </summary> /// <param name="rp">Specifies the RawProto to parse.</param> /// <returns>A new instance of the parameter is returned.</returns> public static new KernelParameter FromProto(RawProto rp) { KernelParameter p = new KernelParameter(); ((EngineParameter)p).Copy(EngineParameter.FromProto(rp)); p.m_rgPad = rp.FindArray <uint>("pad"); p.m_rgStride = rp.FindArray <uint>("stride"); p.m_rgKernelSize = rp.FindArray <uint>("kernel_size"); p.m_rgDilation = rp.FindArray <uint>("dilation"); p.m_nPadH = (uint?)rp.FindValue("pad_h", typeof(uint)); p.m_nPadW = (uint?)rp.FindValue("pad_w", typeof(uint)); p.m_nStrideH = (uint?)rp.FindValue("stride_h", typeof(uint)); p.m_nStrideW = (uint?)rp.FindValue("stride_w", typeof(uint)); p.m_nKernelH = (uint?)rp.FindValue("kernel_h", typeof(uint)); p.m_nKernelW = (uint?)rp.FindValue("kernel_w", typeof(uint)); return(p); }
/** @copydoc EngineParameter::Copy */ public override void Copy(LayerParameterBase src) { base.Copy(src); if (src is KernelParameter) { KernelParameter p = (KernelParameter)src; m_rgPad = Utility.Clone <uint>(p.m_rgPad); m_rgStride = Utility.Clone <uint>(p.m_rgStride); m_rgKernelSize = Utility.Clone <uint>(p.m_rgKernelSize); m_rgDilation = Utility.Clone <uint>(p.m_rgDilation); m_nPadH = p.m_nPadH; m_nPadW = p.m_nPadW; m_nStrideH = p.m_nStrideH; m_nStrideW = p.m_nStrideW; m_nKernelH = p.m_nKernelH; m_nKernelW = p.m_nKernelW; } }