mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Pin "paddlepaddle<3.0.0" to resolve export bug

pull/13550/head
Glenn Jocher 2025-03-27 08:54:46 +01:00
parent 61c22e2d20
commit 44da251b25
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
export.py
View File

@ -510,7 +510,7 @@ def export_paddle(model, im, file, metadata, prefix=colorstr("PaddlePaddle:")):
$ pip install paddlepaddle x2paddle
```
"""
check_requirements(("paddlepaddle", "x2paddle"))
check_requirements(("paddlepaddle<3.0.0", "x2paddle"))
import x2paddle
from x2paddle.convert import pytorch2paddle

2
models/common.py
View File

@ -646,7 +646,7 @@ class DetectMultiBackend(nn.Module):
raise NotImplementedError("ERROR: YOLOv5 TF.js inference is not supported")
elif paddle: # PaddlePaddle
LOGGER.info(f"Loading {w} for PaddlePaddle inference...")
check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle")
check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle<3.0.0")
import paddle.inference as pdi
if not Path(w).is_file(): # if not *.pdmodel