1. 学ぶ
    2. /
  2. コース
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

演習

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

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

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

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

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

指示1 / 3

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