FrancescoSaverioZuppichini

from PIL import Image
import torch
import matplotlib.pyplot as plt
import numpy as np

# credit https://i0.wp.com/craffic.co.in/wp-content/uploads/2021/02/ai-remastered-rick-astley-never-gonna-give-you-up.jpg?w=1600&ssl=1
img = Image.open("./samples/never-gonna-give-you-up.webp")
img

FrancescoSaverioZuppichini

# %matplotlib notebook

FrancescoSaverioZuppichini

from torchvision.ops.boxes import nms as torch_nms
nms_indices = torch_nms(bboxes + labels[..., None], scores, .45)
plot_bboxes(img, 
            bboxes[nms_indices],
            colors=["yellow" if el.item() == 0 else "blue" for el in labels[nms_indices]], 
            labels=["head" if el.item()  == 0 else "mic" for el in labels[nms_indices]]
           )

FrancescoSaverioZuppichini

from ipywidgets import interact

def _interact(threshold: float):
    print(threshold)
    nms_indices = nms(bboxes + labels[..., None], scores, threshold)
    plot_bboxes(img, 
                bboxes[nms_indices],
                colors=["yellow" if el.item() == 0 else "blue" for el in labels[nms_indices]], 
                labels=["head" if el.item()  == 0 else "mic" for el in labels[nms_indices]]
               )

FrancescoSaverioZuppichini

!jupyter nbextension enable --py widgetsnbextension --sys-prefix

FrancescoSaverioZuppichini

nms_indices = nms(bboxes + labels[..., None], scores, .45)
plot_bboxes(img, 
            bboxes[nms_indices],
            colors=["yellow" if el.item() == 0 else "blue" for el in labels[nms_indices]], 
            labels=["head" if el.item()  == 0 else "mic" for el in labels[nms_indices]]
           )

FrancescoSaverioZuppichini

nms_indices = nms(bboxes, scores, .45)
plot_bboxes(img, 
            bboxes[nms_indices],
            colors=["yellow" if el.item() == 0 else "blue" for el in labels[nms_indices]], 
            labels=["head" if el.item()  == 0 else "mic" for el in labels[nms_indices]]
           )

FrancescoSaverioZuppichini

        keep[overlapped + i + 1] = 0

FrancescoSaverioZuppichini

        overlapped = torch.nonzero(iou > iou_threshold)

FrancescoSaverioZuppichini

            (bboxes ...)[order[i + 1:]]