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

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