我有灰度图像,我在x_train和x_test中得到了它们的像素阵列。
x_train大小为(2500,21,512),x_test大小为(500,21,512)。我想做一个CNN,以获得输出y_train以及(2500,21,512)和y_test as (500,21,512),但这是我希望网络预测的其他图像的数组。如果不是为了我的图像,我怎么做同样的事情呢?
若要显示27x35形状的图像,请使用:for x in range(1 , 946):matplotlib.pyplot.imshow(random_image_arr.reshape(27 , 35))
然后,我尝试使用torch.nn.Conv2d对图像应用卷积为了将卷积应用到图像中,我应该改变Conv2d的参数吗?
更新。random_image_arr = n
操作:我有维度[n x m x c]的pytorch张量A和维度[1 x 1 x c]的B。我想将每个1 x 1 x c向量的内积与A的B相乘,从而生成[n x m]维的张量C。在我的网络的前向函数中,在特定的步骤我接收维度[N, channels, Height, Width]的张量,其中N是图像的数量,通道是特征映射中的通道的数量,高度和宽度是当前特征映射的。有人能解释一下pytorch中实现这一步的最佳方式和功能吗? 我刚接触pytorch,找不到合适的方法。Tensorflow支