for index,layer in enumerate(model.children()):
print('\n')
if isinstance(layer, nn.Conv2d):
img = layer(img)
print(img.shape)
elif isinstance(layer, nn.Sequential):
# img = layer(img)
for index2, layer2 in enumerate(layer):
img = layer2(img)
print(img.shape)
elif isinstance(layer, nn.Linear):
print('no')
break
Pytorch得到ResNet各层输出(特征图)
最新推荐文章于 2023-03-22 18:33:19 发布