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 merupakan bagian dari kursus
Deep Learning untuk Gambar dengan PyTorch
Latihan interaktif langsung
Ubah teori menjadi aksi dengan salah satu latihan interaktif kami
Mulai latihan