YOLOv8训练时报错ERROR:albumentations.check_version:Error fetching version info

原因:

根本原因在于 check_version.py 文件中设置的超时时长太短,加载像Github这样的网站较慢,

解决方法:

在ultralytics/yolo/v8/detect/train.py文件中添加代码如下

1.导入包

import warnings
import os

2.修改ultralytics/yolo/v8/detect/train.py中运行代码如下

if __name__ == '__main__':
    # 忽略更新,解决:ERROR:albumentations.check_version:Error fetching version info
    os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'
    # 忽略 UserWarning
    warnings.filterwarnings("ignore", category=UserWarning)
    train()

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐