Tensores de imagem

Uma empresa de café tem um projeto de detecção de objetos em que precisa anotar objetos de interesse — neste caso, doses de espresso. Você criou uma lista com as coordenadas da caixa delimitadora para uma imagem de espresso. Agora, você precisa converter a imagem e as coordenadas em tensores.

torch e torchvision já foram importados. torchvision.transforms foi importado como transforms. A imagem foi carregada como image usando Image.open() da biblioteca PIL. As coordenadas da caixa delimitadora estão armazenadas na variável bbox.

espresso

Este exercício faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

Instruções do exercício

Converta bbox em tensores usando torch.tensor().
Redimensione bbox_tensor adicionando uma dimensão de batch com unsqueeze(0).
Crie uma transformação para redimensionar image para (224) e convertê-la em um tensor de imagem sem escala.
Aplique transform a image.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

# Convert bbox into tensors
bbox_tensor = ____

# Add a new batch dimension
bbox_tensor = bbox_tensor.____

# Resize image and transform tensor
transform = transforms.Compose([
  transforms.____,
  transforms.____
])

# Apply transform to image
image_tensor = ____
print(image_tensor)

Editar e executar o código