1. Learn
  2. /
  3. Cursuri
  4. /
  5. Deep Learning intermediar cu PyTorch

Connected

exercițiu

Set de date cu două intrări

Construirea unui model cu mai multe intrări începe cu crearea unui set de date personalizat care să furnizeze toate intrările necesare modelului. În acest exercițiu, vei construi setul de date Omniglot, care generează triplete formate din:

  • Imaginea unui caracter de clasificat,
  • Vectorul alfabetului codificat one-hot, de lungime 30, cu zerouri peste tot, cu excepția unui singur unu care indică ID-ul alfabetului din care provine caracterul,
  • Eticheta țintă, un număr întreg între 0 și 963.

Îți este furnizată variabila samples, o listă de 3-tupluri care conțin calea fișierului unei imagini, vectorul său de alfabet și eticheta țintă. De asemenea, următoarele importuri au fost deja efectuate, așa că hai să trecem la treabă!

from PIL import Image
from torch.utils.data import DataLoader, Dataset
from torchvision import transforms

Instrucțiuni 1/4

undefined XP
    1
    2
    3
    4
  • Atribuie transform și samples ca atribute de clasă cu aceleași nume.