gpt4 book ai didi

python - detectron 2 输出的类标签指的是什么?

转载 作者:行者123 更新时间:2023-12-05 01:32:46 26 4
gpt4 key购买 nike

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

这里是输出字典的规范:https://detectron2.readthedocs.io/tutorials/models.html

最佳答案

数据集元数据类有一个属性thing_classes,它有一个该数据集的类名列表。只需从 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))

关于python - detectron 2 输出的类标签指的是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/65427337/

26 4 0