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

[Bug Fix] Fix bugs when deploy quantized YOLOv5/v6/v7 model. (#729)

Browse Source 
* Improve the usage of fastdeploy tools

* Fix quantized YOLOv5,v6 and v7 model deployment
This commit is contained in:
yunyaoXYY
2022-11-28 19:19:37 +08:00
committed by GitHub
parent 941057888a
commit dc2dad62a4
14 changed files with 12 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/fastdeploy/vision/detection/contrib/yolov5.py
View File

@@ -151,7 +151,6 @@ class YOLOv5(FastDeployModel):
# 初始化后的option保存在self._runtime_option
super(YOLOv5, self).__init__(runtime_option)
assert model_format == ModelFormat.ONNX, "YOLOv5 only support model format of ModelFormat.ONNX now."
self._model = C.vision.detection.YOLOv5(
model_file, params_file, self._runtime_option, model_format)
# 通过self.initialized判断整个模型的初始化是否成功

1
python/fastdeploy/vision/detection/contrib/yolov7.py
View File

@@ -137,7 +137,6 @@ class YOLOv7(FastDeployModel):
# 初始化后的option保存在self._runtime_option
super(YOLOv7, self).__init__(runtime_option)
assert model_format == ModelFormat.ONNX, "YOLOv7 only support model format of ModelFormat.ONNX now."
self._model = C.vision.detection.YOLOv7(
model_file, params_file, self._runtime_option, model_format)
# 通过self.initialized判断整个模型的初始化是否成功