Передбачення обмежувальних прямокутників

Ви натренували модель розпізнавання об'єктів. Тепер ви хочете використати її, щоб згенерувати обмежувальні прямокутники та класифікації для тестового зображення.

Модель доступна як model і вже переведена в режим оцінювання. Також доступне test_image, а torch і torch.nn як nn імпортовано.

Ця вправа є частиною курсу

Глибоке навчання для зображень із PyTorch

Переглянути курс

Інструкції до вправи

Отримайте вихід моделі для test_image та присвойте його змінній output.
Витягніть передбачені обмежувальні прямокутники з виходу та присвойте їх змінній boxes.
Витягніть передбачені оцінки впевненості з виходу та присвойте їх змінній scores.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

# Get model's prediction
with torch.no_grad():
    output = ____

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

Редагувати та запускати код