pytorch猫狗分类测试 包括批量及单张图片测试

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import os
from torchvision import datasets,transforms
from torch.utils.data import DataLoader
import cv2
import numpy as np

root = "D:\找工作\暑期学习\cats_and_dogs_filtered"
test_data = os.path.join(root,"validation")

BATCH = 2
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
transform = transforms.Compose([transforms.RandomResizedCrop(150),
                                transforms.ToTensor()
                                ])##transforms.Normalize([0.5,0.5,0.5],[0.5,0.5,0.5])

test_gen = datasets.ImageFolder(test