[AI]python使用googLeNet实现鲜花分类，准确度高达99.9%

阿天ATean

于 2024-08-11 10:56:51 发布

阅读量151

点赞数 5

分类专栏：人工智能深度学习文章标签：深度学习 pytorch 人工智能

本文链接：https://blog.csdn.net/cnngavij/article/details/141102678

版权

import os.path

import torch
import torch.nn as nn
import torchvision.models as models
from torch.utils.data import DataLoader
from torchsummary import summary
from torchvision.datasets import ImageFolder
from torchvision.transforms import transforms

model = models.googlenet(pretrained=True)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

model_path = "./model/googleNet_cuda2.pth"
model = models.googlenet().to(device)
if os.path.exists(model_path):
    model.load_state_dict(torch.load(model_path))
def train():
    model.train()
    # summary(model=model, input_size=(3, 227, 227), batch_size=1, device="cpu")
    transform = transforms.Compose([
        transforms.ToTensor(),
        transforms.Resize((224, 224)),
    ])
    train = ImageFolder("./flower_datas/train", transform=transform)
    # test = ImageFolder("./flower_datas/val