我们可以传递哪个高度的图像=通过我们的有线电视新闻网在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)
任何建议都会很有帮助
目前没有回答
相关问题 更多 >
编程相关推荐