使用 Tensorflow 对象检测 API 为每个对象的所有类输出分数/概率

Question

在 TensorFlow 对象检测 API 中，我们通常对每个测试图像都这样做：

output_dict = sess.run(tensor_dict, feed_dict={image_tensor: image_np_expanded})
                    # pdb.set_trace()
                    # all outputs are float32 numpy arrays, so convert types as appropriate
                    output_dict['num_detections'] = int(output_dict['num_detections'][0])
                    output_dict['detection_classes'] = output_dict['detection_classes'][0].astype(np.int64)
                    output_dict['detection_boxes'] = output_dict['detection_boxes'][0]
                    output_dict['detection_scores'] = output_dict['detection_scores'][0]

这给了我们每个对象的分数。例如，有一个狗对象，它的分数是 95%。但是我希望它也输出其他类的分数，例如，对于相同的对象，猫的分数是 3%，自行车的分数是 2%。请帮帮我，非常感谢。

Answer 1

事实证明，在较新版本的 API 中，有 output_dict['detection_multiclass_scores'] 可以为每个类提供分数。我使用的是没有它的旧版本。