def infer()

in cvat-serverless/functions/endpoints/openvino_reidentification.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

def infer(image0, boxes0, image1, boxes1, threshold, distance):
    similarity_matrix = _compute_similarity_matrix(image0, boxes0, image1, boxes1, distance)
    row_idx, col_idx = linear_sum_assignment(similarity_matrix)
    results = [-1] * len(boxes0)
    for idx0, idx1 in zip(row_idx, col_idx):
        if similarity_matrix[idx0, idx1] <= threshold:
            results[idx0] = int(idx1)

    return results