图书馆网站结构怎么做,国外超酷设计网站,定制网络推广计划,做网站咨询1.效果视频#xff1a;36种水果和蔬菜识别#xff08;pytorch框架#xff0c;深度卷积网络模型#xff0c;可以实现照片连续识别和视频识别#xff09;_哔哩哔哩_bilibili 2.项目文件夹 第一个文件夹#xff08;data#xff09;#xff1a; 装载的是原始图像
第二个文…1.效果视频36种水果和蔬菜识别pytorch框架深度卷积网络模型可以实现照片连续识别和视频识别_哔哩哔哩_bilibili 2.项目文件夹 第一个文件夹data 装载的是原始图像
第二个文件夹GUI装载的是随意选取的图像供录屏生成视频。
第三个文件夹piture:装载的是经hf.py对data文件夹处理后生成的训练集和测试集 第四个文件是class_indices.json是装载的标签和对应类别名称
第五个文件CNN.pth是装载训练好的模型参数
第六个文件GUI_VEDIO.py是呈现GUI界面包括对图像连续识别和对视频识别
第七个文件hf.py是对data文件夹进行操作生成训练集和测试集
第八个文件model.py是模型
第九个文件predict.py是对单独的照片tulip.jpg进行识别
第十个文件train.pys是训练脚本
对项目感兴趣的可以关注代码和数据集已经放到压缩包因为上传内容不能超过100M,所以这里把data数据集放在了百度网盘里数据下载后可以按照项目文件夹照片里呈现的样子跟代码放在同一目录下然后运行hf.py生成训练集和测试集。CNN.pth是训练好的模型参数可以直接被GUI_vedio.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
#数据集和代码https://mbd.pub/o/bread/ZJ2Ympdv