- 一个是用tensorflow,但是还没搞清楚怎么实现
- 下面是用pytorch生成的文字版的网络结构
net = nn.Sequential(
.......
)
X = torch.random(1, 1, 1080, 1920)
for layer in net:
X = layer(X)
print(layer.__class__.name__, 'output shape:/t', X.shape)
net = nn.Sequential(
.......
)
X = torch.random(1, 1, 1080, 1920)
for layer in net:
X = layer(X)
print(layer.__class__.name__, 'output shape:/t', X.shape)