图像分类模型批量预测图像到对应分类标签文件夹代码

最新推荐文章于 2022-11-23 20:29:05 发布

会飞的渔WZH

最新推荐文章于 2022-11-23 20:29:05 发布

阅读量458

点赞数

分类专栏： Tips 文章标签：分类 pytorch 深度学习

本文链接：https://blog.csdn.net/wuzhihuaw/article/details/127432257

版权

本文介绍如何使用PyTorch进行图像分类模型的批量预测，并将预测结果按照对应的分类标签归档到不同的文件夹中，详细阐述了深度学习模型在实际应用中的操作流程。

摘要由CSDN通过智能技术生成

import json

import torch
from PIL import Image

from model_v3 import mobilenet_v3_large

from torchvision import transforms

def main():
    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

    data_transform = transforms.Compose(
        [transforms.Resize(256),
         transforms.CenterCrop(224),
         transforms.ToTensor(),
         transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
    #导入库
    import os

    from tqdm import tqdm
    dir_origin_path='F:/Dataset/dataset5/parking_picture/'
    dir_save_path1='F:/Dataset/dataset5/illegal_parking/'
    if not os.path.exists