apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add YOLOv5Face model support (#38)

Browse Source 
* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

* add yolox c++/pybind and model_zoo demo

* move some helpers to private

* fixed CMakeLists.txt typos

* add normalize with alpha and beta

* add version notes for yolov5/yolov6/yolox

* add copyright to yolov5.cc

* revert normalize

* fixed some bugs in yolox

* Add YOLOv5Face Model support

* fixed examples/vision typos

* fixed runtime_option print func bugs
This commit is contained in:
DefTruth
2022-07-25 21:55:56 +08:00
committed by GitHub
parent 36fc77e6b8
commit fc71d79e58
27 changed files with 1240 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/vision/utils/utils.h
View File

@@ -53,9 +53,13 @@ std::vector<int32_t> TopKIndices(const T* array, int array_size, int topk) {
void NMS(DetectionResult* output, float iou_threshold = 0.5);
void NMS(FaceDetectionResult* result, float iou_threshold = 0.5);
// MergeSort
void SortDetectionResult(DetectionResult* output);
void SortDetectionResult(FaceDetectionResult* result);
} // namespace utils
} // namespace vision
} // namespace fastdeploy