mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
15 lines
305 B
C
15 lines
305 B
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <map>
|
||
|
|
#include "NvInfer.h"
|
||
|
|
#include "fastrt/struct.h"
|
||
|
|
using namespace nvinfer1;
|
||
|
|
|
||
|
|
namespace fastrt {
|
||
|
|
|
||
|
|
IScaleLayer* embedding_head(INetworkDefinition *network,
|
||
|
|
std::map<std::string, Weights>& weightMap,
|
||
|
|
ITensor& input,
|
||
|
|
const FastreidConfig& reidCfg);
|
||
|
|
|
||
|
|
}
|