玉米叶病害识别(图像连续识别和视频识别,python代码,pytorch框架)

 代码运行要求:Torch>=1.13.1即可

效果视频玉米叶病害识别(图像连续识别和视频识别,python代码,pytorch框架)_哔哩哔哩_bilibili

玉米叶病害识别(图像连续识别和视频识别,python代码,pytorch框架)_第1张图片 

 

  1.数据集介绍:

玉米叶病害识别(图像连续识别和视频识别,python代码,pytorch框架)_第2张图片

 

 Blight(枯萎病),1145张照片

Common_Rust(褐锈病,1306张照片) 

 Gray_Leaf_Spot(灰斑病,574张照片)

Healthy(健康,1162张照片) 

2.项目文件

玉米叶病害识别(图像连续识别和视频识别,python代码,pytorch框架)_第3张图片 

第一个文件夹(data): 装载的是原始图像

第二个文件夹(GUI):装载的是随意选取的图像,供vedio_creat.py处理后生成视频。

第三个文件夹(piture):装载的是经hf.py对data文件夹处理后,生成的训练集和测试集

 第四个文件是class_indices.json是装载的标签和对应类别名称

第五个文件:CNN.pth是装载训练好的模型参数

第六个文件:GUI_VEDIO.py是呈现GUI界面,包括对图像连续识别和对视频识别

第七个文件:hf.py是对data文件夹进行操作,生成训练集和测试集

第八个文件:model.py是模型 

第九个文件:predict.py是对单独的照片(tulip.jpg)进行识别

第十个文件:train.pys是训练脚本

两个录屏的视频均可以被识别。

对项目感兴趣的可以关注,代码和数据集已经放到压缩包,解压缩后无需配置路径,可以直接运行:

import threading
import os
import json
import torch
import cv2
from PIL import Image
from torchvision import transforms
import tkinter as tk
from tkinter import filedialog
from model import CNN
from PIL import ImageTk
#代码和数据集压缩包https://mbd.pub/o/bread/ZJ2XmpZv

你可能感兴趣的:(生物技术与深度学习结合,python,pytorch,开发语言)