mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[Model] Support DINO & DETR and add PaddleDetectionModel class (#1837)
* 添加paddleclas模型 * 更新README_CN * 更新README_CN * 更新README * update get_model.sh * update get_models.sh * update paddleseg models * update paddle_seg models * update paddle_seg models * modified test resources * update benchmark_gpu_trt.sh * add paddle detection * add paddledetection to benchmark * modified benchmark cmakelists * update benchmark scripts * modified benchmark function calling * modified paddledetection documents * add PaddleDetectonModel * reset examples/paddledetection * resolve conflict * update pybind * resolve conflict * fix bug * delete debug mode * update checkarch log * update trt inputs example * Update README.md --------- Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
Regular → Executable
+17
@@ -18,6 +18,7 @@ PPDetBase::PPDetBase(const std::string& model_file,
|
||||
runtime_option.model_format = model_format;
|
||||
runtime_option.model_file = model_file;
|
||||
runtime_option.params_file = params_file;
|
||||
|
||||
}
|
||||
|
||||
std::unique_ptr<PPDetBase> PPDetBase::Clone() const {
|
||||
@@ -82,6 +83,22 @@ bool PPDetBase::BatchPredict(const std::vector<cv::Mat>& imgs,
|
||||
return true;
|
||||
}
|
||||
|
||||
bool PPDetBase::CheckArch(){
|
||||
std::vector<std::string> archs = {"SOLOv2","YOLO","SSD","RetinaNet","RCNN","Face","GFL","YOLOX","YOLOv5","YOLOv6","YOLOv7","RTMDet","FCOS","TTFNet","TOOD","DETR"};
|
||||
auto arch_ = preprocessor_.GetArch();
|
||||
for (auto item : archs) {
|
||||
if (arch_ == item) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
FDWARNING << "Please set model arch,"
|
||||
<< "support value : SOLOv2, YOLO, SSD, RetinaNet, RCNN, Face , GFL , RTMDet ,"\
|
||||
<<"FCOS , TTFNet , TOOD , DETR." << std::endl;
|
||||
return false;
|
||||
|
||||
|
||||
}
|
||||
|
||||
} // namespace detection
|
||||
} // namespace vision
|
||||
} // namespace fastdeploy
|
||||
|
||||
Reference in New Issue
Block a user