使用scipy阅读BSD500 groundTruth

Question

我试图使用scipyloadmat加载一个基本事实文件，它返回object（dtype='O'）类型的numpy ndarray。你知道吗

从这个对象我到达访问每一个元素，这也是nArray，但我正在努力从这一点，要么访问分割或边界图像。你知道吗

我想把这个转换成一个数字类型的数组的列表我怎么做？你知道吗

提前谢谢你的帮助

Answer 1

我找到了解决问题的方法。 我不认为这是最佳的，但它的工作。你知道吗

def load_bsd_gt(filename):
    gt = loadmat(filename)

    gt = gt['groundTruth']

    cols = gt.shape[1]

    what = ['Segmentation','Boundaries']

    ret = list()

    for i in range(cols):
        j=0
        tmp = list()
        for w in what:
            tmp.append(gt[0][j][w][0][0][:])
            j+=1
        ret.append(tmp)

    return ret

如果有人有更好的方法，请随时添加评论或答案。你知道吗