mirror of https://github.com/JDAI-CV/fast-reid.git
26 lines
584 B
C++
26 lines
584 B
C++
#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);
|
|
};
|
|
|
|
} |