前一阵子其实已经在导师的“威逼利诱”下开始学了,直到昨天才突发奇想,想用博客记录一下自己是怎么从零开始,完成这本该属于是研究生同学去学习的还没入门就放弃的chao难项目!所以今天就把本该半个月前的文章补发吧。
目录
安装开发环境:
这一步我是在各大b站博主们的视频和csdn大佬们的博客共同帮助下,牺牲了两个晚上的优质睡眠后,才寥寥草草把项目给跑起来并成功测试的
本人在本篇也仅仅是粗略介绍流程,毕竟每个人的设备不一,遇到的问题都是千奇百怪,如果你正在苦恼环境安装我只能深表同情并鼓励你不要轻易放弃,去揣摩揣摩博主大佬们是如何解决的,也许是这项目开发必受的煎熬吧!
参考:b站博主<炮哥带你学>,他在csdn同名,也有很多博客可供学习参考
主要直装软件:
Aanaconda(用来创建虚拟开发环境,以及下载各种依赖包)
pycharm(用来运行、测试项目)
CUDA(为项目提供gpu算力,如果只用cpu可以忽略)
创建虚拟环境及下载依赖包:
在anaconda的终端窗口创建python虚拟环境,随后用命令下载各种依赖包,主要有:
torch,torchvision,cuda
这三个的版本是互相依赖、互相对应的,如果下错了要卸载重下,可以去pytorch官网复制下载命令到anaconda终端执行下载,也可以下载好.whl文件到本地安装,本人用方法是后者
完成这三个的下载,其实后面还有很多包是不必你自己手动命令下载了,在yolov5的项目中有一个requirement.txt文件,复制该文件第一行的pip install -r requirements.txt命令到pycharm的Terminal窗口,就会自动帮你完成剩下所有项目依赖包的下载和安装
运行train.py文件:
第一次运行项目,绝对不可能一帆风顺,你会遇到数不胜数的bug,根据报错提示去网上慢慢改吧!
一直改bug直到出现上图这样,在稳定完成一轮又一轮的训练,这才说明项目成功跑起来了,当然我不是在保证你的项目一定能够成功训练到结束哈
运行detect.py文件(即项目测试):
train是为了训练模型,detect则用来实现我们项目本身最想实现的功能,也就是目标检测,按照各平台各博主的教学,成功跑通代码就能出结果(当然这里一定还有一堆bug等着你hhh)
尾声:
再次强调本篇只是用来记录本人自己的学习经历而非正经教学,当然你可以把你遇到的问题评论或私信给我,也许我能提供帮助。不太能帮助到你的话我很抱歉,我能做的就是,当你在浩瀚的互联网中焦头烂额地查找教学但似乎依旧摸不着头脑、在放弃的边缘徘徊之时,给你加油鼓气,这是一定要经历的过程,毕竟,我也是这么过来的/(ㄒoㄒ)/~~!