Tensores de imagem

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

torch e torchvision 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 ficam guardadas na variável bbox.

espresso

Este exercício faz parte do curso

Aprendizado profundo para imagens com PyTorch

Ver curso

Instruções do exercício

Converta o bbox em tensores usando torch.tensor().
Reestruturar bbox_tensor adicionando uma dimensão de lote usando unsqueeze(0).
Crie uma transformação para redimensionar image para (224) e transforme em um tensor de imagem não redimensionado.
Envie um e-mail para transform com o assunto “ ” para 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