0227期基于python深度学习的是否有污渍识别-含数据集

本文介绍了使用Python和PyTorch实现的污渍识别项目,包括下载代码、环境设置、数据集预处理、模型训练、以及创建可视化UI界面进行图像识别。训练过程中生成的日志记录了每个epoch的验证性能。
摘要由CSDN通过智能技术生成

代码下载和视频演示地址:

0227期基于python深度学习的是否有污渍识别_哔哩哔哩_bilibili

本代码是基于python pytorch环境安装的。

下载本代码后,有个环境安装的requirement.txt文本

数据集介绍,下载本资源后,界面如下:

数据集文件夹存放了本次识别的各个类别图片。

本代码对数据集进行了预处理,包括通过在较短边增加灰边,使得图片变为正方形(如果图片原本就是正方形则不会增加灰边),和旋转角度,来扩增增强数据集,

【资源说明】 【博主环境】 *可以在此检测项目基础上增加计数功能,统计当前画面目标总数,或者增加追踪功能,实现追踪计数! python==3.8 pytorch==1.8.1 torchvision==0.9.1 1、搭建环境 建议在anaconda中新建虚拟环境配置,然后在pycharm打开工程,再导入anaconda环境 确保正确安装requirements.txt中的包,可用清华源,下载块! 2、训练好的模型+评估指标曲线+数据集可视化图存放在“ultralytics\yolo\v8\detect\runs\detect”文件夹 3、开始检测识别 a.打开predict.py修改34行模型路径,照葫芦画瓢修改; b.需要检测的图片或视频预先存放在“\ultralytics\assets”文件夹 c.运行predict.py,开始检测检测结果会保存在ultralytics/yolo/v8/detect/runs/detect文件夹下 4、训练自己的模型 a.准备数据集,可参考YOLOv5,拆分为train、val即可,标签为txt b.在yolo\v8\detect\data文件夹下新建.yaml文件,照葫芦画瓢,仿照coco128.yaml c.修改tarin.py中的238行,改成自己新建yaml的路径 d.GPU训练(注释掉241行,修改device参数为0),若CPU训练(注释掉242行即可) e.运行train.py开始训练,当精度不在增加时,会自动停止训练。模型保存在ultralytics\yolo\v8\detect\runs\detect文件夹 【备注】 1、该资源内项目代码都经过测试运行成功,功能ok的情况下才上传的,请放心下载使用!有问题请及时沟通交流。 2、适用人群:计算机相关专业(如计科、信息安全、数据科学与大数据技术、人工智能、通信、物联网、自动化、电子信息等)在校学生、专业老师或者企业员工下载使用。 3、用途:项目具有较高的学习借鉴价值,也适用于小白学习入门进阶。当然也可作为毕设项目、课程设计、大作业、初项目立项演示等。 4、如果基础还行,或者热爱钻研,亦可在此项目代码基础上进行修改添加,实现其他不同功能。 欢迎下载,沟通交流,互相学习,共同进步!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值