WeightInfo.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
 
namespace MyCaffe.common
{
    public class WeightInfo<T>
    {
        Dictionary<string, Tuple<List<int>, BLOB_TYPE>> m_rgBlobInfo = new Dictionary<string, Tuple<List<int>, BLOB_TYPE>>();
        BlobName m_names = new BlobName();
 
        public WeightInfo()
        {
        }
 
        public void AddBlob(string strName, List<int> rgShape, BLOB_TYPE type)
        {
            strName = m_names.GetName(strName);
            m_rgBlobInfo.Add(strName, new Tuple<List<int>, BLOB_TYPE>(rgShape, type));
        }
 
        public void AddBlob(Blob<T> b)
        {
            string strName = m_names.GetName(b.Name);
            m_rgBlobInfo.Add(strName, new Tuple<List<int>, BLOB_TYPE>(b.shape(), b.type));
        }
 
        public Dictionary<string, Tuple<List<int>, BLOB_TYPE>> Blobs
        {
            get { return m_rgBlobInfo; }
        }
    }
 
    public class BlobName
    {
        Dictionary<string, int> m_rgNames = new Dictionary<string, int>();
 
        public BlobName()
        {
        }
 
        public string GetName(string strName)
        {
            if (string.IsNullOrEmpty(strName))
                strName = "b";
 
            if (!m_rgNames.ContainsKey(strName))
            {
                m_rgNames.Add(strName, 1);
            }
            else
            {
                m_rgNames[strName]++;
                strName = strName + m_rgNames[strName].ToString();
            }
 
            return strName;
        }
    }
}