fast-reid/projects/FastRT/include/fastrt/factory.h

#pragma once

#include "struct.h"
#include "module.h"
#include "IPoolingLayerRT.h"

namespace fastrt {
    
    class ModuleFactory {
    public:
        ModuleFactory() = default;
        ~ModuleFactory() = default;

        std::unique_ptr<Module> createBackbone(FastreidConfig& modelCfg);
        std::unique_ptr<Module> createHead(FastreidConfig& modelCfg);
    };

    class LayerFactory {
    public:
        LayerFactory() = default;
        ~LayerFactory() = default;

        std::unique_ptr<IPoolingLayerRT> createPoolingLayer(const FastreidPoolingType& pooltype);
    };

}