Обчисліть NMS
Після того як ви отримали передбачені обмежувальні прямокутники (bounding boxes) і бали з моделі розпізнавання об'єктів, далі потрібно залишити лише найточніші й такі, що не перекриваються. Для цього застосуйте метод придушення немаксимумів (non‑max suppression).
boxes і scores, які ви створили у попередній вправі, доступні у вашому робочому середовищі, а torch і torchvision вже імпортовано.
Ця вправа є частиною курсу
Глибоке навчання для зображень із PyTorch
Інструкції до вправи
- Імпортуйте
nmsізtorchvision.ops. - Встановіть поріг IoU рівним
0.5. - Застосуйте придушення немаксимумів, передавши
boxes,confidence_scoresіiou_thresholdу відповідну функцію. - Використайте індекси з виходу, щоб відфільтрувати передбачені прямокутники.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Import nms
____
# Set the IoU threshold
iou_threshold = ____
# Apply non-max suppression
box_indices = ____
# Filter boxes
filtered_boxes = ____
print("Filtered Boxes:", filtered_boxes)