1. Apprendre
  2. /
  3. Cours
  4. /
  5. Середній рівень Deep Learning з PyTorch

Connected

Exercice

Побудова згорткових мереж

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

Ви вирішуєте створити згортковий класифікатор зображень. Модель складатиметься з двох частин:

  • Екстрактора ознак, який навчається вектору ознак з вхідного зображення,
  • Класифікатора, який передбачає клас зображення на основі здобутих ознак.

torch і torch.nn as nn уже імпортовано, тож почнімо!

Instructions 1/3

undefined XP
    1
    2
    3
  • Визначте частину моделі feature_extractor, додавши ще один згортковий шар із 64 вихідними картами ознак, активацією ELU та шаром максимального пулінгу з вікном розміру два; наприкінці «розплющте» вихід.