Модель Pytorch Выбор случайных входных изображений - PullRequest
Новая
       70

Модель Pytorch Выбор случайных входных изображений

0 голосов
/ 16 февраля 2020

Я применил следующую ссылку для моего набора данных двоичной классификации: https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html Я взял идентификаторы изображений и соответствующие имена классов из файла json.

Код обучает, рассматривая все доступные элементы в обоих классах. Мне нужно выбрать случайно сгенерированные несколько элементов из каждого класса.

Я пытался получить случайные элементы через загрузчик данных, но не смог этого сделать: 

`image_datasets = {x: datasets.ImageFolder(os.path.join(data_dir, x),
                                          data_transforms[x])
                  for x in ['train', 'val']}
dataloaders = {x: torch.utils.data.DataLoader(image_datasets[x], batch_size=4,
                                             shuffle=True, num_workers=4)
              for x in ['train', 'val']}
dataset_sizes = {x: len(image_datasets[x]) for x in ['train', 'val']}
class_names = image_datasets['train'].classes`
...