将检测坐标过滤到对象检测模型中的特定类

Question

我正在使用TensorFlow对象检测API来检测一些自定义对象。如果我有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)