yolov5 train报错：TypeError: expected np.ndarray (got numpy.ndarray)

前言

mac intel 机器上，使用 yolov5 物体检测训练时报错：TypeError: expected np.ndarray (got numpy.ndarray)

这个错误信息 TypeError: expected np.ndarray (got numpy.ndarray) 看起来有些奇怪，因为 numpy.ndarray 实际上就是 np.ndarray。

通常情况下，这种错误可能是由于库版本不兼容，需要确保使用的 NumPy 和 PyTorch 版本是兼容的。

解决

1.卸载当前的 numpy 版本：

复制代码

pip uninstall numpy

2.安装兼容的 numpy 版本：

复制代码

pip install numpy==1.23.0

3.验证安装的 numpy 版本：

复制代码

python -c "import numpy; print(numpy.__version__)"

4.重新安装 ultralytics 和 torch：

复制代码

pip install ultralytics torch

5.确保所有依赖项都已正确安装并且版本兼容：

复制代码

pip list

通过这些步骤，你应该能够解决版本不兼容的问题，并确保代码正常运行。