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] Fix nms problem (#2230)

Browse Source 
fix nms problem
This commit is contained in:
Jason
2023-10-10 01:24:33 -07:00
committed by GitHub
parent aab27b3db5
commit d910d6116c
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/fastdeploy/vision/detection/ppdet/__init__.py
View File

@@ -494,7 +494,6 @@ class PaddleYOLOv5(PPYOLOE):
super(PPYOLOE, self).__init__(runtime_option)
assert model_format == ModelFormat.PADDLE, "PaddleYOLOv5 model only support model format of ModelFormat.Paddle now."
self._model = C.vision.detection.PaddleYOLOv5(
model_file, params_file, config_file, self._runtime_option,
model_format)