1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

Výpočet NMS

Z modelu pro rozpoznávání objektů jsi už extrahoval/a predikované ohraničující rámečky a skóre. Teď je čas zajistit, aby zůstaly pouze ty nejpřesnější a nepřekrývající se rámečky – k tomu využiješ techniku non-max suppression.

boxes a scores vytvořené v předchozím cvičení jsou dostupné v tvém pracovním prostředí a knihovny torch a torchvision jsou naimportovány.

Pokyny

100 XP
  • Importuj nms z torchvision.ops.
  • Nastav práh IoU na hodnotu 0.5.
  • Aplikuj non-max suppression – předej funkci boxes, confidence_scores a iou_threshold.
  • Pomocí výstupních indexů vyfiltruj predikované rámečky.