Menganalisis keluaran model
Setelah memperoleh prediksi dari model pralatih, sekarang Anda dapat menganalisisnya secara detail. prediction tersedia untuk Anda.
Anda akan melihat bahwa ini adalah sebuah list dengan panjang satu (Anda membuat prediksi hanya untuk satu gambar) yang berisi sebuah dictionary dengan empat kunci. labels adalah bilangan bulat yang menunjukkan kelas yang diprediksi, dan scores adalah probabilitas kelas yang bersesuaian. Anda dapat mengakses daftar nama kelas yang bersesuaian dengan labels sebagai class_names.
Apa kategori yang diprediksi dan skor terkaitnya untuk dua mask teratas?
Latihan ini adalah bagian dari kursus
Deep Learning untuk Gambar dengan PyTorch
Latihan interaktif praktis
Ubah teori menjadi tindakan dengan salah satu latihan interaktif kami.
Mulai berolahraga