表示Keras CNN模型的3维数据的正确方法?

时间:2018-09-18 08:46:54

标签: python pandas numpy scipy

我是numpy的新手。我有1000个3D结构的数据集,每个结构都有x,y,z坐标。我正在尝试使用CNN模型,但无法为该架构准备正确的数据集格式。这是我的熊猫数据框数据集:

row_id, xlist, ylist, zlist, target
1, [1,2,3,3], [5,2,6,7], [1,0,4,5], 1
2, [-1,6,4,6], [7,-1,4,1], [2,5,3,5], 1
3, [6,4,-2,-2], [2,-6,6,2], [3,0,7,2], 0
...
n, [5,3,1], [5,3,7], [3,5,3], 1

我尝试过,但这不是我想要的

np.array([dataset['xlist'],dataset['ylist'],dataset['zlist']]) 

我想获得具有x * y * z尺寸的3D结构。如何创建尺寸为4 * 4 * 4(用于x,y和z)的3D矩阵

1 个答案:

答案 0 :(得分:1)

喀拉拉邦可能需要具有(高度,宽度,深度,通道)的表示形式。

因此,在您的情况下,可以先创建一个空白图像,如下所示:

empty_image = np.zeros(20, 20, 20, 1)

这假设您的x,y,z范围是-10到10,像素大小= 1 因此,对于每个点,您都可以将零变成一个:

empty_image[x+10, y+10, z+10, 0] += 1