1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

Przewidywanie ramek ograniczających

Wytrenowano model do rozpoznawania obiektów. Chcesz teraz użyć go do wygenerowania ramek ograniczających i klasyfikacji dla obrazu testowego.

Model jest dostępny jako model i działa już w trybie ewaluacji. Dostępny jest również test_image, a torch i torch.nn jako nn zostały zaimportowane.

Instrukcje

100 XP
  • Uzyskaj wyniki modelu dla test_image i przypisz je do zmiennej output.
  • Wyodrębnij przewidywane ramki ograniczające z wyjścia i przypisz je do zmiennej boxes.
  • Wyodrębnij przewidywane wyniki pewności z wyjścia i przypisz je do zmiennej scores.