运行YOLOV5时numpy版本报错调试

最新推荐文章于 2025-03-30 00:53:35 发布

水花花花花花

最新推荐文章于 2025-03-30 00:53:35 发布

阅读量2k

点赞数 33

分类专栏：人工智能文章标签： YOLO numpy python

本文链接：https://blog.csdn.net/m0_73776435/article/details/137029395

版权

文章讲述了在使用yolov5进行训练时遇到的关于numpy版本升级导致的报错，涉及np.int的缺失。给出了两种解决方案：一是降级numpy到1.20.0版本，二是升级到至少1.22.4版本并保持兼容。建议谨慎修改源码以避免更多错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天在运行yolov5训练时报错：

Traceback (most recent call last):
  File "/home/wh/projects/DenseNet_Demo/train_resnet.py", line 17, in <module>
    from torchtoolbox.transform import Cutout
  File "/home/wh/anaconda3/envs/pytorch39/lib/python3.9/site-packages/torchtoolbox/transform/__init__.py", line 5, in <module>
    from .autoaugment import *
  File "/home/wh/anaconda3/envs/pytorch39/lib/python3.9/site-packages/torchtoolbox/transform/autoaugment.py", line 194, in <module>
    Compose([Posterize(0.4, 8), Rotate(0.6, 9)]),
  File "/home/wh/anaconda3/envs/pytorch39/lib/python3.9/site-packages/torchtoolbox/transform/autoaugment.py", line 104, in __init__
    ranges = np.round(np.linspace(8, 4, 10), 0).astype(np.int)
  File "/home/wh/anaconda3/envs/pytorch39/lib/python3.9/site-packages/numpy/__init__.py", line 305, in __getattr__
    raise AttributeError(__former_attrs__[attr])
AttributeError: module

最低0.47元/天解锁文章