pytorch网络提取的特征可视化

最新推荐文章于 2023-11-09 22:36:51 发布

Flying Stone

最新推荐文章于 2023-11-09 22:36:51 发布

阅读量183

点赞数

分类专栏：随手翻文章标签：可视化网络机器学习

本文链接：https://blog.csdn.net/weixin_42840360/article/details/117122678

版权

随手翻专栏收录该内容

27 篇文章 3 订阅

订阅专栏

        import cv2
        import numpy as np
        import os
        from PIL import Image
        import matplotlib.pyplot as plt
        print(x.shape)
        y=x.clone()
        y=y.cpu()
        y=y.numpy()
        x_show= np.max(y, axis=1)
        x_show =np.squeeze(x_show )
        x_show =(x_show - np.min(x_show)) / (np.max(x_show) - np.min(x_show))

        plt.figure("Image")  # 图像窗口名称
        plt.ion()
        plt.imshow(x_show)
        plt.axis('off')  # 关掉坐标轴为 off
        plt.title('b')  # 图像题目
        plt.show()

        #cv2.imshow('pic_name',x_show)
        #cv2.waitKey(0)