所需的环境:cuda,cudnn,pytorch以及anaconda、pycharm。
- 首先cuda跟cudnn要装对版本,这个根据自己电脑显卡,
- 就是anaconda的用法,这个在以前我都不知道,原来anaconda这么棒,可以建立起虚拟的环境,就不需要在电脑中装那么多版本的python跟软件包,而且很乱
conda create -n 虚拟环境名 python=3.8 建立一个python3.8的虚拟环境
activaye 虚拟环境名 启用虚拟环境
- Pytorch的安装,去pytorch官网直接根据需要将将其装进虚拟环境中具体网上见教程。
- 配置pycharm,打开pycharm,新建项目,选择现有解释器,然后找到anaconda的安装位置中的anaconda/env/虚拟环境名/python.exe,创建。
- 数据集的标注,根据命令 pip install labelimg 下载安装labelimg,使用labelimg对数据进行标注并保存。网上参考教程 labelimg的使用
- 数据集划分,使用网上的一个来实现的(95条消息) 目标检测---数据集格式转化及训练集和验证集划分_炮哥带你学的博客-CSDN博客_目标检测验证集 主要遇见的bug是:UnicodeDecodeError: 'gbk' codec can't decode byte 0x86 in position 121,这个通过给open()函数中加上限制条件encoding=”UTF-8”来强制让文件解码时用UTF-8编码解决。如果此方法不好用,再试一试改变缩进。 SyntaxError: unexpected EOF while parsing 这个bug是由于修改文件错误,代码缺少括号或者中英文符号错误,实在找不出就把代码复制下来新建一个文件,再复制进去。
- YOLOV5的搭建,从GitHub上下载下来就行,(94条消息) 目标检测---教你利用yolov5训练自己的目标检测模型_炮哥带你学的博客-CSDN博客_yolov5训练自己的模型 根据这篇博客我下的是5.5.0版本,然后照着里面做,遇到的bug:运行前要把train.py中参数设置好,同时pycharm的运行配置打开,将工作空间改为项目的位置,不然设置的data.yaml 以及model.yaml文件会找不到。
- 运行train.py开始进行训练,遇到的bug:缺少pycocotools包,就去这个博客(95条消息) win10背景下Yolov5官方代码中pycoctools报错解决方法_一个客户,两个客户的博客-CSDN博客
AttributeError: Cant get attribute SPPF on module models 这条bug是因为model/common.py文件中没有SPPF类,看博客(94条消息) AttributeError: Cant get attribute SPPF on module models_star__king的博客-CSDN博客
没有wandb工具 直接运行pip install wandb安装
wandb.errors.UsageError: api_key not configured (no-tty). call wandb.login(key=[your_api_key]) 这条错误是wandb没有设置,要运行wandb init 然后弹出网址,进入网站注册,会出现密钥,将密钥复制输入就行。见博客(94条消息) 成功解决:wandb.errors.UsageError: api_key not configured (no-tty). call wandb.login(key=[your_api_key])_Jay_Mapp的博客-CSDN博客