Previsão de bounding boxes

Você treinou um modelo de reconhecimento de objetos. Agora, você quer usá-lo para gerar bounding boxes e classificações para uma imagem de teste.

O modelo está disponível como model e já está no modo de avaliação. A test_image também está disponível, e torch e torch.nn como nn já foram importados.

Este exercício faz parte do curso

Deep Learning para Imagens com PyTorch

Ver curso

Instruções do exercício

Obtenha a saída do modelo para test_image e atribua-a a output.
Extraia as bounding boxes previstas da saída e atribua-as a boxes.
Extraia as pontuações previstas da saída e atribua-as a scores.

Exercício interativo prático

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

# Get model's prediction
with torch.no_grad():
    output = ____

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

Editar e executar o código