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
Acest exercițiu face parte din cursul
Deep Learning intermediar cu PyTorch
Instrucțiuni pentru exercițiu
- Compune două transformări: prima, pentru a converti imaginea într-un tensor, și a doua, pentru a redimensiona imaginea la
128x128, și atribuie-le variabileitrain_transforms. - Folosește
ImageFolderpentru a definidataset_train, transmițându-i calea către directorul cu date ("clouds_train") și transformările definite anterior.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)