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数据,这个要慢慢看调试改错