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

exercițiu

Set de date cu imagini

Să începem prin a construi un Dataset Torch de imagini. Îl vei folosi pentru a explora datele și, mai târziu, pentru a-l transmite unui model.

Datele de antrenament pentru sarcina de clasificare a norilor sunt stocate în următoarea structură de directoare:

clouds_train
  - cirriform clouds
    - 539cd1c356e9c14749988a12fdf6c515.jpg
    - ...
  - clear sky
  - cumulonimbus clouds
  - cumulus clouds
  - high cumuliform clouds
  - stratiform clouds
  - stratocumulus clouds

În interiorul clouds_train există șapte foldere, fiecare reprezentând un tip de nor (sau cer senin). În fiecare dintre aceste foldere se află fișierele de imagine corespunzătoare.

Următoarele importuri au fost deja realizate pentru tine:

from torchvision.datasets import ImageFolder
from torchvision import transforms

Instrucțiuni

100 XP
  • Compune două transformări: prima, pentru a converti imaginea într-un tensor, și a doua, pentru a redimensiona imaginea la 128 x 128, și atribuie-le variabilei train_transforms.
  • Folosește ImageFolder pentru a defini dataset_train, transmițându-i calea către directorul cu date ("clouds_train") și transformările definite anterior.