1..马铃薯数据集主要包好三种类别(Early_Blight(早期枯萎病),Late_Blight(晚期枯萎病)Healthy(正常))而且数据集已经分好训练集、验证集和测试集
Early_Blight 照片(训练集有1303张,验证集有163张,测试集有162张)
Late_Blight 照片(训练集有1132张,验证集有151张,测试集有141张)
Healthy照片(训练集有816张,验证集有102张,测试集有102张)
2.以Training文件夹为例展示
Healthy文件夹
Early_Blight(早期枯萎病)
Late_Blight(晚期枯萎病)文件夹
import os
import sys
import json
import torch
import torch.nn as nn
from torchvision import transforms, datasets
import torch.optim as optim
from tqdm import tqdm
#数据集;https://mbd.pub/o/bread/mbd-ZJibmp5v
# from model import GoogLeNet
from model import CNN
import os
from shutil import copy
import random