论文学习 网络可视化

Posted by Sun on August 2, 2020

ResNet50

基于pytorch中的预训练ResNet50模型,在对lena图片推理阶段显示特征图的各个通道。

input image [1,3,224,224]

lena_RGB

  1. 因为pytorch读取图像过程必须按patch来读取,所以需要用unsqueeze(0)[batch_size, n_channels, hight, width]格式

f1_conv1

f2_bn1

f3_relu

f4_maxpool

f5_layer1

f6_layer2

f7_layer3

f8_layer4_1

f8_layer4_2

f9_avgpool

f10_fc