yolov5实现水稻病虫害识别

1、制作合适的数据集

1.1 mask文件和yolov5放在同一个文件夹下

1.2 mask文件夹下包含两个文件（train和valid)和一个data.yaml
train:训练集，包含用于训练的图片和图片的位置信息（txt格式）
valid:验证集，包含用于验证的图片和图片的位置信息（txt格式）
data.yaml:用于存放train和valid的绝对地址以及类别总数（nc）、类别名称

1.3 train文件夹下包括两个文件夹，images用来存放图片，labels存放图片信息（txt格式包括类被和位置信息）。（train文件夹主要用来训练）

1.4 vaild文件夹下存放格式与train相同

1.5 data.yaml文件如下，train and val后面接地址，nc表示类别的总数，name后面是类别名称。

需要注意的是，txt文件中第一个是表示的某一类别，3是指在names数组中第4个类别

数据集到这就可以开始训练了

2、用自己的数据集进行训练
在pycharm下打开yolov5对应的文件夹，激活环境

conda activate yolov5

接着修改yolov5/models/yolov5s.yaml，将nc = 80修改为nc = 4.

把数据集准备好，并且修改好yolov5s.yaml里的参数，接下来就可以执行训练命令，生成一个针对你自己准备的数据集的

python train.py --img 640 --batch 16 --epoch 300 --data C:~\V1\mask\data.yaml --cfg models/yolov5s.yaml --weights weights/yolov5s.pt

data后面接的是上文1.5中data.yaml文件的完整路径
epoch是指训练完整个数据集的次数，可以自己指定
batch是指单次训练的图片数量，和显卡相关，如果报错就调小

经过漫长的等待，训练完成
生成的best.pt就是所得到的权重

3、测试生成的权重

python detect.py --weight runs\exp21\weights\best.pt --source C:\Users\Lenovo\Desktop\IMG_4390.JPG

detect.py:用于检测的程序
weight：选择刚训练好的权重路径
source：选择一张测试图片（没有用过的）的路径

打开得到图片

在这里说一下碰到的问题和我所用的解决的办法

一开始batch size的大小写的16，但是一直不运行也不报错，于是就开始减小，减小为8时，报错

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
You can try to repro this exception using the following code snippet. If that doesn't trigger the error, please include your original repro script when reporting this issue.

于是减小到1，然后开始运行了。

yolov5实现水稻病虫害识别相关推荐

tensorflow实现图像分类解决病虫害识别问题
一,任务本次做的是一个小麦水稻病虫害识别问题,使用tensorflow,利用CNN提取图片像素特征进行分类,并达到了一定的效果. 二,数据准备为了防止过拟合,数据预处理阶段分别经过亮度增强,对比度 ...
基于Yolov5的草莓病虫害检测识别
项目介绍上一篇文章介绍了基于卷积神经网络的交通标志分类识别Python交通标志识别基于卷积神经网络的保姆级教程(Tensorflow),并且最后实现了一个pyqt5的GUI界面,并且还制作了一个简单 ...
农作物病虫害识别进展概述（***）
最近看了Overview: Research Progress on Pest and Disease Identification这篇文章,这篇文章主要也就是对农业病虫害的识别的一个综述. 摘要近 ...
基于深度学习的农作物病虫害识别
摘要:我国有广阔的农作物种植面积,其中病虫害对农作物产量的影响最大,当农作物得了病虫害时,其整体生理机能会大大下降从而导致植株瘦小,无法达到最优生产状态从而产量不高经济效益低.因此农民需要多关注农作物 ...
农作物病虫害识别进展概述
最近看了Overview: Research Progress on Pest and Disease Identification这篇文章,这篇文章主要也就是对农业病虫害的识别的一个综述. 摘要近 ...
同济子豪兄github_【B站UP主-同济子豪兄】华为云ModelArts零代码开发病虫害识别应用...
大家好,我是B站UP主同济子豪兄,我希望让人工智能和大数据赋能每一个行业,每一个人. 今天给大家带来的是使用ModelArts快速开发农作物病虫害识别微信小程序,用人工智能的力量赋能农业,助力乡村振兴 ...
基于MATLAB的农业病虫害识别系统
摘要目前,图像处理技术的应用与研究都有了很大进展,其应用也已经渗透到农业生产的各个领域.在农业生产中,病虫害一直是困扰农作物生长的基本问题.因此研究农作物病虫草害的自动检测与识别,开发智能化控制系 ...
基于深度学习的农作物病虫害识别系统
1 简介今天向大家介绍一个帮助往届学生完成的毕业设计项目,基于深度学习的农作物病虫害识别系统. ABSTRACT 及时.准确地诊断植物病害,对于防止农业生产的损失和农产品的损失或减少具有重要作用.为 ...
基于MATALB的植物病虫害识别系统设计与实现(已完结)
(随便看看就好,主要是课设结束了电脑没空间了舍不得删) 本篇博客主要是有下面两个方面的内容: 一.App Designer工具设计交互界面二.MATLAB深度学习模块中VGG-19模型的使用目录 ...

yolov5实现水稻病虫害识别

yolov5实现水稻病虫害识别相关推荐

最新文章

热门文章