yolov5中train.py运行配置以及报错解决方案

1.依赖库版本要求

yolov5中不同版本的依赖库要求可以在requirements.txt查看。根据你的yolov5版本,选择好满足条件的版本。最新版本链接:

yolov5/requirements.txt at master · ultralytics/yolov5 (github.com)

2.parser = argparse.ArgumentParser()参数配置

这里只需要更改--data的参数,改成训练数据的路径,如下图:

当然也可以在代码中直接修改:

 3.报错信息:FailedPreconditionError: runs is not a directory

解决方案: 删除runs文件夹

4.报错信息: OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading "C:\Users\NL\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\lib\cudnn_adv_train64_8.dll" or one of its dependencies.

原因:batch_size太大
大概在410行处

解决方法,将batch_size改成8或者更小 

5.报错信息:Tensor******

因为有些数据不是tensor数据,这个要慢慢看调试改错

猜你喜欢

转载自blog.csdn.net/AI_dataloads/article/details/134428018