ComeçarComece gratuitamente

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

Ver Curso

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 por 128, atribuindo-as a train_transforms.
  • Use ImageFolder para definir dataset_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=____,
)
Editar e executar código