Conjunto de dados de duas entradas
A criação de um modelo com várias entradas começa com a elaboração de um conjunto de dados personalizado que possa fornecer todas as entradas para o modelo. Neste exercício, você criará o conjunto de dados Omniglot que serve triplas que consistem em:
- A imagem de um personagem a ser classificado,
- O vetor do alfabeto codificado com um único disparo de comprimento 30, com zeros em todos os lugares, exceto por um único que denota o ID do alfabeto de onde vem o caractere,
- O rótulo de destino, um número inteiro entre 0 e 963.
Você recebe samples
, uma lista de três tuplas que inclui o caminho do arquivo de uma imagem, seu vetor de alfabeto e o rótulo de destino. Além disso, as importações a seguir já foram feitas para você, então vamos lá!
from PIL import Image
from torch.utils.data import DataLoader, Dataset
from torchvision import transforms
Este exercício faz parte do curso
Aprendizagem profunda intermediária com PyTorch
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
class OmniglotDataset(Dataset):
def __init__(self, transform, samples):
# Assign transform and samples to class attributes
____ = ____
____ = ____