1. Обучение
    2. /
  2. Курса
    3. /
  3. Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Connected

упражнение

Zbiór danych z augmentacjami

Masz już zbudowany zbiór danych ze zdjęciami chmur oraz model konwolucyjny do klasyfikacji ich rodzajów. Zanim przystąpisz do trenowania, dostosuj zbiór danych, dodając augmentacje, które mogą poprawić skuteczność klasyfikacji.

Kod konfigurujący Dataset i DataLoader jest już przygotowany — powinien wyglądać znajomo. Twoim zadaniem jest zdefiniowanie kompozycji przekształceń stosowanych do obrazów podczas ich wczytywania.

Wcześniej skalowałeś obrazy do rozmiaru 128 × 128, aby wyglądały czytelnie. Teraz użyjesz mniejszych obrazów, żeby przyspieszyć trenowanie. Jak się wkrótce przekonasz, rozmiar 64 × 64 wystarczy, aby model się nauczył.

from torchvision import transforms zostało już wykonane, więc możemy zaczynać!

Инструкции

100 XP
  • Zdefiniuj train_transforms, składając ze sobą pięć przekształceń: losowe odbicie poziome, losowy obrót (o kąt od 0 do 45 stopni), losową automatyczną korektę kontrastu, konwersję do tensora oraz skalowanie do rozmiaru 64 × 64 piksele.