Отфильтруйте координаты обнаружения по определенному классу в модели обнаружения объектов

Question

Я использую API обнаружения объектов TensorFlow для обнаружения некоторых пользовательских объектов. Если у меня есть 4 объекта (например, "кошка", "собака", "лев", "человек". Я хочу получить координату "человека" каждый раз, когда она обнаружена. Я могу получить координаты для каждого обнаруженного объекта, НО я не знаю, как отфильтровать его только к одному конкретному объекту (например, «человек»). Вот часть, которую я использую для получения координат.

for coordinate in coordinates:
    (ymin, ymax, xmin, xmax, acc, classification) = coordinate
    height = np.size(frame, 0)
    width = np.size(frame, 1)
    min_score = 0.6
    true_boxs = boxes[0][scores[0] > min_score]
    for i in range(true_boxs.shape[0]):
        ymin = int(true_boxs[i, 0] * height)
        xmin = int(true_boxs[i, 1] * width)
        ymax = int(true_boxs[i, 2] * height)
        xmax = int(true_boxs[i, 3] * width)
        roi = frame[ymin:ymax, xmin:xmax]
        cv2.imwrite("box_{}.jpg".format(str(i)), roi)