MyCaffeModelData.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.basecode
{
    public class MyCaffeModelData
    {
        string m_strModelDescription;
        string m_strSolverDescription = null;
        byte[] m_rgWeights;
        byte[] m_rgImageMean;
        string m_strLastModelFileName = "";
        string m_strLastSolverFileName = "";
        string m_strLastImageMeanFileName = "";
        string m_strLastWeightsFileName = "";
        string m_strOriginalDownloadFile = null;
        List<int> m_rgInputShape = new List<int>();
 
        public MyCaffeModelData(string strModelDesc, byte[] rgWeights, byte[] rgImageMean = null, string strSolverDescription = null, List<int> rgInputShape = null)
        {
            m_strModelDescription = strModelDesc;
            m_rgWeights = rgWeights;
            m_rgImageMean = rgImageMean;
            m_strSolverDescription = strSolverDescription;
 
            if (rgInputShape != null)
                m_rgInputShape = rgInputShape;
        }
 
        public List<int> InputShape
        {
            get { return m_rgInputShape; }
            set { m_rgInputShape = value; }
        }
 
        public string ModelDescription
        {
            get { return m_strModelDescription; }
            set { m_strModelDescription = value; }
        }
 
        public string SolverDescription
        {
            get { return m_strSolverDescription; }
            set { m_strSolverDescription = value; }
        }
 
        public byte[] Weights
        {
            get { return m_rgWeights; }
        }
 
        public byte[] ImageMean
        {
            get { return m_rgImageMean; }
        }
 
        public string OriginalDownloadFile
        {
            get { return m_strOriginalDownloadFile; }
            set { m_strOriginalDownloadFile = value; }
        }
 
        public void Save(string strFolder, string strName)
        {
            string strModel = strFolder.TrimEnd('\\') + "\\" + strName + "_model_desc.prototxt";
 
            if (File.Exists(strModel))
                File.Delete(strModel);
 
            m_strLastModelFileName = strModel;
            using (StreamWriter sr = new StreamWriter(strModel))
            {
                sr.WriteLine(m_strModelDescription);
            }
 
            if (!string.IsNullOrEmpty(m_strSolverDescription))
            {
                string strSolver = strFolder.TrimEnd('\\') + "\\" + strName + "_solver_desc.prototxt";
 
                if (File.Exists(strSolver))
                    File.Delete(strSolver);
 
                m_strLastSolverFileName = strSolver;
                using (StreamWriter sr = new StreamWriter(strSolver))
                {
                    sr.WriteLine(strSolver);
                }
            }
 
            if (m_rgWeights != null)
            {
                string strWts = strFolder.TrimEnd('\\') + "\\" + strName + "_weights.mycaffemodel";
 
                if (File.Exists(strWts))
                    File.Delete(strWts);
 
                m_strLastWeightsFileName = strWts;
                using (FileStream fs = new FileStream(strWts, FileMode.CreateNew, FileAccess.Write))
                using (BinaryWriter bw = new BinaryWriter(fs))
                {
                    bw.Write(m_rgWeights);
                }
            }
            else
            {
                m_strLastWeightsFileName = "";
            }
 
            if (m_rgImageMean != null)
            {
                string strWts = strFolder.TrimEnd('\\') + "\\" + strName + "_image_mean.bin";
 
                if (File.Exists(strWts))
                    File.Delete(strWts);
 
                m_strLastImageMeanFileName = strWts;
                using (FileStream fs = new FileStream(strWts, FileMode.CreateNew, FileAccess.Write))
                using (BinaryWriter bw = new BinaryWriter(fs))
                {
                    bw.Write(m_rgImageMean);
                }
            }
            else
            {
                m_strLastImageMeanFileName = "";
            }
        }
 
        public string LastSavedModeDescriptionFileName
        {
            get { return m_strLastModelFileName; }
        }
 
        public string LastSaveSolverDescriptionFileName
        {
            get { return m_strLastSolverFileName; }
        }
 
        public string LastSavedWeightsFileName
        {
            get { return m_strLastWeightsFileName; }
        }
 
        public string LastSavedImageMeanFileName
        {
            get { return m_strLastImageMeanFileName; }
        }
    }
}