土豆叶病害识别(图像连续识别和视频识别)

效果视频:土豆叶病害识别(Python代码,pyTorch框架,视频识别)_哔哩哔哩_bilibili

代码运行要求:Torch库>=1.13.1,其它库无版本要求

1..土豆叶数据集主要包好三种类别(Early_Blight(早期枯萎病),Late_Blight(晚期枯萎病)Healthy(正常))

Early_Blight 照片(有1303张)

Late_Blight 照片(有1132张)

Healthy照片(有816张)

2,本次项目文件夹

第一个文件夹(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是训练脚本

第十一个文件:vedio.mp4,是以一帧一秒的速度,将一个个的图像经vedio_creat.py处理后,生成视频,以模拟无人机采集的的农业视频,做实时检测。视频识别的时候,也是以一秒一帧的速度取图像。

对项目感兴趣的可以关注:

复制代码
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/ZJ2Xl59y
相关推荐
kk爱闹几秒前
【挑战14天学完python和pytorch】- day01
android·pytorch·python
Ai墨芯1118 分钟前
靠机器学习+组合优化就发了CCF-A
人工智能
Morpheon10 分钟前
揭开预训练Pre-Training的力量:革新机器学习
人工智能·机器学习
勤奋的大熊猫11 分钟前
机器学习中的 Agent 是什么?
人工智能·机器学习·agent
Blossom.11811 分钟前
机器学习在智能建筑中的应用:能源管理与环境优化
人工智能·python·深度学习·神经网络·机器学习·机器人·sklearn
明朝百晓生14 分钟前
深入理解Vapnik-Chervonenkis(VC)维度:机器学习泛化能力的理论基础
人工智能·机器学习
信息快讯14 分钟前
机器学习驱动的智能化电池管理技术与应用
人工智能·机器学习·锂离子电池
勤奋的大熊猫15 分钟前
机器学习路径规划中的 net 和 netlist 分别是什么?
人工智能·机器学习·自动寻路
还有糕手16 分钟前
西南交通大学【机器学习实验6】
人工智能·机器学习
亚力山大抵21 分钟前
实验六-使用PyMySQL数据存储的Flask登录系统-实验七-集成Flask-SocketIO的实时通信系统
后端·python·flask