detectron 2输出的类别标签是指什么？

Question

在detectron2的文档中，它指出类标签位于output_dict['Instances'].pred_classes中。这一切都很好，我可以很容易地访问它，但是在文档(或输出字典，据我所知)中没有指定哪个整数标签引用哪个类。我假设在某个地方有一本包含类似{0: 'Person', 1: 'Bicycle', 2: 'Car',... }的字典，但我似乎找不到它。有人知道在哪里可以找到这个吗？

以下是输出字典的规范：https://detectron2.readthedocs.io/tutorials/models.html

Answer 1

dataset元数据类有一个属性thing_classes，该属性具有该dataset的类名列表。只需从pred_classes传递类id，就可以获得here所示的类名。

pred_classes = output_dict['instances'].pred_classes.cpu().tolist()
class_names = MetadataCatalog.get("mydataset").thing_classes
pred_class_names = list(map(lambda x: class_names[x], pred_classes))