85 lines
1.8 KiB
Python
85 lines
1.8 KiB
Python
"""The tests for the Deepstack object component."""
|
|
from .image_processing import get_objects
|
|
|
|
TARGET = "person"
|
|
IMG_WIDTH = 960
|
|
IMG_HEIGHT = 640
|
|
|
|
MOCK_PREDICTIONS = [
|
|
{
|
|
"confidence": 0.9995428,
|
|
"label": "person",
|
|
"y_min": 95,
|
|
"x_min": 295,
|
|
"y_max": 523,
|
|
"x_max": 451,
|
|
},
|
|
{
|
|
"confidence": 0.9994912,
|
|
"label": "person",
|
|
"y_min": 99,
|
|
"x_min": 440,
|
|
"y_max": 531,
|
|
"x_max": 608,
|
|
},
|
|
{
|
|
"confidence": 0.9990447,
|
|
"label": "dog",
|
|
"y_min": 358,
|
|
"x_min": 647,
|
|
"y_max": 539,
|
|
"x_max": 797,
|
|
},
|
|
]
|
|
|
|
PARSED_PREDICTIONS = [
|
|
{
|
|
"bounding_box": {
|
|
"height": 0.669,
|
|
"width": 0.163,
|
|
"y_min": 0.148,
|
|
"x_min": 0.307,
|
|
"y_max": 0.817,
|
|
"x_max": 0.47,
|
|
},
|
|
"box_area": 0.109,
|
|
"centroid": {"x": 0.389, "y": 0.483},
|
|
"name": "person",
|
|
"confidence": 99.954,
|
|
},
|
|
{
|
|
"bounding_box": {
|
|
"height": 0.675,
|
|
"width": 0.175,
|
|
"y_min": 0.155,
|
|
"x_min": 0.458,
|
|
"y_max": 0.83,
|
|
"x_max": 0.633,
|
|
},
|
|
"box_area": 0.118,
|
|
"centroid": {"x": 0.545, "y": 0.493},
|
|
"name": "person",
|
|
"confidence": 99.949,
|
|
},
|
|
{
|
|
"bounding_box": {
|
|
"height": 0.283,
|
|
"width": 0.156,
|
|
"y_min": 0.559,
|
|
"x_min": 0.674,
|
|
"y_max": 0.842,
|
|
"x_max": 0.83,
|
|
},
|
|
"box_area": 0.044,
|
|
"centroid": {"x": 0.752, "y": 0.701},
|
|
"name": "dog",
|
|
"confidence": 99.904,
|
|
},
|
|
]
|
|
|
|
|
|
def test_get_objects():
|
|
objects = get_objects(MOCK_PREDICTIONS, IMG_WIDTH, IMG_HEIGHT)
|
|
assert len(objects) == 3
|
|
assert objects[0] == PARSED_PREDICTIONS[0]
|