我们可以传递哪个高度的图像=通过我们的CNN进行pytorch培训?

2022-12-01 05:09:58 发布

您现在位置:Python中文网/ 问答频道 /正文

我们可以传递哪个高度的图像=通过我们的有线电视新闻网在Pytorch的宽度

在CNN中,我有卷积、批处理规范、最大池、relu和完全连接的层

我的网络

self.conv_seqn = nn.Sequential(
nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3, padding=1),
nn.BatchNorm2d(32),
nn.ReLU(inplace=True),
nn.MaxPool2d(kernel_size=2, stride=2),
nn.Conv2d(in_channels=32, out_channels=64, kernel_size=3, padding=1),
nn.ReLU(inplace=True),
nn.MaxPool2d(kernel_size=4, stride=4),
nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3, padding=1),
nn.ReLU(inplace=True),
nn.MaxPool2d(kernel_size=4, stride=4),
)
self.fc_seqn = nn.Sequential(
nn.Linear(1843200, 256),
nn.ReLU(inplace=True),
nn.Linear(256, total_configs)
)

我的前进功能

forward()
{
 x = self.conv_seqn(x)
 x = x.view(x.size(0), -1)
 x = self.fc_seqn(x)
 return x
}

如果在应用conv_seqn()后输入大小为3840*1920*3的图像,则其大小应为[1、128、120、60],但我得到的大小为[1128120120](此处批量大小为1)

任何建议都会很有帮助