Аналіз виходу моделі
Отримавши прогноз від попередньо натренованої моделі, тепер ви можете детально його проаналізувати. Змінна prediction доступна для вас.
Ви побачите, що це список довжини один (ви зробили прогноз лише для одного зображення) зі словником із чотирма ключами. labels — це цілі числа, що позначають передбачені класи, а scores — відповідні ймовірності класів. Список назв класів, що відповідають labels, доступний як class_names.
Які категорії передбачені та які їхні оцінки для двох найкращих масок?
Ця вправа є частиною курсу
Глибоке навчання для зображень із PyTorch
Практична інтерактивна вправа
Перетворіть теорію на практику за допомогою однієї з наших інтерактивних вправ
Почати вправу