mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
821dc756e6
* Fix runtime with python * Add CenterNet/PicoDet/PPYOLO/PPYOLOv2/YOLOv3 * add more ppdet models * add model * fix some usage bugs for detection models
26 lines
691 B
C++
26 lines
691 B
C++
#pragma once
|
|
#include "fastdeploy/vision/ppdet/ppyoloe.h"
|
|
|
|
namespace fastdeploy {
|
|
namespace vision {
|
|
namespace ppdet {
|
|
|
|
class FASTDEPLOY_DECL PPYOLO : public PPYOLOE {
|
|
public:
|
|
PPYOLO(const std::string& model_file, const std::string& params_file,
|
|
const std::string& config_file,
|
|
const RuntimeOption& custom_option = RuntimeOption(),
|
|
const Frontend& model_format = Frontend::PADDLE);
|
|
|
|
virtual std::string ModelName() const { return "PaddleDetection/PPYOLO"; }
|
|
|
|
virtual bool Preprocess(Mat* mat, std::vector<FDTensor>* outputs);
|
|
virtual bool Initialize();
|
|
|
|
protected:
|
|
PPYOLO() {}
|
|
};
|
|
} // namespace ppdet
|
|
} // namespace vision
|
|
} // namespace fastdeploy
|