ResNet50
基于pytorch中的预训练ResNet50模型,在对lena图片推理阶段显示特征图的各个通道。
input image [1,3,224,224]
- 因为pytorch读取图像过程必须按patch来读取,所以需要用
unsqueeze(0)
为[batch_size, n_channels, hight, width]
格式
基于pytorch中的预训练ResNet50模型,在对lena图片推理阶段显示特征图的各个通道。
unsqueeze(0)
为[batch_size, n_channels, hight, width]
格式