/// <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); }
/** @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); }