(高光谱)数据集分割
从img_gt(存储一幅图像每个像素点的标签)中拆分出两部分:将矩阵分成两个相同矩阵,当中的非0部分按设定比例分到两个矩阵中
# output:
# (array([ 1, 1, 1, ..., 609, 609, 609], dtype=int64),
# array([ 96, 97, 115, ..., 270, 271, 278], dtype=int64))
indices = np.nonzero(ttt)
# [(1, 96),(1, 97),(1, 115),(1, 171),
原创
2021-06-15 21:14:39 ·
1084 阅读 ·
0 评论