Conjunto de dados de imagem
Vamos começar criando um conjunto de dados Torch de imagens. Você o usará para explorar os dados e, posteriormente, para alimentá-los em um modelo.
Os dados de treinamento para a tarefa de classificação da nuvem são armazenados na seguinte estrutura de diretórios:
clouds_train
- cirriform clouds
- 539cd1c356e9c14749988a12fdf6c515.jpg
- ...
- clear sky
- cumulonimbus clouds
- cumulus clouds
- high cumuliform clouds
- stratiform clouds
- stratocumulus clouds
Há sete pastas em clouds_train
, cada uma representando um tipo de nuvem (ou um céu claro). Dentro de cada uma dessas pastas, você encontrará os arquivos de imagem correspondentes.
As seguintes importações já foram feitas para você:
from torchvision.datasets import ImageFolder
from torchvision import transforms
Este exercício faz parte do curso
Aprendizagem profunda intermediária com PyTorch
Instruções de exercício
- Componha duas transformações: a primeira, para analisar a imagem em um tensor, e uma para redimensionar a imagem para
128
por128
, atribuindo-as atrain_transforms
. - Use
ImageFolder
para definirdataset_train
, passando a ele o caminho do diretório para os dados ("clouds_train"
) e as transformações definidas anteriormente.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# Compose transformations
train_transforms = ____([
transforms.____,
transforms.____,
])
# Create Dataset using ImageFolder
dataset_train = ____(
____,
transform=____,
)