1. Learn
    2. /
  2. Курси
    3. /
  3. Глибоке навчання для зображень із PyTorch
Connected

Вправа

Модель Faster R-CNN

Ваше наступне завдання — зібрати модель Faster R-CNN, яка зможе знаходити об'єкти різних розмірів на зображенні. Для цього скористайтеся зручною класою MultiScaleRoIAlign() з torchvision.ops.

Клас FasterRCNN імпортовано з torchvision.models.detection. Ваш anchor_generator з попередньої вправи доступний у робочому середовищі, а також імпортовано torch, torch.nn як nn і torchvision.

Інструкції

100 XP
  • Імпортуйте MultiScaleRoIAlign з torchvision.ops.
  • Створіть RoI-пулер за допомогою MultiScaleRoIAlign, встановивши featmap_names у ["0"], output_size — 7, а sampling_ratio — 2.
  • Створіть модель Faster R-CNN, передавши їй backbone, num_class для бінарної класифікації, anchor_generator і roi_pooler.