LayerFactory.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.db.image;
using MyCaffe.param;
 
namespace MyCaffe.layers.ssd
{
    public class LayerFactory : ILayerCreator
    {
        public Layer<double> CreateDouble(CudaDnn<double> cuda, Log log, LayerParameter p, CancelEvent evtCancel, IXDatabaseBase db)
        {
            switch (p.type)
            {
                case LayerParameter.LayerType.ANNOTATED_DATA:
                    return new AnnotatedDataLayer<double>(cuda, log, p, db, evtCancel);
 
                case LayerParameter.LayerType.DETECTION_EVALUATE:
                    return new DetectionEvaluateLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.DETECTION_OUTPUT:
                    return new DetectionOutputLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.MULTIBOX_LOSS:
                    return new MultiBoxLossLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.NORMALIZATION2:
                    return new Normalization2Layer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.PERMUTE:
                    return new PermuteLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.PRIORBOX:
                    return new PriorBoxLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.SMOOTHL1_LOSS:
                    return new SmoothL1LossLayer<double>(cuda, log, p);
 
                case LayerParameter.LayerType.VIDEO_DATA:
                    return new VideoDataLayer<double>(cuda, log, p, db, evtCancel);
 
                default:
                    return null;
            }
        }
 
        public Layer<float> CreateSingle(CudaDnn<float> cuda, Log log, LayerParameter p, CancelEvent evtCancel, IXDatabaseBase db)
        {
            switch (p.type)
            {
                case LayerParameter.LayerType.ANNOTATED_DATA:
                    return new AnnotatedDataLayer<float>(cuda, log, p, db, evtCancel);
 
                case LayerParameter.LayerType.DETECTION_EVALUATE:
                    return new DetectionEvaluateLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.DETECTION_OUTPUT:
                    return new DetectionOutputLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.MULTIBOX_LOSS:
                    return new MultiBoxLossLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.NORMALIZATION2:
                    return new Normalization2Layer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.PERMUTE:
                    return new PermuteLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.PRIORBOX:
                    return new PriorBoxLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.SMOOTHL1_LOSS:
                    return new SmoothL1LossLayer<float>(cuda, log, p);
 
                case LayerParameter.LayerType.VIDEO_DATA:
                    return new VideoDataLayer<float>(cuda, log, p, db, evtCancel);
 
                default:
                    return null;
            }
        }
    }
}