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

Вправа

Обчисліть NMS

Після того як ви отримали передбачені обмежувальні прямокутники (bounding boxes) і бали з моделі розпізнавання об'єктів, далі потрібно залишити лише найточніші й такі, що не перекриваються. Для цього застосуйте метод придушення немаксимумів (non‑max suppression).

boxes і scores, які ви створили у попередній вправі, доступні у вашому робочому середовищі, а torch і torchvision вже імпортовано.

Інструкції

100 XP
  • Імпортуйте nms із torchvision.ops.
  • Встановіть поріг IoU рівним 0.5.
  • Застосуйте придушення немаксимумів, передавши boxes, confidence_scores і iou_threshold у відповідну функцію.
  • Використайте індекси з виходу, щоб відфільтрувати передбачені прямокутники.