yolov5 train報錯:TypeError: expected np.ndarray (got numpy.ndarray)

牛奔發表於2024-10-29

前言

mac intel 機器上,使用 yolov5 物體檢測訓練時報錯:TypeError: expected np.ndarray (got numpy.ndarray)

這個錯誤資訊 TypeError: expected np.ndarray (got numpy.ndarray) 看起來有些奇怪,因為 numpy.ndarray 實際上就是 np.ndarray

通常情況下,這種錯誤可能是由於庫版本不相容,需要確保使用的 NumPyPyTorch 版本是相容的。

解決

1.解除安裝當前的 numpy 版本:

pip uninstall numpy

2.安裝相容的 numpy 版本:

pip install numpy==1.23.0

3.驗證安裝的 numpy 版本:

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

4.重新安裝 ultralyticstorch

pip install ultralytics torch

5.確保所有依賴項都已正確安裝並且版本相容:

pip list

透過這些步驟,你應該能夠解決版本不相容的問題,並確保程式碼正常執行。

相關文章