Previsão de caixas delimitadoras
Você treinou um modelo de reconhecimento de objetos. Agora, você quer usar para gerar caixas delimitadoras e classificações para uma imagem de teste.
O modelo tá disponível como um model e já tá no modo de avaliação. O test_image também está disponível, e torch e torch.nn, assim como nn, foram importados.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Pega o resultado do modelo para o
test_imagee coloca emoutput. - Pega as caixas delimitadoras previstas da saída e coloca elas num
boxes. - Pega as pontuações previstas da saída e coloca em um
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)