/// <summary> /// Creates a new copy of this instance of the parameter. /// </summary> /// <returns>A new instance of this parameter is returned.</returns> public override LayerParameterBase Clone() { DetectionOutputParameter p = new DetectionOutputParameter(); p.Copy(this); return(p); }
/// <summary> /// Load the parameter from a binary reader. /// </summary> /// <param name="br">Specifies the binary reader.</param> /// <param name="bNewInstance">When <i>true</i> a new instance is created (the default), otherwise the existing instance is loaded from the binary reader.</param> /// <returns>Returns an instance of the parameter.</returns> public override object Load(System.IO.BinaryReader br, bool bNewInstance = true) { RawProto proto = RawProto.Parse(br.ReadString()); DetectionOutputParameter p = FromProto(proto); if (!bNewInstance) { Copy(p); } return(p); }
/// <summary> /// Copy on parameter to another. /// </summary> /// <param name="src">Specifies the parameter to copy.</param> public override void Copy(LayerParameterBase src) { DetectionOutputParameter p = (DetectionOutputParameter)src; m_nNumClasses = p.m_nNumClasses; m_bShareLocation = p.m_bShareLocation; m_nBackgroundLabelId = p.m_nBackgroundLabelId; m_nmsParam = p.m_nmsParam.Clone(); m_saveOutputParam = p.save_output_param.Clone() as SaveOutputParameter; m_codeType = p.m_codeType; m_bVarianceEncodedInTarget = p.m_bVarianceEncodedInTarget; m_nKeepTopK = p.m_nKeepTopK; m_fConfidenceThreshold = p.m_fConfidenceThreshold; m_bVisualize = p.m_bVisualize; m_fVisualizeThreshold = p.m_fVisualizeThreshold; m_strSaveFile = p.m_strSaveFile; }
/// <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 DetectionOutputParameter FromProto(RawProto rp) { DetectionOutputParameter p = new DetectionOutputParameter(); string strVal; if ((strVal = rp.FindValue("num_classes")) != null) { p.num_classes = uint.Parse(strVal); } if ((strVal = rp.FindValue("share_location")) != null) { p.share_location = bool.Parse(strVal); } if ((strVal = rp.FindValue("background_label_id")) != null) { p.background_label_id = int.Parse(strVal); } RawProto rpNms = rp.FindChild("nms_param"); if (rpNms != null) { p.nms_param = NonMaximumSuppressionParameter.FromProto(rpNms); } RawProto rpSave = rp.FindChild("save_output_param"); if (rpSave != null) { p.save_output_param = SaveOutputParameter.FromProto(rpSave); } if ((strVal = rp.FindValue("code_type")) != null) { if (strVal == PriorBoxParameter.CodeType.CENTER_SIZE.ToString()) { p.code_type = PriorBoxParameter.CodeType.CENTER_SIZE; } else if (strVal == PriorBoxParameter.CodeType.CORNER.ToString()) { p.code_type = PriorBoxParameter.CodeType.CORNER; } else if (strVal == PriorBoxParameter.CodeType.CORNER_SIZE.ToString()) { p.code_type = PriorBoxParameter.CodeType.CORNER_SIZE; } else { throw new Exception("Unknown PriorBoxParameter.CodeType '" + strVal + "'!"); } } if ((strVal = rp.FindValue("variance_encoded_in_target")) != null) { p.variance_encoded_in_target = bool.Parse(strVal); } if ((strVal = rp.FindValue("keep_top_k")) != null) { p.keep_top_k = int.Parse(strVal); } if ((strVal = rp.FindValue("confidence_threshold")) != null) { p.confidence_threshold = ParseFloat(strVal); } if ((strVal = rp.FindValue("visualize")) != null) { p.visualize = bool.Parse(strVal); } if ((strVal = rp.FindValue("visualize_threshold")) != null) { p.visualize_threshold = ParseFloat(strVal); } if ((strVal = rp.FindValue("save_file")) != null) { p.save_file = strVal; } return(p); }