feature map的可视化
最近使用yolo-tiny进行目标检测,需要用tensorboard将特征图可视化,从网上搜索了一大圈,找到了比较靠谱的方法。`
img_grid = vutils.make_grid(x, normalize=True, scale_each=True, nrow=2)
# 绘制原始图像
writer.add_image('raw img', img_grid, global_step=666) # j 表示feature map数
print(x.size())
model.eval()
for n
原创
2020-07-21 09:46:12 ·
2247 阅读 ·
1 评论