- 博客(2)
- 收藏
- 关注
RSS订阅原创 解决Yolov5-7-0 运行classify/train.py时，报错：AttributeError: module ‘PIL.Image‘ has no attribute ‘ANTIALIAS‘
可以运行classify/train.py，但requirements.txt文件中 ['Pillow>=10.0.1'] ，会自动更新Pillow-10.1.0，并在重新启动运行时或重新运行命令以使更新生效，需要在每次运行前重新安装Pillow或修改requirements.txt文件，或者使用方法二。的10.0.0版本中ANTIALIAS方法被删除，ANTIALIAS方法被删除需要使用PIL.Image.LANCZOS或PIL.Image.Resampling.LANCZOS。修改完成后可以正确运行。
2023-11-24 23:59:15
1158
2
原创 解决yolov5 使用GTX 1660 Ti训练时出现NAN的问题
yolov5训练时， box obj cls 为 nan，P R mAP@.5 mAP@.5:.95为 0 的问题。PyTorch版本与CUDA版本不匹配，导致（float16）数据类型在卷积等一些运算的时候会出现nan值。效果：训练时间从03: 40增加到06: 11，但显存增加了很多，考虑到显存可能不够的原因，决定换解决办法。修改train.py，不使用自动混合精度（amp），以及半精度浮点型数据改为单精度的浮点型运算。效果：训练时间比方法一略久一点，但显存是方法一的一半。
2023-11-17 19:19:09
1622
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人