Francesco Saverio Zuppichini FrancescoSaverioZuppichini

🧸

Focusing

Computer Vision Engineer @ 🦝 BSc informatics. MSc AI. Artificial Intelligence /Deep Learning Enthusiast & Full Stack developer

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-21.py

Created October 26, 2022 07:57

Test

overlapped = torch.nonzero(iou > iou_threshold)

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-20.py

Created October 26, 2022 07:57

Test

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

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-19.py

Last active October 27, 2022 08:34

Test

(bboxes[...] * keep[...][...,None])

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-18.txt

Last active October 27, 2022 08:31

Test

iou = box_iou(bbox[None,...], (bboxes[order[i + 1:]]) * keep[i + 1:][...,None])

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-17.py

Last active October 27, 2022 08:37

Test

bbox = bboxes[order[i]]]

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-16.py

Created October 26, 2022 07:57

Test

if keep[i]:

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-15.py

Created October 26, 2022 07:57

Test

	for i in indices:
	...

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-14.py

Created October 26, 2022 07:57

Test

keep = torch.ones_like(indices, dtype=torch.bool)

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-13.py

Created October 26, 2022 07:57

Test

indices = torch.arange(bboxes.shape[0])

FrancescoSaverioZuppichini / non-max-suppression-in-pytorch-12.py

Created October 26, 2022 07:57

Test

order = scores.argsort()