Как я могу применить переменную изображения к нейронной сети Pytorch, обученной на преобразованных данных? - PullRequest
0 голосов
/ 13 декабря 2018

Нейронная сеть была обучена с использованием входных данных, которые были импортированы и преобразованы следующим образом:

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 и все преобразования, чтобы превратить этот массив в пакет из одного элемента для передачи через мою нейронную сеть?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...