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

[Model] Support YOLOv8 (#1137)

Browse Source 
* add GPL lisence

* add GPL-3.0 lisence

* add GPL-3.0 lisence

* add GPL-3.0 lisence

* support yolov8

* add pybind for yolov8

* add yolov8 readme

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
WJJ1995
2023-01-16 11:24:23 +08:00
committed by GitHub
parent a4b94b2c93
commit 02bd22422e
28 changed files with 1448 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/fastdeploy/vision/detection/contrib/yolov5seg.py Normal file → Executable file
View File

@@ -180,7 +180,7 @@ class YOLOv5Seg(FastDeployModel):
model_file, params_file, self._runtime_option, model_format)
assert self.initialized, "YOLOv5Seg initialize failed."
def predict(self, input_image, conf_threshold=0.25, nms_iou_threshold=0.5):
def predict(self, input_image):
"""Detect an input image
:param input_image: (numpy.ndarray)The input image data, 3-D array with layout HWC, BGR format
@@ -189,8 +189,6 @@ class YOLOv5Seg(FastDeployModel):
:return: DetectionResult
"""
self.postprocessor.conf_threshold = conf_threshold
self.postprocessor.nms_threshold = nms_iou_threshold
return self._model.predict(input_image)
def batch_predict(self, images):