Нейронная сеть была обучена с использованием входных данных, которые были импортированы и преобразованы следующим образом:
transform = transforms.Compose(
[transforms.ToTensor()])
data_path = '/home/tim/Documents/trainset/'
traindataset = torchvision.datasets.ImageFolder(
root=data_path,
transform=transform
)
trainloader = torch.utils.data.DataLoader(
traindataset,
batch_size=3,
num_workers=1,
shuffle=True
)
Теперь у меня есть один массив numy, называемый img, который представляет изображение в RGB со значениями пикселей в диапазоне отОт 0 до 255. Как я могу применить trainloader и все преобразования, чтобы превратить этот массив в пакет из одного элемента для передачи через мою нейронную сеть?