从具有 multiple 掩码的 multiple 数阵列中选择
17:27:22
我正在尝试构建一个示例来理解 Image 分割,给您一个形状的 Image (1,2,2,3),这是一个2x2 Image ,其中每个像素具有3个数字,表明此像素属于特定类别的概率。我想要的是让输出(1,2,2,1),即基于概率值(选择最高)代码以 showing 问题的每个像素的类。 np.random.seed(2)
pixel=np.random.random((1,2,2,3)) #the image
pixel[0,0,1,:] #pixel(1,1) #with three c