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.
Toto cvičení je součástí kurzu
Deep Learning pro obrázky s PyTorchem
Pokyny k cvičení
- Importuj
nmsztorchvision.ops. - Nastav práh IoU na hodnotu
0.5. - Aplikuj non-max suppression – předej funkci
boxes,confidence_scoresaiou_threshold. - Pomocí výstupních indexů vyfiltruj predikované rámečky.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# Import nms
____
# Set the IoU threshold
iou_threshold = ____
# Apply non-max suppression
box_indices = ____
# Filter boxes
filtered_boxes = ____
print("Filtered Boxes:", filtered_boxes)