python 访问返回的变量

Question

我对 python 仍然不太满意。尝试深度学习开源代码，我正在编写一个测试代码，我可以看到它的运行情况如下。 (使用数据集包和其中的 pascal_voc 模块，顺便说一句，这是来自 py-faster-rcnn 代码)

>>> import datasets
>>> import datasets.pascal_voc as pv

>>> d = datasets.pascal_voc('trainval', '2007')
>>> d._load_pascal_annotation('{0:06d}'.format(5))
Removed 2 difficult objects
{'boxes': array([[262, 210, 323, 338],
       [164, 263, 252, 371],
       [240, 193, 294, 298]], dtype=uint16), 'flipped': False, 'gt_classes': array([9, 9, 9], dtype=int32), 'gt_overlaps': <3x21 sparse matrix of type '<type 'numpy.float32'>'

函数_load_pascal_annotation返回如下所示的值。

def _load_pascal_annotation(self, index):
....
    return {'boxes' : boxes,
        'gt_classes': gt_classes,
        'gt_overlaps' : overlaps,
        'flipped' : False}

我想从返回的字典中提取“盒子”数组并用它来绘制一些东西。我尝试了 d['boxes'] 但在下面给出了错误。

>>> d['boxes']
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'pascal_voc' object has no attribute '__getitem__'

如何从_load_pascal_annotation的返回值中提取坐标值？

Answer 1

您调用_load_pascal_annotation方法，但不保留对其返回值的引用。

尝试:

r_value = d._load_pascal_annotation('{0:06d}'.format(5))
r_value['boxes']