深度学习入门——yolov5模型目标检测(人与车的辨别)

1.关于anaconda和pytorch的下载以及各种配置

参考下面这篇文章,真的很详细,按着操作即可

http://t.csdnimg.cn/Rf2lYicon-default.png?t=N7T8http://t.csdnimg.cn/Rf2lY

2.自制数据集

这里原本用labelimg扣标签的,后面发现我把xml格式转txt格式的代码走不通,找不到解决方法,所以我用了make sense(贴个链接Make Senseicon-default.png?t=N7T8https://www.makesense.ai/),标签导出直接就是txt格式了。

3.训练集和验证集的划分

按道理来说这里本来应该随机分的,但是想了想手动分应该没什么大问题,再加上代码问题,真就手工训练集和验证集的划分处理了,花了有五六个小时才把两个集搞好。

4.模型下载以及预训练权重下载

还是参考这篇文章下载,很全http://t.csdnimg.cn/tw6spicon-default.png?t=N7T8http://t.csdnimg.cn/tw6sp

嫌进网站慢可以换个外国的vpn

不过里面还是有点小问题,有些代码是老版本的需要改改,其中有一些pycharm会提醒你,下个网易有道翻译实时翻译就行,还有一些没提醒的在下面这篇文章里找http://t.csdnimg.cn/nImYbicon-default.png?t=N7T8http://t.csdnimg.cn/nImYb

这样一套流程下来基本问题就解决了

5.模型训练

最开始用的是5s的权重,准度低但是真的快,然后试了一把5x的权重,笔记本跑不动(哭),

最后折中选了5m的权重,跑了一个半小时,结果还算能打

附上一些图片检测结果

  • 14
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值