1.可视化调试

可视化方法可分为：各通到相加可视化、

1.1 各通道相加可视化

def visualize_feature_map(img_batch,out_path,type,BI):
    feature_map = torch.squeeze(img_batch)
    feature_map = feature_map.detach().cpu().numpy()

    feature_map_sum = feature_map[0, :, :]
    feature_map_sum = np.expand_dims(feature_map_sum, axis=2)
    for i in range(0, 2048): #各通道遍历
        feature_map_split = feature_map[i,:, :]
        feature_map_split = np.expand_dims(feature_map_split,axis=2)
        if i != 0:
            feature_map_sum +=feature_map_split # 各通到相加
        feature_map_split = BI.transform(feature_map_split) #非线性差值，用于恢复图片尺寸。

        plt.imshow(feature_map_split)
        plt.savefig(out_path + str(i) + "_{}.jpg".format(type) )
        plt.yticks()
        plt.axis('off')

    feature_map_sum = BI.transform(feature_map_sum)
    plt.imshow(feature_map_sum)
    plt.savefig(out_path + "sum_{}.jpg".format(type))
    print("save sum_{}.jpg".format(type))   

可视化效果如下：

1.2 降维到3维或2维

现在主流的手段就是PCA和t-SNE。