"Pytorch：创建节点特征矩阵"

Question

我正在尝试创建形状[num_nodes, num_features]的节点特征矩阵，其中num_nodes由图像中的多个边界框（cropped_image）表示。 num_features由每个边界框的像素值表示。

我下面的代码返回[num_nodes, num_features] (=> [87, 1080,1920,3])

     cropped_images =[]

     imgs = PIL.Image.open(img_path).convert('RGB')
     image_width, image_height = imgs.size
     imgArrays = np.array(imgs)
      
     X = (xCenter*image_width)          
     Y = (yCenter*image_height)         
     W = (Width*image_width)        
     H = (Height*image_height)         
     cropped_image = np.zeros((image_height, image_width))
                    
     for i in range(len(X)):
             x1, y1, w, h = X[i], Y[i], W[i], H[i]
             x_start = int(x1 - (w/2))
             y_start = int(y1 - (h/2))
             x_end = int(x_start + w)
             y_end = int(y_start + h)
                          
             temp = imgArrays[y_start: y_end, x_start: x_end]
             dim = (image_width, image_height)
         
             # resize image
             resized = cv2.resize(temp, dim, interpolation = cv2.INTER_AREA)
             cropped_image_pixels = resized 
             cropped_images.append(cropped_image_pixels)

             Tensor_cropped_image = torch.as_tensor(np.array(cropped_images))

如何返回[num_nodes, num_features] (=> [87, 1080*1920*3]) ？

您的帮助将不胜感激。