Francesco Saverio Zuppichini FrancescoSaverioZuppichini

## non-max-suppression-in-pytorch-25.py
!jupyter nbextension enable --py widgetsnbextension --sys-prefix

## non-max-suppression-in-pytorch-24.py
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]]
           )

## non-max-suppression-in-pytorch-23.py
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]]
           )

## non-max-suppression-in-pytorch-22.py
        keep[overlapped + i + 1] = 0

## non-max-suppression-in-pytorch-21.py
        overlapped = torch.nonzero(iou > iou_threshold)

## non-max-suppression-in-pytorch-20.py
            (bboxes ...)[order[i + 1:]]

## non-max-suppression-in-pytorch-16.py
     if keep[i]:

## non-max-suppression-in-pytorch-15.py
for i in indices:
    ...

## non-max-suppression-in-pytorch-14.py
keep = torch.ones_like(indices, dtype=torch.bool)

## non-max-suppression-in-pytorch-13.py
indices = torch.arange(bboxes.shape[0])
	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]]
	)
	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]]
	)