在 detectron2 中，如何为同一类的边界框分配相同的颜色？

Question

我正在学习使用 Faster R-CNN 的 detectron2。

我正在尝试绘制同一类的边界框。例如，如果边界框包含“猫”，则它是蓝色的，而如果另一个边界框包含“狗”，则它被指定为红色。 然而，下面的代码只能输出每个边界框有不同的颜色 请你给我建议好吗？

from detectron2.utils.visualizer import ColorMode
import glob
for imageName in random.sample(glob.glob(os.path.join(test_path, '*.png')), 3):
  im = cv2.imread(imageName)
  outputs = predictor(im)
  v = Visualizer(im[:, :, ::-1],
        metadata=train_metadata,
        scale=0.8
         )
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
  cv2_imshow(out.get_image()[:, :, ::-1])