Reshaping arrays in Python

最新推荐文章于 2023-06-20 23:51:30 发布

weixin_45405128

最新推荐文章于 2023-06-20 23:51:30 发布

阅读量149

点赞数 1

分类专栏： Python

本文链接：https://blog.csdn.net/weixin_45405128/article/details/102327910

版权

Python 专栏收录该内容

81 篇文章 1 订阅

订阅专栏

在计算机科学中，图像通过3D array(length,height,depth=3)来表示。然而，图像作为算法的输入时，需转化成一维列向量(length∗height∗3,1)。

def image2vector(image):
    v = image.reshape((image.shape[0]*image.shape[1]*image.shape[2]),1)
    return v

image = np.array([[[ 0.67826139,  0.29380381],
        [ 0.90714982,  0.52835647],
        [ 0.4215251 ,  0.45017551]],

       [[ 0.92814219,  0.96677647],
        [ 0.85304703,  0.52351845],
        [ 0.19981397,  0.27417313]],

       [[ 0.60659855,  0.00533165],
        [ 0.10820313,  0.49978937],
        [ 0.34144279,  0.94630077]]])

print ("image2vector(image) = " + str(image2vector(image)))

OUT：

image2vector(image) = [[ 0.67826139]
 [ 0.29380381]
 [ 0.90714982]
 [ 0.52835647]
 [ 0.4215251 ]
 [ 0.45017551]
 [ 0.92814219]
 [ 0.96677647]
 [ 0.85304703]
 [ 0.52351845]
 [ 0.19981397]
 [ 0.27417313]
 [ 0.60659855]
 [ 0.00533165]
 [ 0.10820313]
 [ 0.49978937]
 [ 0.34144279]
 [ 0.94630077]]