def boxes_MiniXceptionFER():
boxes2D = [
Box2D(np.array([855, 466, 974, 585]), 0.98590159, 'happy'),
Box2D(np.array([701, 362, 827, 488]), 0.49472683, 'neutral'),
Box2D(np.array([488, 408, 612, 532]), 0.28161105, 'sad'),
]
return boxes2D
def boxes_FaceKeypointNet2D32():
boxes2D = [
Box2D(np.array([855, 466, 974, 585]), 1.0, 'Face'),
Box2D(np.array([701, 362, 827, 488]), 1.0, 'Face'),
Box2D(np.array([488, 408, 612, 532]), 1.0, 'Face'),
]
return boxes2D
def boxes_HaarCascadeFace():
boxes2D = [
Box2D(np.array([855, 466, 974, 585]), 1.0, 'Face'),
Box2D(np.array([701, 362, 827, 488]), 1.0, 'Face'),
Box2D(np.array([488, 408, 612, 532]), 1.0, 'Face'),
]
return boxes2D
def boxes_SSD512YCBVideo():
boxes2D = [
Box2D(np.array([115, 121, 215, 155]), 0.9056605, '011_banana'),
Box2D(np.array([38, 93, 162, 164]), 0.839406490, '035_power_drill'),
Box2D(np.array([173, 25, 217, 123]), 0.99889081, '006_mustard_bottle'),
Box2D(np.array([93, 11, 148, 104]), 0.825154304, '003_cracker_box')
]
return boxes2D
def boxes_SSD300FAT():
boxes2D = [
Box2D(np.array([110, 96, 150, 125]), 0.626648, '007_tuna_fish_can'),
Box2D(np.array([41, 93, 146, 167]), 0.7510558, '035_power_drill'),
Box2D(np.array([171, 22, 227, 131]), 0.793466, '006_mustard_bottle'),
Box2D(np.array([99, 6, 151, 107]), 0.50704032, '003_cracker_box')
]
return boxes2D
def boxes_SSD512COCO():
boxes2D = [
Box2D(np.array([544, 373, 1018, 807]), 0.9982471, 'person'),
Box2D(np.array([483, 710, 569, 819]), 0.78569597, 'cup'),
Box2D(np.array([943, 182, 1083, 341]), 0.7874794, 'potted plant'),
Box2D(np.array([150, 721, 1413, 993]), 0.6786173, 'dining table'),
Box2D(np.array([577, 619, 895, 820]), 0.83648031, 'laptop')
]
return boxes2D
def boxes_SSD300VOC():
boxes2D = [
Box2D(np.array([510, 383, 991, 806]), 0.99694544, 'person'),
Box2D(np.array([954, 192, 1072, 350]), 0.7211749, 'pottedplant')
]
return boxes2D
def test_Box2D(box_coordinates, score, test_point):
""" Unit test for Bounding box 2D with class
and score
"""
box2D = Box2D(box_coordinates, score)
assert (box2D.contains(test_point))