machine-learning-deep-learning/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-02-21 11:18:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
Aladdin Persson
2021-01-30 21:49:15 +01:00
commit 65b8c80495
432 changed files with 1290844 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
ML_tests/Object_detection_tests/nms_test.py Executable file
View File

@@ -0,0 +1,95 @@
import sys
import unittest
import torch
sys.path.append("ML/Pytorch/object_detection/metrics/")
from nms import nms
class TestNonMaxSuppression(unittest.TestCase):
def setUp(self):
# test cases we want to run
self.t1_boxes = [
[1, 1, 0.5, 0.45, 0.4, 0.5],
[1, 0.8, 0.5, 0.5, 0.2, 0.4],
[1, 0.7, 0.25, 0.35, 0.3, 0.1],
[1, 0.05, 0.1, 0.1, 0.1, 0.1],
]
self.c1_boxes = [[1, 1, 0.5, 0.45, 0.4, 0.5], [1, 0.7, 0.25, 0.35, 0.3, 0.1]]
self.t2_boxes = [
[1, 1, 0.5, 0.45, 0.4, 0.5],
[2, 0.9, 0.5, 0.5, 0.2, 0.4],
[1, 0.8, 0.25, 0.35, 0.3, 0.1],
[1, 0.05, 0.1, 0.1, 0.1, 0.1],
]
self.c2_boxes = [
[1, 1, 0.5, 0.45, 0.4, 0.5],
[2, 0.9, 0.5, 0.5, 0.2, 0.4],
[1, 0.8, 0.25, 0.35, 0.3, 0.1],
]
self.t3_boxes = [
[1, 0.9, 0.5, 0.45, 0.4, 0.5],
[1, 1, 0.5, 0.5, 0.2, 0.4],
[2, 0.8, 0.25, 0.35, 0.3, 0.1],
[1, 0.05, 0.1, 0.1, 0.1, 0.1],
]
self.c3_boxes = [[1, 1, 0.5, 0.5, 0.2, 0.4], [2, 0.8, 0.25, 0.35, 0.3, 0.1]]
self.t4_boxes = [
[1, 0.9, 0.5, 0.45, 0.4, 0.5],
[1, 1, 0.5, 0.5, 0.2, 0.4],
[1, 0.8, 0.25, 0.35, 0.3, 0.1],
[1, 0.05, 0.1, 0.1, 0.1, 0.1],
]
self.c4_boxes = [
[1, 0.9, 0.5, 0.45, 0.4, 0.5],
[1, 1, 0.5, 0.5, 0.2, 0.4],
[1, 0.8, 0.25, 0.35, 0.3, 0.1],
]
def test_remove_on_iou(self):
bboxes = nms(
self.t1_boxes,
threshold=0.2,
iou_threshold=7 / 20,
box_format="midpoint",
)
self.assertTrue(sorted(bboxes) == sorted(self.c1_boxes))
def test_keep_on_class(self):
bboxes = nms(
self.t2_boxes,
threshold=0.2,
iou_threshold=7 / 20,
box_format="midpoint",
)
self.assertTrue(sorted(bboxes) == sorted(self.c2_boxes))
def test_remove_on_iou_and_class(self):
bboxes = nms(
self.t3_boxes,
threshold=0.2,
iou_threshold=7 / 20,
box_format="midpoint",
)
self.assertTrue(sorted(bboxes) == sorted(self.c3_boxes))
def test_keep_on_iou(self):
bboxes = nms(
self.t4_boxes,
threshold=0.2,
iou_threshold=9 / 20,
box_format="midpoint",
)
self.assertTrue(sorted(bboxes) == sorted(self.c4_boxes))
if __name__ == "__main__":
print("Running Non Max Suppression Tests:")
unittest.main()