BilinearFiller.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
using MyCaffe.param;
using MyCaffe.common;
 
namespace MyCaffe.fillers
{
    public class BilinearFiller<T> : Filler<T>
    {
        public BilinearFiller(CudaDnn<T> cuda, Log log, FillerParameter p)
            : base(cuda, log, p)
        {
        }
 
        public override void Fill(int nCount, long hMem, int nNumAxes = 1, int nNumOutputs = 1, int nNumChannels = 1, int nHeight = 1, int nWidth = 1)
        {
            m_log.CHECK_EQ(nNumAxes, 4, "Blob must be 4 dim.");
            m_log.CHECK_EQ(nWidth, nHeight, "Filter must be square.");
 
            T[] rgData = m_cuda.GetMemory(hMem);
            int nF = (int)Math.Ceiling(nWidth / 2.0);
            double dfC = (nWidth - 1) / (2.0 * nF);
 
            for (int i = 0; i < nCount; i++)
            {
                double dfX = i % nWidth;
                double dfY = (i / nWidth) % nHeight;
                double dfVal = (1 - Math.Abs(dfX / nF - dfC)) * (1 - Math.Abs(dfY / nF - dfC));
 
                rgData[i] = (T)Convert.ChangeType(dfVal, typeof(T));
            }
 
            m_cuda.SetMemory(hMem, rgData);
 
            m_log.CHECK_EQ(-1, m_param.sparse, "Sparsity not supported by this Filler.");
        }
    }
}