InputLayer.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using MyCaffe.basecode;
using MyCaffe.common;
using MyCaffe.param;
 
namespace MyCaffe.layers
{
    public class InputLayer<T> : Layer<T>
    {
        public InputLayer(CudaDnn<T> cuda, Log log, LayerParameter p) 
            : base(cuda, log, p)
        {
            m_type = LayerParameter.LayerType.INPUT;
        }
 
        public override void LayerSetUp(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
            int nNumTop = colTop.Count;
            InputParameter p = m_param.input_param;
            int nNumShape = p.shape.Count();
 
            m_bUseHalfSize = m_param.use_halfsize;
            m_log.CHECK(nNumShape == 0 || nNumShape == 1 || nNumShape == nNumTop, "Must specify 'shape' once, once per top blob, or not at all: " + nNumTop.ToString() + " top vs. " + nNumShape.ToString() + " shapes.");
 
            if (nNumShape > 0)
            {
                for (int i = 0; i < nNumTop; i++)
                {
                    int nShapeIdx = (p.shape.Count() == 1) ? 0 : i;
                    colTop[i].Reshape(p.shape[nShapeIdx], m_bUseHalfSize);
                }
            }
 
            m_bConvertTopOnFwd = m_param.use_halfsize;
        }
 
        public override void Reshape(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        public override int ExactNumBottomBlobs
        {
            get { return 0; }
        }
 
        public override int MinTopBlobs
        {
            get { return 1; }
        }
 
 
        protected override void forward(BlobCollection<T> colBottom, BlobCollection<T> colTop)
        {
        }
 
        protected override void backward(BlobCollection<T> colTop, List<bool> rgbPropagateDown, BlobCollection<T> colBottom)
        {
        }
    }
}