python 改变数组元素值

最新推荐文章于 2024-08-24 02:23:59 发布

XBeiKer

最新推荐文章于 2024-08-24 02:23:59 发布

阅读量5.9k

点赞数 2

文章标签： python list

本文链接：https://blog.csdn.net/sinat_38778796/article/details/123912684

版权

这篇博客介绍了在计算像素级别的交并比（MIOU）时，如何将二值像素标签中的255值转换为1，以便在混淆矩阵中进行正确的分类计数。博主展示了使用NumPy和Python实现的方法一和方法二，通过遍历和直接赋值来完成转换，并展示了转换前后的数组对比。

摘要由CSDN通过智能技术生成

import numpy as np
from keras import backend
from PIL import Image

a = np.array([0,255,255,0,255,0,0,255,0,0])
print(a)
print(a.shape)
b = np.array([0,255,0,0,255,0,0,255,0,0])
c = a.flatten()
d = a.flatten()
print(c.shape)
print(c)

#方法一：
b[b==255]=1  
# print(b)

#方法二：
for i, ele in enumerate(a):
    if ele  == 255:
        a[i] = 1

print(c)
print(a)
print(b)

简单记录一下，在计算MIOU的时候，像素为二值像素，像素值为255的标签为第一类，在混淆矩阵的时候，需要将255像素改成1像素，和预测标签进行计算对角线上的分类正确的像素点