Prediksi bounding box
Anda telah melatih sebuah model pengenalan objek. Sekarang, Anda ingin menggunakannya untuk menghasilkan bounding box dan klasifikasi untuk sebuah citra uji.
Model tersedia sebagai model dan sudah dalam mode evaluasi. test_image juga tersedia, dan torch serta torch.nn sebagai nn telah diimpor.
Latihan ini adalah bagian dari kursus
Deep Learning untuk Gambar dengan PyTorch
Petunjuk latihan
- Dapatkan keluaran model untuk
test_imagedan simpan keoutput. - Ekstrak bounding box yang diprediksi dari keluaran dan simpan ke
boxes. - Ekstrak skor yang diprediksi dari keluaran dan simpan ke
scores.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)