mirror of https://github.com/JDAI-CV/fast-reid.git
25 lines
680 B
C++
25 lines
680 B
C++
#pragma once
|
|
|
|
#include "model.h"
|
|
#include "struct.h"
|
|
#include <memory>
|
|
#include <opencv2/opencv.hpp>
|
|
using namespace trtxapi;
|
|
|
|
namespace fastrt {
|
|
|
|
class Baseline : public Model {
|
|
public:
|
|
Baseline(const trt::ModelConfig &modelcfg, const FastreidConfig& reidcfg);
|
|
~Baseline() = default;
|
|
|
|
private:
|
|
void preprocessing_cpu(const cv::Mat& img, float* const data, const std::size_t stride);
|
|
ITensor* preprocessing_gpu(INetworkDefinition* network,
|
|
std::map<std::string, Weights>& weightMap,
|
|
ITensor* input);
|
|
private:
|
|
std::string _input_name{"data"};
|
|
std::string _output_name{"embd"};
|
|
};
|
|
} |